| Current File : //opt/imunify360/venv/lib/python3.11/site-packages/im360/internals/__pycache__/geo.cpython-311.pyc |
�
�*(i� � � � d dl mZ d dlmZmZmZmZ d dlmZ d dl Z
d dl m Z d dl
mZ d dlmZ G d� d � � Zed
� � � ZdS ) � )�contextmanager)� IPv4Address� IPv4Network� IPv6Address� IPv6Network)�UnionN)�AddressNotFoundError)� CountryInfo)�IPc �z � e Zd Zd� Zdeeeeee f fd�Z
deeeeee f fd�Z deeeeee f fd�Z dS )�Readerc � � || _ d S )N)�_geoip2_reader)�self�
geoip2_readers �H/opt/imunify360/venv/lib/python3.11/site-packages/im360/internals/geo.py�__init__zReader.__init__
s � � +����� �addressc �� � t j |� � }n# t $ r Y dS w xY w | j � t |j � � � � }n# t $ r Y dS w xY w|r|j ndS )z�
Returns geo country information from max mind's db request
:param address: ip or network address
e.g. '4.4.4.4, 1.2.0.0/16, 2001:678:4c::/48'
:return: maxmind's geo info
N)r �adopt_to_ipvX_network�
ValueErrorr �country�str�network_addressr )r r �ip�objs r �getz
Reader.get s� � � ��)�'�2�2�B�B��� � � ��4�4� ���� ��%�-�-�c�"�2D�.E�.E�F�F�C�C��#� � � ��4�4� ���� "�+�s�{�{�t�+s � �
%�%�,A �
A$�#A$c �B � | � |� � }|r|j S dS )za
:param address: valid ipv4 address
:return: maxmind's id of the country
N)r �
geoname_id�r r � country_infos r �get_idz
Reader.get_id- s+ � � �x�x��(�(� � � +� �*� *��tr c �B � | � |� � }|r|j S dS )ze
:param address: valid ipv4 address
:return: country code in ISO-3166 format
N)r �iso_coder! s r �get_codezReader.get_code< s+ � � �x�x��(�(� � � )� �(� (��tr N)
�__name__�
__module__� __qualname__r r r r r r r r r# r& � r r r
r
s� � � � � � �,� ,� ,�,�� ��k�;� � C�
�,� ,� ,� ,�:
�� ��k�;� � C�
�
�
�
�
�
�� ��k�;� � C�
�
�
�
�
�
�
r r
c # � K � t j � t j � � 5 } t | � � V � ddd� � dS # 1 swxY w Y dS )zH
:return Reader obj: instance to be reused to it's method calls
N)�geoip2�databaser
r
�DB)r s r �readerr/ L s� � � � �
� � � � �� /� /� $�=��]�#�#�#�#�#�$� $� $� $� $� $� $� $� $� $� $� $���� $� $� $� $� $� $s �A � A�A)�
contextlibr � ipaddressr r r r �typingr �geoip2.databaser, �
geoip2.errorsr � defence360agent.contracts.configr
� defence360agent.utils.validater r
r/ r* r r �<module>r7 s� �� %� %� %� %� %� %� H� H� H� H� H� H� H� H� H� H� H� H� � � � � � � � � � � .� .� .� .� .� .� 8� 8� 8� 8� 8� 8� -� -� -� -� -� -�=� =� =� =� =� =� =� =�@ �$� $� � �$� $� $r