o Fh/'@sUddlZddlZddlZddlmZmZmZmZmZm Z m Z m Z ddl m Z mZdZz ddlZeejdZWn eyDdZdZYnwejejBZejejBZejZeedr^eejMZGd d d e ZGd d d e ZGd ddZ e e eefZ!ereneZ"e!e#d<dS)N)AnyDictFinalListOptionalTupleTypeUnion)AbstractResolver ResolveResult)ThreadedResolver AsyncResolverDefaultResolver getaddrinfoFAI_MASKc @s\eZdZdZddeejddfddZdej fde d e d ej de efd d Zdd dZdS)r zThreaded resolver. Uses an Executor for synchronous getaddrinfo() calls. concurrent.futures.ThreadPoolExecutor is used by default. NloopreturncCs|pt|_dSN)asyncioget_running_loop_loop)selfrrC/var/www/html/venv/lib/python3.10/site-packages/aiohttp/resolver.py__init__"szThreadedResolver.__init__rhostportfamilyc s|jj||tj|tdIdH}g}|D]I\}}}}}|tjkrEt|dkr'q|dr<|j|tIdH\} } t | }n|dd\} }n |tj ksLJ|\} }| t || |||t dq|S)N)typerflagshostnamerrrprotor )rrsocket SOCK_STREAM_AI_ADDRCONFIGAF_INET6len getnameinfo_NAME_SOCKET_FLAGSintAF_INETappendr _NUMERIC_SOCKET_FLAGS) rrrrinfoshosts_r%address resolved_host_portrrrresolve%s@     zThreadedResolver.resolvecsdSrrrrrrcloseQszThreadedResolver.closerrN)__name__ __module__ __qualname____doc__rrAbstractEventLooprr&r.strr- AddressFamilyrr r7r9rrrrr s ,r c @seZdZdZ ddeejdededdfddZd e j fd e d e d e j deefd dZd e j fd e d e d e deee effddZdddZdS)rz9Use the `aiodns` package to make asynchronous DNS lookupsNrargskwargsrcOsxtdurtd|p t|_d|_|s|r!tj|i||_dSt|_|j ||j|_t |jds:|j |_ dSdS)Nz Resolver requires aiodns library gethostbyname) aiodns RuntimeErrorrrr_manager DNSResolver _resolver_DNSResolverManager get_resolverhasattr_resolve_with_queryr7)rrrBrCrrrrXs  zAsyncResolver.__init__rrrrc sBz|jj||tj|tdIdH}Wn"tjjy4}zt|j dkr(|j dnd}t d||d}~wwg}|j D]]}|j } |j }|tjkrxt| dkrl| drl|j| ddg| ddRtIdH} | j} n| dd} | d}n|tjksJ| dd} | d}|t|| ||dtdq:|st dd|S)N)rrrr r DNS lookup failedr!rasciir#)rIrr&r'r(rEerrorDNSErrorr*rBOSErrornodesaddrrr)r+decoder,noder.r/r r0) rrrrrespexcmsgr2rVr4resultr5rrrr7qsV      zAsyncResolver.resolvec s|tjkrdnd}z |j||IdH}Wn"tjjy8}zt|jdkr,|jdnd}t d||d}~wwg}|D]} | || j ||dtj dq=|sUt dd|S)NAAAAAr rNrr#) r&r)rIqueryrErPrQr*rBrRr/rAI_NUMERICHOST) rrrrqtyperWrXrYr2rrrrrrMs.  z!AsyncResolver._resolve_with_querycsF|jr|j||jd|_d|_dS|jdur|jd|_dSr)rGrelease_resolverrrIcancelr8rrrr9s   zAsyncResolver.closerr:)r;r<r=r>rrr?rrr&r.r@r-rArr r7rrMr9rrrrrUsB  4 rcsneZdZUdZdZeded<dfdd ZdddZd d d e j dd fd dZ d d d e j ddfddZ Z S)rJzManager for aiodns.DNSResolver objects. This class manages shared aiodns.DNSResolver instances with no custom arguments across different event loops. N _instancercs(|jdurt||_|j|jSr)rcsuper__new___init)cls __class__rrres  z_DNSResolverManager.__new__cCst|_dSr)weakrefWeakKeyDictionary _loop_datar8rrrrfsz_DNSResolverManager._initclientrrzaiodns.DNSResolvercCsJ||jvrtj|d}t}||f|j|<n|j|\}}|||S)a&Get or create the shared aiodns.DNSResolver instance for a specific event loop. Args: client: The AsyncResolver instance requesting the resolver. This is required to track resolver usage. loop: The event loop to use for the resolver. )r)rlrErHrjWeakSetadd)rrmrresolver client_setrrrrKs  z _DNSResolverManager.get_resolvercCsN|j|}|dur dS|\}}|||s%|dur||j|=dSdS)zRelease the resolver for an AsyncResolver client when it's closed. Args: client: The AsyncResolver instance to release. loop: The event loop the resolver was using. N)rlgetdiscardrb)rrmrcurrent_loop_datarprqrrrras   z$_DNSResolverManager.release_resolver)rrJr:)r;r<r=r>rcr__annotations__rerfrr?rKra __classcell__rrrhrrJs&   rJr)$rr&rjtypingrrrrrrrr abcr r __all__rErLrHaiodns_default ImportErrorr^AI_NUMERICSERVr0NI_NUMERICHOSTNI_NUMERICSERVr, AI_ADDRCONFIGr(rr rrJ _DefaultTyperrurrrrs. (     :yC