File: //lib/python3/dist-packages/twisted/protocols/__pycache__/dict.cpython-310.pyc
o
�b�* � @ s� d Z ddlmZ ddlmZmZ ddlmZ ddlm Z dd� Z
dd � Z d
d � Z d d
� Z
G dd� d�ZG dd� dej�ZG dd� de�ZG dd� de�ZG dd� dej�Zdd� Zdd� Zd S ) zE
Dict client protocol implementation.
@author: Pavel Pergamenshchik
� )�BytesIO)�defer�protocol)�basic)�logc C s� | dkrdS | dd� dkrd}nd}d}t | �}|dkr"|�d� |�d�}|dkr<|dkr;|�d� ||�� fS n2|dkrN|�d�}|dkrMd| fS n |dkr`|dkr\||�� fS d| fS |d krn|dkrn||�� fS ||7 }q#)
zQChew one dqstring or atom from beginning of line and return (param, remaningline)� )Nr r � � "� � \N� )r �read)�line�mode�res�io�a� r �8/usr/lib/python3/dist-packages/twisted/protocols/dict.py�
parseParam s<
�
� �r c C s t dd� | �S )z Munch a string into an 'atom'c S s | t ttd�g d� �vS )N�! ��" �' �\ )�map�chr�range)�xr r r �<lambda>6 s zmakeAtom.<locals>.<lambda>)�filter�r r r r �makeAtom3 s r" c C sL t d�g d� }g }| D ]}t|�|v r|�d� |�|� q d�|�} | S )Nr r r r )r �ord�append�join)�s� mustquote�result�cr r r �makeWord9 s
r* c C sD t | �dkr | dkr d S t | �dkr | dd� dkr | dd � } | S )Nr � .r r
s ..)�lenr! r r r � parseTextD s
r- c @ s e Zd ZdZdd� ZdS )�
DefinitionzA word definitionc C s || _ || _|| _|| _d S �N)�name�db�dbdesc�text)�selfr0 r1 r2 r3 r r r �__init__P �
zDefinition.__init__N)�__name__�
__module__� __qualname__�__doc__r5 r r r r r. M s r. c @ s e Zd ZdZdZdZdZdZdZdZ dd� Z
dd� Z dd � Z d
d � Z
d d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Z d*d+� Z d,d-� Z d.d/� Z d0d1� Z d2d3� Z!d4d5� Z"d6d7� Z#d8d9� Z$d:d;� Z%d<d=� Z&d>d?� Z'dS )@�
DictClientzdict (RFC2229) clientNi c C s d | _ d | _d S r/ )�datar( �r4 r r r r5 a �
zDictClient.__init__c C s d| _ d| _d S )N�conn�command)�stater r= r r r �connectionMadee r> zDictClient.connectionMadec C s, t |�| jd kr
td��tj�| |� dS )z3Throw up if the line is longer than 1022 charactersr
z(DictClient tried to send a too long lineN)r, �
MAX_LENGTH�
ValueErrorr � LineReceiver�sendLine�r4 r r r r rF i s zDictClient.sendLinec C s� z|� d�}W n
ty Y d S w | jdkrd}n't|�dk r3t�d| � | �d� | j�� d S t |d d� �}|dd � }t
| d|� d| j � �| j �}||� d S ) Nzutf-8r3 � z-DictClient got invalid line from server -- %szInvalid line from server� � dictCode_�_)
�decode� UnicodeErrorr r, r �msg�
protocolError� transport�LoseConnection�int�getattrrA �dictCode_default)r4 r �code�methodr r r � lineReceivedo s �
zDictClient.lineReceivedc C s& t �d| � | �d� | j�� dS )zUnknown messagez3DictClient got unexpected message from server -- %szUnexpected server messageN)r rN rO rP �loseConnectionrG r r r rT � s
zDictClient.dictCode_defaultc C � dS )z*We are about to get kicked off, do nothingNr rG r r r �dictCode_221_ready� � z DictClient.dictCode_221_readyc C s d| _ | �� dS )zGreeting message�readyN)rA �
dictConnectedrG r r r �dictCode_220_conn� s z DictClient.dictCode_220_connc C � | � d� | j�� d S )Nz
Access denied�rO rP rX r= r r r �dictCode_530_conn� �
z DictClient.dictCode_530_connc C r_ )Nz Server temporarily unavailabler` r= r r r �dictCode_420_conn� rb z DictClient.dictCode_420_connc C r_ )Nz(Server shutting down at operator requestr` r= r r r �dictCode_421_conn� rb z DictClient.dictCode_421_connc C sR | j dks J d��d| _d| _d| _ d�t|�d��t|�d���}| �|� dS )zSend a dict DEFINE commandr\ z4DictClient.sendDefine called when not in ready stateN�definez DEFINE {} {}�UTF-8)rA r( r< �formatr" �encoder* rF )r4 �database�wordr@ r r r �
sendDefine� s �� �zDictClient.sendDefinec C sR | j dks J d��d| _d| _d| _ d�t|�t|�t|��}| �|�d�� dS )zSend a dict MATCH commandr\ z3DictClient.sendMatch called when not in ready stateN�matchzMATCH {} {} {}rf )rA r( r< rg r" rF rh )r4 ri �strategyrj r@ r r r � sendMatch� s ��zDictClient.sendMatchc C � d| _ | �d � dS �zInvalid databaser\ N�r � defineFailedrG r r r �dictCode_550_define� � z DictClient.dictCode_550_definec C ro rp �r � matchFailedrG r r r �dictCode_550_match� rt z DictClient.dictCode_550_matchc C ro )zInvalid strategyr\ Nru rG r r r �dictCode_551_match� rt z DictClient.dictCode_551_matchc C ro �zNo matchr\ Nrq rG r r r �dictCode_552_define� rt z DictClient.dictCode_552_definec C ro ry ru rG r r r �dictCode_552_match� rt z DictClient.dictCode_552_matchc C s
g | _ dS )zn definitions retrievedN)r( rG r r r �dictCode_150_define� s
z DictClient.dictCode_150_definec C sn d| _ t|�\}}t|�\}}t|�\}}|r|r|s'| �d� | j�� dS | j�t|||g �� g | _dS )zDefinition text followsr3 zInvalid server responseN) r r rO rP rX r( r$ r. r< )r4 r rj r1 r2 r r r �dictCode_151_define� s
z DictClient.dictCode_151_definec C s d| _ g | _g | _dS )z n matches found, text followsr3 N)r r( r< rG r r r �dictCode_152_match� s
z DictClient.dictCode_152_matchc C s> t |�}|dkrd| _| j| jd _d| _dS | j�|� dS )z"A line of definition text receivedNr@ ���)r- r r< r( r3 r$ )r4 r r r r r �dictCode_text_define� s
z DictClient.dictCode_text_definec C sF dd� }t |�}|dkrd| _t|| j�| _d| _dS | j�|� dS )z One line of match text receivedc S s t | �\}}t |�\}}||fS r/ )r )r& �p1�t�p2r r r �l� s z)DictClient.dictCode_text_match.<locals>.lNr@ )r- r r r< r( r$ )r4 r r� r r r r �dictCode_text_match� s
z DictClient.dictCode_text_matchc C � | j }d| _ d| _| �|� dS ��okNr\ )r( rA �
defineDone�r4 r r� r r r �dictCode_250_define � z DictClient.dictCode_250_definec C r� r� )r( rA � matchDoner� r r r �dictCode_250_match r� z DictClient.dictCode_250_matchc C rY )z5override to catch unexpected dict protocol conditionsNr �r4 �reasonr r r rO r[ zDictClient.protocolErrorc C rY )zCoverride to be notified when the server is ready to accept commandsNr r= r r r r] r[ zDictClient.dictConnectedc C rY )z8override to catch reasonable failure responses to DEFINENr r� r r r rr r[ zDictClient.defineFailedc C rY )z#override to catch successful DEFINENr �r4 r( r r r r� r[ zDictClient.defineDonec C rY )z7override to catch reasonable failure responses to MATCHNr r� r r r rv # r[ zDictClient.matchFailedc C rY )z"override to catch successful MATCHNr r� r r r r� ' r[ zDictClient.matchDone)(r7 r8 r9 r: r< rC rA r r( �factoryr5 rB rF rW rT rZ r^ ra rc rd rk rn rs rw rx rz r{ r| r} r~ r� r� r� r� rO r] rr r� rv r� r r r r r; W sL
r; c @ s e Zd ZdS )�InvalidResponseN)r7 r8 r9 r r r r r� , s r� c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�
DictLookupzNUtility class for a single dict transaction. To be used with DictLookupFactoryc C s, | j js| j j�t|�� | j �� d S d S r/ )r� �done�d�errbackr� �
clientDoner� r r r rO 3 s �zDictLookup.protocolErrorc C s@ | j jdkr| j| j j� d S | j jdkr | j| j j� d S d S )Nre rl )r� � queryTyperk �paramrn r= r r r r] 8 s
�zDictLookup.dictConnectedc C �& | j j�g � | j �� | j�� d S r/ �r� r� �callbackr� rP rX r� r r r rr > �
zDictLookup.defineFailedc C �& | j j�|� | j �� | j�� d S r/ r� r� r r r r� C r� zDictLookup.defineDonec C r� r/ r� r� r r r rv H r� zDictLookup.matchFailedc C r� r/ r� r� r r r r� M r� zDictLookup.matchDoneN)
r7 r8 r9 r: rO r] rr r� rv r� r r r r r� 0 s r� c @ s@ e Zd ZdZeZdZdd� Zdd� Zdd� Z d d
� Z
d d � Z dS )
�DictLookupFactoryz-Utility factory for a single dict transactionNc C s || _ || _|| _d| _d S )Nr )r� r� r� r� )r4 r� r� r� r r r r5 Y r6 zDictLookupFactory.__init__c C s d| _ | `dS )zCalled by client when done.r N)r� r� r= r r r r� _ s z DictLookupFactory.clientDonec C s | j �|� d S r/ )r� r� �r4 � connector�errorr r r �clientConnectionFailedd s z(DictLookupFactory.clientConnectionFailedc C s | j s | j�|� d S d S r/ )r� r� r� r� r r r �clientConnectionLostg s �z&DictLookupFactory.clientConnectionLostc C s | � � }| |_|S r/ )r r� )r4 �addr�pr r r �
buildProtocolk s z DictLookupFactory.buildProtocol) r7 r8 r9 r: r� r r� r5 r� r� r� r� r r r r r� S s r� c C s6 t �� }td||f|�}ddlm} |�| ||� |S )z"Look up a word using a dict serverre r ��reactor�r �Deferredr� �twisted.internetr� �
connectTCP)�host�portri rj r� r� r� r r r re q s
re c C s8 t �� }td|||f|�}ddlm} |�| ||� |S )z Match a word using a dict serverrl r r� r� )r� r� ri rm rj r� r� r� r r r rl | s
rl N)r: r r r� r r �twisted.protocolsr �twisted.pythonr r r" r* r- r. rE r; � Exceptionr� r� �
ClientFactoryr� re rl r r r r �<module> s" !
V#