U

_�_�C�@s
dZd
dl
Z
d
dlZd
dlmZmZmZmZmZm	Z	m
Z

d
dlZd
dlZd
dl
Z
d
dlZ
d
dlZd
dlZd
dlmZmZmZmZmZmZmZ
d
dlmZmZmZ
d
dlmZ
dej�dejj ��Z!dej�de
j"�#���Z$Gd	d
�d
�Z%Gdd�de%�Z&Gd
d�de%�Z'dS)a{
============================
WebServices Client API
============================

This class provides a client API for all the GeoIP2 Precision web service end
points. The end points are Country, City, and Insights. Each end point returns
a different set of data about an IP address, with Country returning the least
data and Insights the most.

Each web service end point 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 web service does not return a particular piece of data for an IP
address, the associated attribute is not populated.

The web 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)�Any�Dict�cast�List�Optional�Type�Union)�AddressNotFoundError�AuthenticationError�GeoIP2Error�	HTTPError�InvalidRequestError�OutOfQueriesError�PermissionRequiredError)�City�Country�Insights)
�	IPAddresszGeoIP2-Python-Client/�
 c@s
eZ
dZUeed
<eed<eed<eeed<eed<eeeeeeedd�dd	�Z	ee
ed
�dd�Zeeee
d
�dd��
Zeeeeed�dd�Zeeeeed�dd�Zeeeeeeeeeeefd�dd��
Zeeeeeed�dd��
Zeeeeeed�dd��
ZdS)�
BaseClient�_account_idZ_host�_license_key�_locales�_timeoutN)�
account_id�license_key�host�locales�timeout�returncCsJ|d
krdg
}||_t
|
t�r"|
nt|
�
|_||_d|�d�|_||_d
S)zConstruct a Client.NZenzhttps://z/geoip/v2.1)r�
isinstance�bytes�strrr�	_base_urir)�selfrrrrr�r%�B/opt/alt/python38/lib/python3.8/site-packages/geoip2/webservice.py�__init__Cs

�

zBaseClient.__init__)�path�
ip_addressrcCs(|d
krt�
|�
}d�|j|
t|�
g�
S)N�me�
/)�	ipaddressr)�joinr#r")r$r(r)r%r%r&�_uriZs



zBaseClient._uri)�body�urirc
CsVzt�
|�
WStk
rP
}
z&td
|
�d�d�|j�
d|
�|�W5d}~XYnXdS)NzReceived a 200 response for z, but could not decode the response as JSON: �, ��)�json�loads�
ValueErrorrr-�args)r/r0�exr%r%r&�_handle_success_s





�
��zBaseClient._handle_success)�status�content_typer/r0rcCs\d
|
k
rdkr(n
n|�|
|||�Sd|
k
r<dkrNn
n|�
|
||�S|�|
||�S)Ni�
i�
iX)�_exception_for_4xx_status�_exception_for_5xx_status�_exception_for_non_200_status)r$r9r:r/r0r%r%r&�_exception_for_errorls




zBaseClient._exception_for_errorc
Cs�|s td
|
�d|�d�|
||�S|�
d�
dkrNtd
|
�d|�d|��|
||�Szt�|�
}WnRtk
r�
}
z4td
|
�d|�d�d	d
�|j�
|
||�WY�Sd}~XYn<Xd|kr�d|kr�|�|�d�
|�d�
|
|�Std
|
||�SdS)NzReceived a z error for z with no body.r3���z for z with the following body: z but it did not include zthe expected JSON body: r1�code�errorzAResponse contains JSON but it does not specify code or error keys)	r�findr3r4r5r-r6� _exception_for_web_service_error�get)r$r9r:r/r0�decoded_bodyr7r%r%r&r;usP




�




�




�
�

�	

�



�z$BaseClient._exception_for_4xx_status)�messager@r9r0rcCsN|
d
krt|�
S|
dkr t
|�
S|
dkr0t|�
S|
dkr@t|�
St||
||�S)N)ZIP_ADDRESS_NOT_FOUNDZIP_ADDRESS_RESERVED)ZACCOUNT_ID_REQUIREDZACCOUNT_ID_UNKNOWNZAUTHORIZATION_INVALIDZLICENSE_KEY_REQUIREDZUSER_ID_REQUIREDZUSER_ID_UNKNOWN)ZINSUFFICIENT_FUNDSZOUT_OF_QUERIESZPERMISSION_REQUIRED)r	r
rrr
)rFr@r9r0r%r%r&rC�s






z+BaseClient._exception_for_web_service_error)r9r0r/rcCstd
|�d|
��||
|�S)NzReceived a server error (�) for �
r�r9r0r/r%r%r&r<�s



