Edit File: geo.cpython-311.pyc
� ��\h� � � � 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