U j7g.4@sdZddlmZmZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZmZdd lmZmZmZmZmZmZmZdd lmZdd lmZdd lmZmZmZm Z m!Z!Gd ddZ"dS)zQ Main API client asynchronous handler for fetching data from the IPinfo service. ) IPv4Address IPv6AddressN)APIError) DefaultCache)Details)RequestQuotaExceededErrorTimeoutExceededError)API_URLBATCH_MAX_SIZE CACHE_MAXSIZE CACHE_TTLREQUEST_TIMEOUT_DEFAULTBATCH_REQ_TIMEOUT_DEFAULT cache_key) handler_utils)is_bogon) continents countriescountries_currencies eu_countriescountries_flagsc@s`eZdZdZdddZddZddZdd d Zdedd fd d Z ddZ ddZ dddZ dS) AsyncHandlerz Allows client to request data for specified IP address asynchronously. Instantiates and maintains access to cache. NcKs||_|dpt|_|dp"t|_|dp2t|_|dpBt|_|dpRt|_|di|_d|jkrxt|jd<d|_ d |kr|d |_ n8|d i}d |krt |d <d |krt |d <t f||_ |d d|_dS)zi Initialize the Handler object with country name list and the cache initialized. rrrrZ continentrequest_optionstimeoutNcache cache_optionsmaxsizeZttlheaders) access_tokengetrrrrrrrhttpsessrr r rr)selfrkwargsrr$8/tmp/pip-unpacked-wheel-bv8pilsi/ipinfo/handler_async.py__init__*s(      zAsyncHandler.__init__cs|IdHdS)z Initializes internal aiohttp connection pool. This isn't _required_, as the pool is initialized lazily when needed. But in case you require non-lazy initialization, you may await this. This is idempotent. N)_ensure_aiohttp_readyr"r$r$r%initXs zAsyncHandler.initcs |jr|jIdHd|_dS)z Deinitialize the async handler. This is required in case you need to let go of the memory/state associated with the async handler in a long-running process. This is idempotent. N)r!closer(r$r$r%deinitcs zAsyncHandler.deinitc sx|t|tst|tr"|j}|r@t|r@|dd}t|Sz|jt|}t|WSt k rlYnXt }|r|d|7}t |j |j}i}|dk r||d<|jj|fd|i|4IdHr}|jdkrt|jdkr&|j} |jd } | d kr|IdH} n d |i} t| | |IdH}W5QIdHRXt ||j|j|j|j|j||jt|<t|S) z9Get details for specified IP address as a Details object.Tipbogon/Nrriz Content-Typeapplication/jsonerror)r' isinstancerrexplodedrrrrKeyErrorr r get_headersrrr!r statusrjsontextrformat_detailsrrrrr) r" ip_addressrdetails cached_ipaddrurlrZreq_optsrespZ error_code content_typeZerror_responser$r$r% getDetailspsT     "      zAsyncHandler.getDetailsTc sdkrtig|D]X}t|ts8t|tr>|j}zjt|}||<Wq tk rv |Yq Xq sS|dk rt }t dt jjdd<fddtdtD} zxtj| |tjdIdH\} } | s WS| D]6} z| | IdHWntjk r@YnXqt tWStk r} zt | WYSd} ~ XYnXS) a Get details for a batch of IP addresses at once. There is no specified limit to the number of IPs this function can accept; it can handle as much as the user can fit in RAM (along with all of the response data, which is at least a magnitude larger than the input list). The input list is broken up into batches to abide by API requirements. The batch size can be adjusted with `batch_size` but is clipped to `BATCH_MAX_SIZE`. Defaults to `BATCH_MAX_SIZE`. For each batch, `timeout_per_batch` indicates the maximum seconds to spend waiting for the HTTP request to complete. If any batch fails with this timeout, the whole operation fails. Defaults to `BATCH_REQ_TIMEOUT_DEFAULT` seconds. `timeout_total` is a seconds-denominated hard-timeout for the time spent in HTTP operations; regardless of whether all batches have succeeded so far, if `timeout_total` is reached, the whole operation will fail by raising `TimeoutExceededError`. Defaults to being turned off. `raise_on_fail`, if turned off, will return any result retrieved so far rather than raise an exception when errors occur, including timeout and quota errors. Defaults to on. The concurrency level is currently unadjustable; coroutines will be created and consumed for all batches at once. N/batchr1 content-typec s,g|]$}||qSr$) _do_batch_req).0i batch_sizerlookup_addresses raise_on_failresultr"timeout_per_batchr>r$r% s z0AsyncHandler.getBatchDetails..r)rZ return_when)r'r r3rrr4rrr5appendtimer rr6rrrangelenasynciowaitZFIRST_EXCEPTIONcancelZCancelledErrorreturn_or_failr Exception)r" ip_addressesrHrLZ timeout_totalrJr;r= start_timereqs_pendingcoer$rGr%getBatchDetailss`(     $zAsyncHandler.getBatchDetailsc sz$|jj|t|||dIdH}Wn4tk rX}zt||dWYSd}~XYnXz|jdkrlt| Wn4tk r}zt||dWYSd}~XYnX|IdH} | D]>\} } t | t rt | |j|j|j|j|j| |jt| <q|| dS)zV Coroutine which will do the actual POST request for getBatchDetails. )datarrNr0)r!postr8dumpsrVrrUr7rraise_for_statusitemsr3dictr:rrrrrrrupdate) r"chunkr>rrLrJrKr?r]Z json_respr;r<r$r$r%rD!s8$  $ zAsyncHandler._do_batch_reqcCs.|jr dStj|jdd}tj|d|_dS)z.Ensures aiohttp internal state is initialized.Nr)total)r)r!aiohttpZ ClientTimeoutr ClientSession)r"rr$r$r%r'Ksz"AsyncHandler._ensure_aiohttp_readyc sT|dkr t}ig}|D]z}t|ts0t|tr6|j}|rXt|rX|dd}t|Vqzjt|}||<Wqt k r| |YqXq|s Vt dt jjdd<fdd}tdt||D]j} || | |} || IdH D]>\}}t|tr@t |jjjjj||fVqqdS) NTr,rBr1rCc s|tjd4IdHX}|j|dIdH}||IdH}|D]\}}|jt|<||<qHW5QIdHRXdS)N)r)r8)rhrir`rbr8rcrr)batchsessionresponseZ json_responser;r<rresultsr"r>r$r% process_batchusz7AsyncHandler.getBatchDetailsIter..process_batchr)r r3rrr4rrrrr5rNrcr rr6rrrPrQrdr:rrrrr) r"rWrHrJrIr;r<r=rorFrjr$rmr%getBatchDetailsIterSsL        z AsyncHandler.getBatchDetailsIter)N)NN)NT) __name__ __module__ __qualname____doc__r&r)r+rArr^rDr'rpr$r$r$r%r$s .  = w* r)#rt ipaddressrrrRr8rOrhr2rZ cache.defaultrr<r exceptionsrr rr r r r rrrr.rr_rrrrrrr$r$r$r%s   $