Z @ s d d l Z d d l Z d d l m Z d
Z y d d l Z Wn e k
rX d Z Yn Xd Z Gd d d e Z Gd d d e Z e r e n e Z
d S) N )AbstractResolverThreadedResolver
AsyncResolverDefaultResolverFc @ sF e Z d Z d Z d d d Z d e j d d Z d d Z d S)
r zsUse Executor for synchronous getaddrinfo() calls, which defaults to
concurrent.futures.ThreadPoolExecutor.
Nc C s% | d k r t j } | | _ d S)N)asyncioget_event_loop_loop)selfloop r /resolver.py__init__ s zThreadedResolver.__init__r c s | j j | | d t j d | Id H} g } xY | D]Q \ } } } } } | j d | d | d d | d d | d | d t j i q6 W| S)
Ntypefamilyhostnamehostr portr protoflags)r getaddrinfosocket SOCK_STREAMappendAI_NUMERICHOST) r
r r r Zinfoshosts_r addressr r r
resolve s zThreadedResolver.resolvec s d S)Nr )r
r r r
close* s zThreadedResolver.close) __name__
__module__ __qualname____doc__r r AF_INETr r r r r r
r s c @ s[ e Z d Z d Z d d d Z d e j d d Z d e j d d Z d
d Z d S) r z9Use the `aiodns` package to make asynchronous DNS lookupsNc O sv | d k r t j } t d k r0 t d | | _ t j | d | | | _ t | j d sr | j | _ d S)Nz Resolver requires aiodns libraryr
gethostbyname)
r r aiodns RuntimeErrorr Z DNSResolver _resolverhasattr_resolve_with_queryr )r
r argskwargsr r r
r 1 s zAsyncResolver.__init__r c s y | j j | | Id H} Wnb t j j k
r } z< t | j d k r[ | j d n d } t | | WYd d } ~ Xn Xg } xE | j D]: } | j d | d | d | d | d d d t
j i q W| s t d | S)
Nr zDNS lookup failedr r r r r r r ) r( r% r& errorDNSErrorlenr+ OSErrorZ addressesr r r ) r
r r r respexcmsgr r r r r
r ? s (" zAsyncResolver.resolvec
s | t j k r d } n d } y | j j | | Id H} Wnb t j j k
r } z< t | j d k ry | j d n d } t | | WYd d } ~ Xn Xg } xE | D]= } | j
d | d | j d | d | d d
d t j i q W| st d | S) NZAAAAAr zDNS lookup failedr r r r r r r )
r AF_INET6r( Zqueryr& r- r. r/ r+ r0 r r r )
r
r r r Zqtyper1 r2 r3 r Zrrr r r
r* R s$ ("
z!AsyncResolver._resolve_with_queryc s
| j j S)N)r( cancel)r
r r r
r k s zAsyncResolver.close)
r r! r" r# r r r$ r r* r r r r r
r . s
)r r r ) r r abcr __all__r& ImportErrorZaiodns_defaultr r r r r r r
<module> s
A