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 sd |p
t | _| jd k s t|dd tjf d| ji|| _|rJ|| _ t
| _ t
| _ d | _
d S )NZ
sock_state_cb)asyncioZget_event_loopr AssertionErrorpoppycaresChannel_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 r6 M s zDNSResolver.query)r* familyr c C s0 t j| jd}t| j|}| j||| |S Nr- )r r3 r r4 r5 r) r
gethostbyname)r r* r8 r$ r7 r r r r: ] s zDNSResolver.gethostbyname)namer c C s. t j| jd}t| j|}| j|| |S r9 )r r3 r r4 r5 r) r
gethostbyaddr)r r; r$ r7 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_BADrC rE r
process_fd)r r> rI Zread_fdZwrite_fdr r r rB s zDNSResolver._handle_eventc C s< | j s | jr2| jtjtj | jd| j| _ nd | _ d S rA )
r r r rK r rJ r rF rG r r" r r r rG s zDNSResolver._timer_cb)NN)N) __name__
__module__ __qualname__r r strr ZAbstractEventLoopr r propertyr r r
setter staticmethodr3 intr) r6 socketZ
AddressFamilyr: r< r= boolr rB rG r r r r r . s(
r )#r r4 r rT typingr r r r r __version____all__rC rE 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_ANYr2 r r r r r <module> s: