jl 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 fdZ
deeeeee f fdZ deeeeee f fdZ 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 countrystrnetwork_addressr )r r ipobjs r getz
Reader.get s )'22BB 44 %--c"2D.E.EFFCC# 44 "+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_idr r country_infos r get_idz
Reader.get_id- s+ xx(( + * *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+ xx(( ) ( (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)geoip2databaser
r
DB)r s r readerr/ L s
/ / $=]#####$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $s A AA)
contextlibr ipaddressr r r r typingr geoip2.databaser,
geoip2.errorsr defence360agent.contracts.configr
defence360agent.utils.validater r
r/ r* r r r7 s % % % % % % H H H H H H H H H H H H . . . . . . 8 8 8 8 8 8 - - - - - -= = = = = = = =@ $ $ $ $ $r