File: //opt/alt/python38/lib/python3.8/site-packages/aiodns/__pycache__/__init__.cpython-38.pyc
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)�Any�List�Optional�Set� )�errorz3.0.0)� DNSResolverr � ) �AZAAAA�ANYZCAAZCNAMEZMXZNAPTRZNSZPTRZSOAZSRVZTXT)�INZCHAOSZHSZNONEr c @ s� e Zd Zd eee eej edd�dd�Z e
e j d�dd��Z
e
jee dd�d d��Z
eejeedd
�d d ��Zd!eeeejd
�dd�Zeejejd�dd�Zeejd�dd�Zdd�dd�Zeeedd�dd�Zeedd�d d �Zdd�d d �Z dS )"r N)� nameservers�loop�kwargs�returnc 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 �AssertionError�pop�pycares�Channel�_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 �N�r Zservers�r r r r r
< s zDNSResolver.nameservers)�valuer c C s || j _d S r r! )r r# r r r r
@ s )�fut�result�errornor c C s> | � � r d S |d k r0| �t�|tj�|��� n
| �|� d S r )� cancelled�
set_exceptionr ZDNSErrorr �errno�strerror�
set_result)r$ r% r& r r r � _callbackD s
zDNSResolver._callback)�host�qtype�qclassr 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_map�KeyError�
ValueError�format�query_class_mapr �Futurer � functools�partialr, r �query)r r- r. r/ r$ �cbr r r r9 M s zDNSResolver.query)r- �familyr c C s0 t j| jd�}t�| j|�}| j�|||� |S �Nr0 )r r6 r r7 r8 r, r �
gethostbyname)r r- r; r$ r: r r r r= ] s zDNSResolver.gethostbyname)�namer c C s. t j| jd�}t�| j|�}| j�||� |S r< )r r6 r r7 r8 r, r �
gethostbyaddr)r r> r$ r: r r r r? c s zDNSResolver.gethostbyaddrc C s | j �� d S r )r �cancelr" r r r r@ i s zDNSResolver.cancel)�fd�readable�writabler 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_event�READr �addZ
add_writer�WRITEr r �
call_later� _timer_cb�discardZ
remove_readerZ
remove_writerr@ )r rA rB rC r r r r l s$
zDNSResolver._sock_state_cb)rA �eventr c C s8 t j}t j}|tkr|}n |tkr&|}| j�||� d S r )r �ARES_SOCKET_BADrF rH r �
process_fd)r rA rL Zread_fdZwrite_fdr r r rE � s zDNSResolver._handle_eventc C s< | j s | jr2| j�tjtj� | j�d| j�| _ nd | _ d S rD )
r r r rN r rM r rI rJ r r" r r r rJ � s zDNSResolver._timer_cb)NN)N) �__name__�
__module__� __qualname__r r �strr ZAbstractEventLoopr r �propertyr r r
�setter� staticmethodr6 �intr, r9 �socket�
AddressFamilyr= r? r@ �boolr rE rJ r r r r r . s( � �
r )#r r7 r rW �typingr r r r � r � __version__�__all__rF rH 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_TXTr1 ZQUERY_CLASS_INZQUERY_CLASS_CHAOSZQUERY_CLASS_HSZQUERY_CLASS_NONEZQUERY_CLASS_ANYr5 r r r r r �<module> s: ��