File: //usr/lib/python3/dist-packages/twisted/protocols/__pycache__/dict.cpython-38.pyc
U
W[* � @ 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
� )�basic)�defer�protocol)�log)�BytesIOc C s� | dkr dS | dd� dkr"d}nd}d}t | �}|dkrD|�d� |�d�}|dkrv|dkr�|�d� ||�� fS nf|dkr�|�d�}|dkr�d| fS nB|dkr�|dkr�||�� fS d| fS n |d kr�|dkr�||�� fS ||7 }qDdS )
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 s4
r c C s t dd� | �S )z Munch a string into an 'atom'c S s | t ttd�dddg �kS )N�! �" �' �\ )�map�chr�range)�xr r r �<lambda>5 r zmakeAtom.<locals>.<lambda>)�filter�r r r r �makeAtom2 s r! c C sN t d�dddg }g }| D ]$}t|�|kr4|�d� |�|� qd�|�} | S )Nr r r r r r )r �ord�append�join)�sZ mustquote�result�cr r r �makeWord9 s
r( c C sH t | �dkr| dkrd S t | �dkr@| dd� dkr@| dd � } | S d S )Nr � .r r
s ..)�lenr r r r � parseTextE s
r+ c @ s e Zd ZdZdd� ZdS )�
DefinitionzA word definitionc C s || _ || _|| _|| _d S �N)�name�db�dbdesc�text)�selfr. r/ r0 r1 r r r �__init__Q s zDefinition.__init__N)�__name__�
__module__� __qualname__�__doc__r3 r r r r r, O 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& �r2 r r r r3 c s zDictClient.__init__c C s d| _ d| _d S )NZconn�command)�stater r: r r r �connectionMadeh s zDictClient.connectionMadec C s, t |�| jd krtd��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�r2 r r r r rA m s zDictClient.sendLinec C s� z|� d�}W n tk
r$ Y d S X | jdkr6d}nNt|�dk rht�d| � | �d� | j�� d S t |d d� �}|dd � }t
| d|| j f | j �}||� d S )Nzutf-8r1 � z-DictClient got invalid line from server -- %szInvalid line from server� zdictCode_%s_%s)
�decode� UnicodeErrorr r* r �msg�
protocolError� transportZLoseConnection�int�getattrr< �dictCode_default)r2 r �code�methodr r r � lineReceivedt s
zDictClient.lineReceivedc C s& t �d| � | �d� | j�� dS )zUnknown messagez3DictClient got unexpected message from server -- %szUnexpected server messageN)r rG rH rI �loseConnectionrB r r r rL � s
zDictClient.dictCode_defaultc C s dS )z*We are about to get kicked off, do nothingNr rB r r r �dictCode_221_ready� s z DictClient.dictCode_221_readyc C s d| _ | �� dS )zGreeting message�readyN)r< �
dictConnectedrB r r r �dictCode_220_conn� s z DictClient.dictCode_220_connc C s | � d� | j�� d S )Nz
Access denied�rH rI rP r: r r r �dictCode_530_conn� s
z DictClient.dictCode_530_connc C s | � d� | j�� d S )Nz Server temporarily unavailablerU r: r r r �dictCode_420_conn� s
z DictClient.dictCode_420_connc C s | � d� | j�� d S )Nz(Server shutting down at operator requestrU r: r r r �dictCode_421_conn� s
z DictClient.dictCode_421_connc C sR | j dkstd��d| _d| _d| _ dt|�d��t|�d��f }| �|� dS )zSend a dict DEFINE commandrR z4DictClient.sendDefine called when not in ready stateN�definez DEFINE %s %s�UTF-8)r< �AssertionErrorr& r9 r! �encoder( rA )r2 �database�wordr; r r r �
sendDefine� s zDictClient.sendDefinec C sR | j dkstd��d| _d| _d| _ dt|�t|�t|�f }| �|�d�� dS )zSend a dict MATCH commandrR z3DictClient.sendMatch called when not in ready stateN�matchzMATCH %s %s %srZ )r< r[ r& r9 r! rA r\ )r2 r] �strategyr^ r; r r r � sendMatch� s zDictClient.sendMatchc C s d| _ | �d � dS �zInvalid databaserR N�r � defineFailedrB r r r �dictCode_550_define� s z DictClient.dictCode_550_definec C s d| _ | �d � dS rc �r � matchFailedrB r r r �dictCode_550_match� s z DictClient.dictCode_550_matchc C s d| _ | �d � dS )zInvalid strategyrR Nrg rB r r r �dictCode_551_match� s z DictClient.dictCode_551_matchc C s d| _ | �d � dS �zNo matchrR Nrd rB r r r �dictCode_552_define� s z DictClient.dictCode_552_definec C s d| _ | �d � dS rk rg rB r r r �dictCode_552_match� s z DictClient.dictCode_552_matchc C s
g | _ dS )zn definitions retrievedN)r& rB r r r �dictCode_150_define� s z DictClient.dictCode_150_definec C sl d| _ t|�\}}t|�\}}t|�\}}|r6|r6|sL| �d� | j�� n | j�t|||g �� g | _dS )zDefinition text followsr1 zInvalid server responseN) r r rH rI rP r& r# r, r9 )r2 r r^ r/ r0 r r r �dictCode_151_define� s
z DictClient.dictCode_151_definec C s d| _ g | _g | _dS )z n matches found, text followsr1 N)r r& r9 rB r r r �dictCode_152_match� s z DictClient.dictCode_152_matchc C s< t |�}|dkr,d| _| j| jd _d| _n | j�|� dS )z"A line of definition text receivedNr; ���)r+ r r9 r& r1 r# )r2 r r r r r �dictCode_text_define� s z DictClient.dictCode_text_definec C sD dd� }t |�}|dkr4d| _t|| j�| _d| _n | j�|� dS )z One line of match text receivedc S s t | �\}}t |�\}}||fS r- )r )r% Zp1�tZp2r r r �l s z)DictClient.dictCode_text_match.<locals>.lNr; )r+ r r r9 r& r# )r2 r rt r r r r �dictCode_text_match� s z DictClient.dictCode_text_matchc C s | j }d| _ d| _| �|� dS ��okNrR )r& r< �
defineDone�r2 r rs r r r �dictCode_250_define
s z DictClient.dictCode_250_definec C s | j }d| _ d| _| �|� dS rv )r& r< � matchDonery r r r �dictCode_250_match s z DictClient.dictCode_250_matchc C s dS )z5override to catch unexpected dict protocol conditionsNr �r2 �reasonr r r rH s zDictClient.protocolErrorc C s dS )zCoverride to be notified when the server is ready to accept commandsNr r: r r r rS " s zDictClient.dictConnectedc C s dS )z8override to catch reasonable failure responses to DEFINENr r} r r r re ' s zDictClient.defineFailedc C s dS )z#override to catch successful DEFINENr �r2 r& r r r rx , s zDictClient.defineDonec C s dS )z6override to catch resonable failure responses to MATCHNr r} r r r rh 1 s zDictClient.matchFailedc C s dS )z"override to catch successful MATCHNr r r r r r{ 6 s zDictClient.matchDone)(r4 r5 r6 r7 r9 r> r< r r& �factoryr3 r= rA rO rL rQ rT rV rW rX r_ rb rf ri rj rl rm rn ro rp rr ru rz r| rH rS re rx rh r{ r r r r r8 Y sJ
r8 c @ s e Zd ZdS )�InvalidResponseN)r4 r5 r6 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 r- )r� �done�d�errbackr� �
clientDoner} r r r rH D s zDictLookup.protocolErrorc C s> | j jdkr t| j| j j� n | j jdkr:t| j| j j� d S )NrY r` )r� � queryTypeZapplyr_ �paramrb r: r r r rS J s zDictLookup.dictConnectedc C s& | j j�g � | j �� | j�� d S r- �r� r� �callbackr� rI rP r} r r r re Q s
zDictLookup.defineFailedc C s& | j j�|� | j �� | j�� d S r- r� r r r r rx W s
zDictLookup.defineDonec C s& | j j�g � | j �� | j�� d S r- r� r} r r r rh ] s
zDictLookup.matchFailedc C s& | j j�|� | j �� | j�� d S r- r� r r r r r{ c s
zDictLookup.matchDoneN)
r4 r5 r6 r7 rH rS re rx rh r{ r r r r r� A 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� )r2 r� r� r� r r r r3 o s zDictLookupFactory.__init__c C s d| _ | `dS )zCalled by client when done.r N)r� r� r: r r r r� v s z DictLookupFactory.clientDonec C s | j �|� d S r- )r� r� �r2 Z connector�errorr r r �clientConnectionFailed| s z(DictLookupFactory.clientConnectionFailedc C s | j s| j�|� d S r- )r� r� r� r� r r r �clientConnectionLost� s z&DictLookupFactory.clientConnectionLostc C s | � � }| |_|S r- )r r� )r2 Zaddr�pr r r �
buildProtocol� s z DictLookupFactory.buildProtocol) r4 r5 r6 r7 r� r r� r3 r� r� r� r� r r r r r� j s r� c C s6 t �� }td||f|�}ddlm} |�| ||� |S )z"Look up a word using a dict serverrY r ��reactor�r ZDeferredr� �twisted.internetr� Z
connectTCP)�host�portr] r^ r� r� r� r r r rY � s
rY c C s8 t �� }td|||f|�}ddlm} |�| ||� |S )z Match a word using a dict serverr` r r� r� )r� r� r] ra r^ r� r� r� r r r r` � s
r` N)r7 Ztwisted.protocolsr r� r r Ztwisted.pythonr r r r r! r( r+ r, r@ r8 � Exceptionr� r� Z
ClientFactoryr� rY r` r r r r �<module> s "
d)"