Your IP : 216.73.216.61
Upz\� � @ s� d d l Z d d l Z d d l Z d d l m Z m Z m Z d d l m Z d Z d Z
d Z d Z d e j
d
e j d e j d e j d
e j d e j d e j d e j d e j d e j d e j i Z Gd d � d � Z d S)� N)�Any�List�Optional� )�errorz2.0.0� DNSResolverr � �AZAAAA�ANYZCNAMEZMXZNAPTRZNSZPTRZSOAZSRVZTXTc @ s� e Z d Z d d d d � Z e d d � � Z e j d d � � Z e d d � � Z d d
� Z d d � Z
d
d � Z d d � Z d d � Z
d d � Z d d � Z d S)r Nc K s� | p t j � | _ | j d k s* t � | j d d � t j d | j | � | _ | rd | | _ t
� | _ t
� | _ d | _
d S)NZ
sock_state_cb)�asyncioZget_event_loop�loop�AssertionError�pop�pycaresZChannel�_sock_state_cb�_channel� nameservers�set� _read_fds�
_write_fds�_timer)�selfr r �kwargs� r �@/opt/alt/python35/lib/python3.5/site-packages/aiodns/__init__.py�__init__' s zDNSResolver.__init__c C s
| j j S)N)r �servers)r r r r r 3 s zDNSResolver.nameserversc C s | | j _ d S)N)r r )r �valuer r r r 8 s c C sU | j � r d S| d k rD | j t j | t j j | � � � n
| j | � d S)N)Z cancelledZ
set_exceptionr ZDNSErrorr �errno�strerrorZ
set_result)�fut�resultZerrornor r r � _callback= s
(zDNSResolver._callbackc C s| y t | } Wn' t k
r7 t d j | � � � Yn Xt j d | j � } t j | j | � } | j
j | | | � | S)Nzinvalid query type: {}r ) �query_type_map�KeyError�
ValueError�formatr �Futurer � functools�partialr" r �query)r �hostZqtyper �cbr r r r* G s
zDNSResolver.queryc C sD t j d | j � } t j | j | � } | j j | | | � | S)Nr )r r' r r( r) r" r �
gethostbyname)r r+ �familyr r, r r r r- R s zDNSResolver.gethostbynamec C sA t j d | j � } t j | j | � } | j j | | � | S)Nr )r r' r r( r) r" r �
gethostbyaddr)r �namer r, r r r r/ Y s zDNSResolver.gethostbyaddrc C s | j j � d S)N)r �cancel)r r r r r1 ` s zDNSResolver.cancelc C s8 | s | r� | r> | j j | | j | t � | j j | � | rp | j j | | j | t � | j j | � | j d k r4| j j
d | j � | _ n� | | j k r� | j j | � | j j
| � | | j k r� | j j | � | j j | � | j r4| j r4| j d k r4| j j � d | _ d S)Ng �?)r Z
add_reader�
_handle_event�READr �addZ
add_writer�WRITEr r �
call_later� _timer_cb�discardZ
remove_readerZ
remove_writerr1 )r �fd�readable�writabler r r r d s$ #
zDNSResolver._sock_state_cbc C sP t j } t j } | t k r' | } n | t k r9 | } | j j | | � d S)N)r �ARES_SOCKET_BADr3 r5 r �
process_fd)r r9 �eventZread_fdZwrite_fdr r r r2 } s zDNSResolver._handle_eventc C sV | j s | j rI | j j t j t j � | j j d | j � | _ n d | _ d S)Ng �?)
r r r r= r r< r r6 r7 r )r r r r r7 � s zDNSResolver._timer_cb)�__name__�
__module__� __qualname__r �propertyr �setter� staticmethodr" r* r- r/ r1 r r2 r7 r r r r r & s
)r r )r r( r �typingr r r � r � __version__�__all__r3 r5 Z QUERY_TYPE_AZQUERY_TYPE_AAAAZQUERY_TYPE_ANYZQUERY_TYPE_CNAMEZ
QUERY_TYPE_MXZQUERY_TYPE_NAPTRZ
QUERY_TYPE_NSZQUERY_TYPE_PTRZQUERY_TYPE_SOAZQUERY_TYPE_SRVZQUERY_TYPE_TXTr# r r r r r �<module> s(