�z$BaseClient._exception_for_5xx_statuscCstd
|�d|
��||
|�S)Nz(Received a very surprising HTTP status (rGrHrIr%r%r&r=�s



�z(BaseClient._exception_for_non_200_status)�__name__�
__module__�__qualname__r"�__annotations__r�float�intrr'rr.�staticmethodrr8rr>r;rr
r	rrr
rCrr<r=r%r%r%r&r<sj











�


�

�(





���
�

�rc	s�eZ
dZUd
Zejed<eeed<de	eeee
eeeedd��f
dd	�
Zd e
ed�dd
�Zd!e
ed�dd�Zd"e
ed�dd�Zejd�
dd�Zeeeeeeeefe
eeeefd�dd�Zdd�Zdd�
dd�Zddddd�dd�Z�ZS)#�AsyncClientaUAn 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 GeoIP2
      Precision, 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�_proxy�geoip.maxmind.comN�<�rrrrr�proxyrcst��
|
||||�
||_dS�
N)�superr'rS�r$rrrrrrW�
�	__class__r%r&r'
s	




�zAsyncClient.__init__r*�r)rc�stt
|�d
tjj
|
�IdH�S)�
Call City 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

        �cityN�rr�
_response_for�geoip2�models�r$r)r%r%r&r_
s

�zAsyncClient.cityc�stt
|�d
tjj
|
�IdH�S)�(
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

        �countryN�rrrarbrcrdr%r%r&rf$
s


�zAsyncClient.countryc�stt
|�d
tjj
|
�IdH�S)�
Call the Insights endpoint with the specified IP.

        Insights is only supported by GeoIP2 Precision. 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

        �insightsN�rrrarbrcrdr%r%r&ri3
s


�zAsyncClient.insights�
rc

�s>t|d
�s8t
jt
�|j|j�dtd�t
j|jd�
d�|_	|j	S)NrR�application/json)�Accept�
User-Agent)
Ztotal)�auth�headersr)
�hasattr�aiohttp�
ClientSessionZ	BasicAuthrr�_AIOHTTP_UAZ
ClientTimeoutrrR�
r$r%r%r&�_sessionE
s





�zAsyncClient._session�r(�model_classr)rc�s�|�|
|�}|�
�IdH}|j||jd
�IdH4IdH�`}|j}|j}|��IdH}	|dkrl|�|||	|��
|�|	|�}
||
|j	d�W5QIdHR�SQIdHRXdS)N)
rWr2�
r)
r.rvrDrSr9r:�textr>r8r)r$r(rxr)r0Zsession�responser9r:r/rEr%r%r&raO
s

 






zAsyncClient._response_forc

�st|d
�r|j
��IdH
dS)�fClose underlying session

        This will close the session and any associated connections.
        rRN)rqrR�closerur%r%r&r}`
s

zAsyncClient.closec


�s|SrXr%rur%r%r&�
__aenter__h
s
zAsyncClient.__aenter__��exc_type�	exc_value�	tracebackrc�s|��IdH
dSrX�
r}�r$r�r�r�r%r%r&�	__aexit__k
s
zAsyncClient.__aexit__)rTNrUN)
r*)
r*)
r*)rJrKrL�__doc__rrrsrMrr"rOrrNr'rrr_rrfrrirvrrrar}r~r��
__classcell__r%r%r[r&rQ�s8

0




�






�


�rQc	s�eZ
dZUd
Zejed<eee	e	fed<de
e	e	eee	eee	dd��f
dd	�
Z
deed�dd
�Zdeed�dd�Zd eed�dd�Ze	eeeeeeefeeeeefd�dd�Zdd�Zdd�
dd�Zddddd�dd�Z�ZS)!�Clienta[A 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 GeoIP2
      Precision, 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``.


    rv�_proxiesrTNrUrVcsdt��
|
||||�
t��|_|j|jf|j_d
|jjd<t	|jjd<|dkrVd|_
n
d|i
|_
dS)NrlrmrnZhttps)rYr'�requests�Sessionrvrrrorp�_REQUEST_UAr�rZr[r%r&r'�
s	






zClient.__init__r*r]cCstt
|�d
tjj
|
��S)r^r_r`rdr%r%r&r_�
s
zClient.citycCstt
|�d
tjj
|
��S)rerfrgrdr%r%r&rf�
s

�zClient.countrycCstt
|�d
tjj
|
��S)rhrirjrdr%r%r&ri�
s

�zClient.insightsrwc
Csj|�|
|�}|j
j||j|jd
�}|j}|jd}|j}|dkrP|�||||��
|�	||�}	||	|j
d�S)N)ZproxiesrzContent-Typer2ry)r.rvrDr�rZstatus_coderprzr>r8r)
r$r(rxr)r0r{r9r:r/rEr%r%r&ra�
s








zClient._response_forc

Cs|j�
�
d
S)r|N)rvr}rur%r%r&r}�
szClient.closerkc


Cs|SrXr%rur%r%r&�	__enter__�
s
zClient.__enter__rcCs|��
dSrXr�r�r%r%r&�__exit__�
s
zClient.__exit__)rTNrUN)
r*)
r*)
r*)rJrKrLr�r�r�rMrrr"rOrrNr'rrr_rrfrrirrrar}r�r�r�r%r%r[r&r�o
s6

1




�






�


�r�)(r�r,r3�typingrrrrrrrrrZaiohttp.httpr�Zrequests.utilsrbZ
geoip2.modelsZ
geoip2.errorsr	r
rrr
rrrrrZgeoip2.typesr�__version__ZhttpZSERVER_SOFTWARErtZutilsZdefault_user_agentr�rrQr�r%r%r%r&�<module>
s*

$




$	
��