o i/'@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)selfrrJ/home/kim/smarthome/.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_hostZ_portrrrresolve%s@     zThreadedResolver.resolvecsdSrrrrrrcloseQszThreadedResolver.closerrN)__name__ __module__ __qualname____doc__rrAbstractEventLooprr&r.strr- AddressFamilyrr r6r8rrrrr 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_queryr6)rrrArBrrrrXs  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#)rHrr&r'r(rDerrorDNSErrorr*rAOSErrornodesaddrrr)r+decoder,noder.r/r r0) rrrrrespexcmsgr2rUr4resultr5rrrr6qsV      zAsyncResolver.resolvec s|tjkrdnd}z |j||IdH}Wn"tjjy8}zt|jdkr,|jdnd}t d||d}~wwg}|D]} | || j ||dtj dq=|sUt dd|S)NZAAAAAr rMrr#) r&r)rHqueryrDrOrPr*rArQr/rAI_NUMERICHOST) rrrrZqtyperVrWrXr2ZrrrrrrLs.  z!AsyncResolver._resolve_with_querycsF|jr|j||jd|_d|_dS|jdur|jd|_dSr)rFrelease_resolverrrHcancelr7rrrr8s   zAsyncResolver.closerr9)r:r;r<r=rrr>rrr&r.r?r-r@rr r6rrLr8rrrrrUsB  4 rcsneZdZUdZdZeded<dfdd ZdddZd d d e j dd fd dZ d d d e j ddfddZ Z S)rIzManager 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__rrras  z_DNSResolverManager.__new__cCst|_dSr)weakrefWeakKeyDictionary _loop_datar7rrrrbsz_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)rhrDrGrfWeakSetadd)rrirresolver client_setrrrrJs  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)rhgetdiscardr^)rrirZcurrent_loop_datarlrmrrrr]s   z$_DNSResolverManager.release_resolver)rrIr9)r:r;r<r=r_r__annotations__rarbrr>rJr] __classcell__rrrdrrIs&   rIr)$rr&rftypingrrrrrrrr abcr r __all__rDrKrGZaiodns_default ImportErrorr\AI_NUMERICSERVr0NI_NUMERICHOSTNI_NUMERICSERVr, AI_ADDRCONFIGr(rr rrIZ _DefaultTyperrprrrrs. (     :yC