U [eC@s dZddlZddlZddlmZmZmZmZmZm Z m Z ddl Z ddl Z ddl Z ddlZ ddlZddlZddlmZmZmZmZmZmZmZddlmZmZmZddlmZdejde jj Z!dejde j"#Z$Gd d d Z%Gd d d e%Z&Gd dde%Z'dS)aV ============================ WebServices Client API ============================ This class provides a client API for all the GeoIP2 web services. The web services are Country, City Plus, and Insights. Each service returns a different set of data about an IP address, with Country returning the least data and Insights the most. Each service is represented by a different model class, and these model classes in turn contain multiple record classes. The record classes have attributes which contain data about the IP address. If the service does not return a particular piece of data for an IP address, the associated attribute is not populated. The service may not return any information for an entire record, in which case all of the attributes for that record class will be empty. SSL --- Requests to the web service are always made with SSL. N)AnyDictcastListOptionalTypeUnion)AddressNotFoundErrorAuthenticationError GeoIP2Error HTTPErrorInvalidRequestErrorOutOfQueriesErrorPermissionRequiredError)CityCountryInsights) IPAddresszGeoIP2-Python-Client/ c @seZdZUeed<eed<eed<eeed<eed<eeeeeeedddd Z ee ed d d Z e eee d ddZeeeeedddZeeeeedddZe eeeeeeeeeefdddZe eeeeedddZe eeeeedddZdS) BaseClient _account_idZ_host _license_key_locales_timeoutN) account_id license_keyhostlocalestimeoutreturncCsJ|dkrdg}||_t|tr"|nt||_||_d|d|_||_dS)zConstruct a Client.Nenzhttps://z /geoip/v2.1)r isinstancebytesstrrr _base_urir)selfrrrrrr&S/var/www/html/services/stratfitenv/lib/python3.8/site-packages/geoip2/webservice.py__init__Cs zBaseClient.__init__)path ip_addressrcCs(|dkrt|}d|j|t|gS)Nme/) ipaddressr*joinr$r#)r%r)r*r&r&r'_uriZs zBaseClient._uri)bodyurirc CsVz t|WStk rP}z&td|dd|jd||W5d}~XYnXdS)NzReceived a 200 response for z, but could not decode the response as JSON: , )jsonloads ValueErrorr r.args)r0r1exr&r&r'_handle_success_s   zBaseClient._handle_success)status content_typer0r1rcCs\d|krdkr(nn|||||Sd|krs z(BaseClient._exception_for_non_200_status)__name__ __module__ __qualname__r#__annotations__rfloatintrr(rr/ staticmethodrr9r r?r<rr r rrr rDr r=r>r&r&r&r'r<sj       ( rc seZdZUdZejed<eeed<de eeee ee eeddfdd Z d e ed d d Zd!e ed ddZd"e ed ddZejdddZeeeeeeeefe eeeefdddZddZddddZdddddddZZS)# AsyncClienta[An async GeoIP2 client. It accepts the following required arguments: :param account_id: Your MaxMind account ID. :param license_key: Your MaxMind license key. Go to https://www.maxmind.com/en/my_license_key to see your MaxMind account ID and license key. The following keyword arguments are also accepted: :param host: The hostname to make a request against. This defaults to "geoip.maxmind.com". To use the GeoLite2 web service instead of the GeoIP2 web service, set this to "geolite.info". :param locales: This is list of locale codes. This argument will be passed on to record classes to use when their name properties are called. The default value is ['en']. The order of the locales is significant. When a record class has multiple names (country, city, etc.), its name property will return the name in the first locale that has one. Note that the only locale which is always present in the GeoIP2 data is "en". If you do not include this locale, the name property may end up returning None even when the record has an English name. Currently, the valid locale codes are: * de -- German * en -- English names may still include accented characters if that is the accepted spelling in English. In other words, English does not mean ASCII. * es -- Spanish * fr -- French * ja -- Japanese * pt-BR -- Brazilian Portuguese * ru -- Russian * zh-CN -- Simplified Chinese. :param timeout: The timeout in seconds to use when waiting on the request. This sets both the connect timeout and the read timeout. The default is 60. :param proxy: The URL of an HTTP proxy to use. It may optionally include a basic auth username and password, e.g., ``http://username:password@host:port``. _existing_session_proxygeoip.maxmind.comN<rrrrrproxyrcst|||||||_dSN)superr(rTr%rrrrrrX __class__r&r'r(s zAsyncClient.__init__r+r*rcstt|dtjj|IdHS)Call City Plus endpoint with the specified IP. :param ip_address: IPv4 or IPv6 address as a string. If no address is provided, the address that the web service is called from will be used. :returns: :py:class:`geoip2.models.City` object cityNrr _response_forgeoip2modelsr%r*r&r&r'r`s zAsyncClient.citycstt|dtjj|IdHS)(Call the GeoIP2 Country endpoint with the specified IP. :param ip_address: IPv4 or IPv6 address as a string. If no address is provided, the address that the web service is called from will be used. :returns: :py:class:`geoip2.models.Country` object countryNrrrbrcrdrer&r&r'rg$s zAsyncClient.countrycstt|dtjj|IdHS)Call the Insights endpoint with the specified IP. Insights is only supported by the GeoIP2 web service. The GeoLite2 web service does not support it. :param ip_address: IPv4 or IPv6 address as a string. If no address is provided, the address that the web service is called from will be used. :returns: :py:class:`geoip2.models.Insights` object insightsNrrrbrcrdrer&r&r'rj3s zAsyncClient.insightsrcs>t|ds8tjt|j|jdtdtj|jdd|_ |j S)NrSapplication/json)Accept User-Agent)total)authheadersr) hasattraiohttp ClientSessionZ BasicAuthrr _AIOHTTP_UAZ ClientTimeoutrrSr%r&r&r'_sessionEs  zAsyncClient._sessionr) model_classr*rc s|||}|IdH}|j||jdIdH4IdH`}|j}|j}|IdH} |dkrl|||| ||| |} || |j dW5QIdHRSQIdHRXdS)N)rXr3r) r/rxrErTr:r;textr?r9r) r%r)rzr*r1sessionresponser:r;r0rFr&r&r'rbOs   zAsyncClient._response_forcst|dr|jIdHdS)fClose underlying session This will close the session and any associated connections. rSN)rsrScloserwr&r&r'r`s zAsyncClient.closecs|SrYr&rwr&r&r' __aenter__hszAsyncClient.__aenter__exc_type exc_value tracebackrcs|IdHdSrYrr%rrrr&r&r' __aexit__kszAsyncClient.__aexit__)rUNrVN)r+)r+)r+)rKrLrM__doc__rtrurNrr#rPrrOr(rrr`rrgrrjrxrrrbrrr __classcell__r&r&r\r'rRs8 0     rRc seZdZUdZejed<eee e fed<de e e ee e e ee ddfdd Z deed d d Zdeed ddZd eed ddZe eeeeeeefeeeeefdddZddZddddZdddddddZZS)!ClientaaA synchronous GeoIP2 client. It accepts the following required arguments: :param account_id: Your MaxMind account ID. :param license_key: Your MaxMind license key. Go to https://www.maxmind.com/en/my_license_key to see your MaxMind account ID and license key. The following keyword arguments are also accepted: :param host: The hostname to make a request against. This defaults to "geoip.maxmind.com". To use the GeoLite2 web service instead of the GeoIP2 web service, set this to "geolite.info". :param locales: This is list of locale codes. This argument will be passed on to record classes to use when their name properties are called. The default value is ['en']. The order of the locales is significant. When a record class has multiple names (country, city, etc.), its name property will return the name in the first locale that has one. Note that the only locale which is always present in the GeoIP2 data is "en". If you do not include this locale, the name property may end up returning None even when the record has an English name. Currently, the valid locale codes are: * de -- German * en -- English names may still include accented characters if that is the accepted spelling in English. In other words, English does not mean ASCII. * es -- Spanish * fr -- French * ja -- Japanese * pt-BR -- Brazilian Portuguese * ru -- Russian * zh-CN -- Simplified Chinese. :param timeout: The timeout in seconds to use when waiting on the request. This sets both the connect timeout and the read timeout. The default is 60. :param proxy: The URL of an HTTP proxy to use. It may optionally include a basic auth username and password, e.g., ``http://username:password@host:port``. rx_proxiesrUNrVrWcsdt|||||t|_|j|jf|j_d|jjd<t |jjd<|dkrVd|_ n d|i|_ dS)Nrmrnrohttps) rZr(requestsSessionrxrrrqrr _REQUEST_UArr[r\r&r'r(s    zClient.__init__r+r^cCstt|dtjj|S)r_r`rarer&r&r'r`s z Client.citycCstt|dtjj|S)rfrgrhrer&r&r'rgs zClient.countrycCstt|dtjj|S)rirjrkrer&r&r'rjs zClient.insightsryc Csj|||}|jj||j|jd}|j}|jd}|j}|dkrP|||||| ||} || |j dS)N)proxiesrz Content-Typer3r{) r/rxrErr status_coderrr|r?r9r) r%r)rzr*r1r~r:r;r0rFr&r&r'rbs   zClient._response_forcCs|jdS)rN)rxrrwr&r&r'rsz Client.closerlcCs|SrYr&rwr&r&r' __enter__szClient.__enter__rcCs |dSrYrrr&r&r'__exit__szClient.__exit__)rUNrVN)r+)r+)r+)rKrLrMrrrrNrrr#rPrrOr(rrr`rrgrrjrrrbrrrrr&r&r\r'ros6 1    r)(rr-r4typingrrrrrrrrtZ aiohttp.httprZrequests.utilsrcZ geoip2.modelsZ geoip2.errorsr r r r r rrrrrZ geoip2.typesr __version__httpSERVER_SOFTWARErvutilsdefault_user_agentrrrRrr&r&r&r's*$$