o m=8j&@sUddlZddlZddlZddlmZmZmZddlmZm Z dZ z ddl Z e e j dZWn ey:dZ dZYnwejejBZejejBZejZe edrTeejMZGd d d eZGd d d eZGd ddZeeeBZerueneZeed<dS)N)AnyFinalOptional)AbstractResolver ResolveResult)ThreadedResolver AsyncResolverDefaultResolver getaddrinfoFAI_MASKc @s\eZdZdZddejdBddfddZdejfde d e d ej de e fd d Zdd dZdS)rzThreaded resolver. Uses an Executor for synchronous getaddrinfo() calls. concurrent.futures.ThreadPoolExecutor is used by default. NloopreturncCs|pt|_dSN)asyncioget_running_loop_loop)selfr rW/home/nk/hobo-godmode/plappi-mvp/.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)rr socket 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__rAbstractEventLooprr!r)strr( AddressFamilylistrr2r4rrrrrs ,rc @seZdZdZ ddejdBdededdfddZd ej fd e d e d ej de efd dZd ej fd e d e d e de ee effddZdddZdS)r z9Use the `aiodns` package to make asynchronous DNS lookupsNr argskwargsrcOsxtdurtd|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_queryr2)rr r>r?rrrrXs  zAsyncResolver.__init__rrrrc sBz|jj||tj|tdIdH}Wn"tjjy4}zt|j dkr(|j dnd}t d||d}~wwg}|j D]]}|j } |j tjkrvt| dkrj| drj|j| ddg| ddRtIdH} | j} n| dd} | d}n|j tjks~J| dd} | d}|t|| ||j dtdq:|st dd|S)N)rrrrrDNS lookup failedrrasciir)rEr r!r"r#rAerrorDNSErrorr%r>OSErrornodesaddrrr$r&decoder'noder)r*rr+) rrrrrespexcmsgr-rRr/resultr0rrrr2qsT      zAsyncResolver.resolvec s|tjkrdnd}z |j||IdH}Wn"tjjy8}zt|jdkr,|jdnd}t d||d}~wwg}|D]} | || j ||dtj dq=|sUt dd|S)NAAAAArrJrr) r!r$rEqueryrArLrMr%r>rNr*rAI_NUMERICHOST) rrrrqtyperSrTrUr-rrrrrrIs.  z!AsyncResolver._resolve_with_querycsF|jr|j||jd|_d|_dS|jdur|jd|_dSr)rCrelease_resolverrrEcancelr3rrrr4s   zAsyncResolver.closerr5)r6r7r8r9rr:rrr!r)r;r(r<r=rr2dictrIr4rrrrr UsB  3 r csneZdZUdZdZeded<dfdd ZdddZd d d e j dd fd dZ d d d e j ddfddZ Z S)rFzManager for aiodns.DNSResolver objects. This class manages shared aiodns.DNSResolver instances with no custom arguments across different event loops. N _instancercs(|jdurt||_|j|jSr)r`super__new___init)cls __class__rrrbs  z_DNSResolverManager.__new__cCst|_dSr)weakrefWeakKeyDictionary _loop_datar3rrrrcsz_DNSResolverManager._initclientr r zaiodns.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 )rirArDrgWeakSetadd)rrjr resolver client_setrrrrGs  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)rigetdiscardr^)rrjr current_loop_datarmrnrrrr]s   z$_DNSResolverManager.release_resolver)rrFr5)r6r7r8r9r`r__annotations__rbrcrr:rGr] __classcell__rrrerrFs&   rFr ) rr!rgtypingrrrabcrr__all__rArHrDaiodns_default ImportErrorrZAI_NUMERICSERVr+NI_NUMERICHOSTNI_NUMERICSERVr' AI_ADDRCONFIGr#r rr rFr _DefaultTyper rrrrrrs.      :x C