U ™j7g…1ã@sÂdZddlmZmZddlZddlZddlmZddlm Z ddl m Z ddl m Z mZdd lmZmZmZmZmZmZmZdd lmZdd lmZdd lmZmZmZmZmZGd d„dƒZ dS)zD Main API client handler for fetching data from the IPinfo service. é)Ú IPv4AddressÚ IPv6AddressNé)ÚAPIError)Ú DefaultCache)ÚDetails)ÚRequestQuotaExceededErrorÚTimeoutExceededError)ÚAPI_URLÚBATCH_MAX_SIZEÚ CACHE_MAXSIZEÚ CACHE_TTLÚREQUEST_TIMEOUT_DEFAULTÚBATCH_REQ_TIMEOUT_DEFAULTÚ cache_key)Ú handler_utils)Úis_bogon)Ú continentsÚ countriesÚcountries_currenciesÚ eu_countriesÚcountries_flagsc@sHeZdZdZddd„Zddd„Zdeddfdd „Zd d „Zdd d „Z dS)ÚHandlerzq Allows client to request data for specified IP address. Instantiates and maintains access to cache. NcKsÖ||_| d¡pt|_| d¡p"t|_| d¡p2t|_| d¡pBt|_| d¡pRt|_| di¡|_d|jkrxt|jd<d|krŒ|d|_ n8| d i¡}d |kr¨t |d <d |kr¸t |d <t f|Ž|_ | d d ¡|_ d S)zi Initialize the Handler object with country name list and the cache initialized. rrrrZ continentÚrequest_optionsÚtimeoutÚcacheÚ cache_optionsÚmaxsizeZttlÚheadersN)Ú access_tokenÚgetrrrrrrrrr r rr)ÚselfrÚkwargsr©r#ú2/tmp/pip-unpacked-wheel-bv8pilsi/ipinfo/handler.pyÚ__init__(s& ÿ     zHandler.__init__c CsTt|tƒst|tƒr|j}|rBt|ƒrBi}||d<d|d<t|ƒSz|jt|ƒ}t|ƒWStk rnYnX|j –}|dk rˆ||d<t }|rœ|d|7}t   |j |j¡}tj|fd|i|—Ž}|jdkrÒtƒ‚|jd kr|j} |j d ¡} | d kr| ¡} n d |ji} t| | ƒ‚| ¡}t  ||j|j|j|j|j¡||jt|ƒ<t|ƒS) z» Get details for specified IP address as a Details object. If `timeout` is not `None`, it will override the client-level timeout just for this operation. ÚipTÚbogonNrú/ré­iz Content-Typeúapplication/jsonÚerror)Ú isinstancerrÚexplodedrrrrÚKeyErrorrr rÚ get_headersrrÚrequestsr Ú status_coderÚjsonÚtextrÚformat_detailsrrrrr) r!Ú ip_addressrÚdetailsÚ cached_ipaddrÚreq_optsÚurlrÚresponseZ error_codeÚ content_typeZerror_responser#r#r$Ú getDetailsSsV ÿ          úzHandler.getDetailsTc Csj|dkr t}i}g}|D]†}t|tƒs0t|tƒr6|j}|rdt|ƒrdi} || d<d| d<t| ƒ||<qz|jt|ƒ} | ||<Wqt k rœ|  |¡YqXqt |ƒdkr°|S|dk rÀt   ¡} |j d|i–} td} t |j|j¡}d|d <tdt |ƒ|ƒD]d}|dk r4t   ¡| |kr4t |tƒ|¡S||||…}ztj| f||d œ| —Ž}Wn:tk rš}zt |||¡WY¢Sd}~XYnXz|jd kr°tƒ‚| ¡Wn:tk rô}zt |||¡WY¢Sd}~XYnX| ¡}| ¡D]\}} | |jt|ƒ<q| |¡| ¡D]0}t|t ƒr2t !||j"|j#|j$|j%|j&¡q2qþ|S) 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. Nr&Tr'rrú/batchr*ú content-type©r2rr))'r r,rrr-rrrrr.ÚappendÚlenÚtimerr rr/rrÚrangeÚreturn_or_failr r0ÚpostÚ Exceptionr1rÚraise_for_statusr2ÚitemsÚupdateÚvaluesÚdictr4rrrrr)r!Ú ip_addressesÚ batch_sizeZtimeout_per_batchZ timeout_totalÚ raise_on_failÚresultÚlookup_addressesr5r6r7Ú start_timer8r9rÚiÚchunkr:ÚeZ json_responseÚdetailr#r#r$ÚgetBatchDetails”sŽ% ÿ   ÿþÿÿÿÿ (  (   ú zHandler.getBatchDetailscCs†g}|D](}t|tƒs t|tƒr&|j}| |¡q|j–}t›d}t d|j ¡}d|d<t j |f||dœ|—Ž}|  ¡|  ¡dS)zi Gets a URL to a map on https://ipinfo.io/map given a list of IPs (max 500,000). z /map?cli=1Nr*r>r?Z reportUrl)r,rrr-r@rr rr/rr0rErGr2)r!ÚipsZip_strsr&r8r9rr:r#r#r$ÚgetMaps&  ÿÿÿzHandler.getMapc csæ|dkr t}i}g}|D]„}t|tƒs0t|tƒr6|j}|rbt|ƒrbi}||d<d|d<t|ƒVqz|jt|ƒ}|||<Wqt k rš|  |¡YqXqt |ƒdkr¶t |  ¡ƒ‚td} t |j|j¡} d| d<tdt |ƒ|ƒD]ú} || | |…} ztj| | | d} Wn(tk r6}z|‚W5d}~XYnXz| jd krLtƒ‚|  ¡Wn8tk rŽ}zt ||¡WY¢Sd}~XYnX|  ¡}t ||j|j|j|j |j!¡| D]$}| "|¡}||jt|ƒ<|VqºqædS) Nr&Tr'rr=r*r>r?r))#r r,rrr-rrrrr.r@rAÚ StopIterationrHr rr/rrrCr0rErFr1rrGrDr2r4rrrrrr )r!rLrMrNrOrPr5r6r7r9rrRÚbatchr:rTr&rUr#r#r$ÚgetBatchDetailsIter0sd ÿ       &ú zHandler.getBatchDetailsIter)N)NN)NT) Ú__name__Ú __module__Ú __qualname__Ú__doc__r%r<rrVrXr[r#r#r#r$r"s + Dú ür)!r_Ú ipaddressrrrBr0r+rZ cache.defaultrr6rÚ exceptionsrr rr r r r rrrÚr'rÚdatarrrrrrr#r#r#r$Ús   $