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 )Z cancelledZ
set_exceptionr ZDNSErrorr �errno�strerrorZ
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 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)�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 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| j�tjtj� | j�d| 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: ��