U
9`
@ s d dl Z d dlZd dlZd dlZd dlmZmZmZmZ ddl m
Z
dZ dZ dZ
dZejejejejejejejejejejejejd Zej ej ej ej ej d Z!G d
d d Z"dS ) N)AnyListOptionalSet )errorz3.0.0) DNSResolverr ) AZAAAAANYZCAAZCNAMEZMXZNAPTRZNSZPTRZSOAZSRVZTXT)INZCHAOSZHSZNONEr c @ s e Zd Zd eee eej eddddZ e
e j dddZ
e
jee ddd dZ
eejeedd
d d Zd!eeeejd
ddZeejejdddZeejdddZddddZeeeddddZeeddd d Zddd d Z dS )"r N) nameserversloopkwargsreturnc K sV |p
t | _|dd tjf d| ji|| _|r<|| _t | _
t | _ d | _ d S )NZ
sock_state_cb)
asyncioZget_event_loopr poppycaresChannel_sock_state_cb_channelr
set _read_fds
_write_fds_timer)selfr
r r r @/opt/alt/python38/lib/python3.8/site-packages/aiodns/__init__.py__init__/ s zDNSResolver.__init__)r c C s | j jS Nr Zserversr r r r r
< s zDNSResolver.nameservers)valuer c C s || j _d S r r )r r" r r r r
@ s )futresulterrornor c C s> | r d S |d k r0| t|tj| n
| | d S r )Z cancelledZ
set_exceptionr ZDNSErrorr errnostrerrorZ
set_result)r# r$ r% r r r _callbackD s
zDNSResolver._callback)hostqtypeqclassr c C s z t | }W n" tk
r. td|Y nX |d k rhz t| }W n" tk
rf td|Y nX tj| jd}t | j
|}| j j ||||d |S )Nzinvalid query type: {}zinvalid query class: {}r )Z query_class)
query_type_mapKeyError
ValueErrorformatquery_class_mapr Futurer functoolspartialr( r query)r r) r* r+ r# cbr r r r5 M s zDNSResolver.query)r) familyr c C s0 t j| jd}t| j|}| j||| |S Nr, )r r2 r r3 r4 r( r
gethostbyname)r r) r7 r# r6 r r r r9 ] s zDNSResolver.gethostbyname)namer c C s. t j| jd}t| j|}| j|| |S r8 )r r2 r r3 r4 r( r
gethostbyaddr)r r: r# r6 r r r r; c s zDNSResolver.gethostbyaddrc C s | j d S r )r cancelr! r r r r< i s zDNSResolver.cancel)fdreadablewritabler c C s |s|rn|r,| j || j|t | j| |rP| j || j|t | j| | j d kr| j
d| j | _ nj|| jkr| j | | j
| || jkr| j | | j | | js| js| j d k r| j d | _ d S Ng ?)r Z
add_reader
_handle_eventREADr addZ
add_writerWRITEr r
call_later _timer_cbdiscardZ
remove_readerZ
remove_writerr< )r r= r> r? r r r r l s$
zDNSResolver._sock_state_cb)r= eventr c C s8 t j}t j}|tkr|}n |tkr&|}| j|| d S r )r ARES_SOCKET_BADrB rD r
process_fd)r r= rH Zread_fdZwrite_fdr r r rA s zDNSResolver._handle_eventc C s< | j s | jr2| jtjtj | jd| j| _ nd | _ d S r@ )
r r r rJ r rI r rE rF r r! r r r rF s zDNSResolver._timer_cb)NN)N) __name__
__module__ __qualname__r r strr ZAbstractEventLoopr r propertyr r r
setter staticmethodr2 intr( r5 socketZ
AddressFamilyr9 r; r< boolr rA rF r r r r r . s(
r )#r r3 r rS typingr r r r r __version____all__rB rD Z QUERY_TYPE_AZQUERY_TYPE_AAAAZQUERY_TYPE_ANYZQUERY_TYPE_CAAZQUERY_TYPE_CNAMEZ
QUERY_TYPE_MXZQUERY_TYPE_NAPTRZ
QUERY_TYPE_NSZQUERY_TYPE_PTRZQUERY_TYPE_SOAZQUERY_TYPE_SRVZQUERY_TYPE_TXTr- ZQUERY_CLASS_INZQUERY_CLASS_CHAOSZQUERY_CLASS_HSZQUERY_CLASS_NONEZQUERY_CLASS_ANYr1 r r r r r <module> s: