Your IP : 216.73.216.213
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
� )�basic)�defer�protocol)�log)�BytesIOc C s� | dkr dS | dd� dkr"d}nd}d}t | �}|dkrD|�d� x�|�d�}|dkrx|dkr�|�d� ||�� fS nf|dkr�|�d�}|dkr�d| fS nB|dkr�|dkr�||�� fS d| fS n |d kr�|dkr�||�� fS ||7 }qFW dS )
zQChew one dqstring or atom from beginning of line and return (param, remaningline)� )Nr r � � "� � \N� )r �read)�line�mode�res�io�a� r �I/opt/alt/python37/lib64/python3.7/site-packages/twisted/protocols/dict.py�
parseParam s6
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 sR t d�dddg }g }x,| D ]$}t|�|kr6|�d� |�|� q W d�|�} | 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 ..)�len)r 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- r. r/ r r r �__init__Q s zDefinition.__init__N)�__name__�
__module__� __qualname__�__doc__r1 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 )N)�datar% )r0 r r r r1 c s zDictClient.__init__c C s d| _ d| _d S )NZconn�command)�stater )r0 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)r0 r r r r r> m s zDictClient.sendLinec C s� y|� d�}W n tk
r" d S X | jdkr4d}nNt|�dk rft�d| � | �d� | j�� d S t |d d� �}|dd � }t
| d|| j f | j �}||� d S )Nzutf-8r/ � z-DictClient got invalid line from server -- %szInvalid line from server� zdictCode_%s_%s)
�decode� UnicodeErrorr r) r �msg�
protocolError� transportZLoseConnection�int�getattrr9 �dictCode_default)r0 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 rC rD rE �loseConnection)r0 r r r r rH � s
zDictClient.dictCode_defaultc C s dS )z*We are about to get kicked off, do nothingNr )r0 r r r r �dictCode_221_ready� s z DictClient.dictCode_221_readyc C s d| _ | �� dS )zGreeting message�readyN)r9 �
dictConnected)r0 r r r r �dictCode_220_conn� s z DictClient.dictCode_220_connc C s | � d� | j�� d S )Nz
Access denied)rD rE rL )r0 r r r �dictCode_530_conn� s
z DictClient.dictCode_530_connc C s | � d� | j�� d S )Nz Server temporarily unavailable)rD rE rL )r0 r r r �dictCode_420_conn� s
z DictClient.dictCode_420_connc C s | � d� | j�� d S )Nz(Server shutting down at operator request)rD rE rL )r0 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 commandrN z4DictClient.sendDefine called when not in ready stateN�definez DEFINE %s %szUTF-8)r9 �AssertionErrorr% r7 r �encoder' r> )r0 �database�wordr8 r r r �
sendDefine� s zDictClient.sendDefinec C sR | j dkstd��d| _d| _d| _ dt|�t|�t|�f }| �|�d�� dS )zSend a dict MATCH commandrN z3DictClient.sendMatch called when not in ready stateN�matchzMATCH %s %s %szUTF-8)r9 rU r% r7 r r> rV )r0 rW �strategyrX r8 r r r � sendMatch� s zDictClient.sendMatchc C s d| _ | �d � dS )zInvalid databaserN N)r � defineFailed)r0 r r r r �dictCode_550_define� s z DictClient.dictCode_550_definec C s d| _ | �d � dS )zInvalid databaserN N)r � matchFailed)r0 r r r r �dictCode_550_match� s z DictClient.dictCode_550_matchc C s d| _ | �d � dS )zInvalid strategyrN N)r r_ )r0 r r r r �dictCode_551_match� s z DictClient.dictCode_551_matchc C s d| _ | �d � dS )zNo matchrN N)r r] )r0 r r r r �dictCode_552_define� s z DictClient.dictCode_552_definec C s d| _ | �d � dS )zNo matchrN N)r r_ )r0 r r r r �dictCode_552_match� s z DictClient.dictCode_552_matchc C s
g | _ dS )zn definitions retrievedN)r% )r0 r 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 followsr/ zInvalid server responseN) r r rD rE rL r% r" r+ r7 )r0 r rX r- r. r r r �dictCode_151_define� s
z DictClient.dictCode_151_definec C s d| _ g | _g | _dS )z n matches found, text followsr/ N)r r% r7 )r0 r 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 receivedNr8 ���)r* r r7 r% r/ r" )r0 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 )N)r )r$ �p1�t�p2r r r �l s z)DictClient.dictCode_text_match.<locals>.lNr8 )r* r r r7 r% r" )r0 r rl r r r r �dictCode_text_match� s z DictClient.dictCode_text_matchc C s | j }d| _ d| _| �|� dS )�okNrN )r% r9 �
defineDone)r0 r rj r r r �dictCode_250_define
s z DictClient.dictCode_250_definec C s | j }d| _ d| _| �|� dS )rn NrN )r% r9 � matchDone)r0 r rj r r r �dictCode_250_match s z DictClient.dictCode_250_matchc C s dS )z5override to catch unexpected dict protocol conditionsNr )r0 �reasonr r r rD s zDictClient.protocolErrorc C s dS )zCoverride to be notified when the server is ready to accept commandsNr )r0 r r r rO " s zDictClient.dictConnectedc C s dS )z8override to catch reasonable failure responses to DEFINENr )r0 rs r r r r] ' s zDictClient.defineFailedc C s dS )z#override to catch successful DEFINENr )r0 r% r r r ro , s zDictClient.defineDonec C s dS )z6override to catch resonable failure responses to MATCHNr )r0 rs r r r r_ 1 s zDictClient.matchFailedc C s dS )z"override to catch successful MATCHNr )r0 r% r r r rq 6 s zDictClient.matchDone)(r2 r3 r4 r5 r7 r; r9 r r% �factoryr1 r: r> rK rH rM rP rQ rR rS rY r\ r^ r` ra rb rc rd re rf rh rm rp rr rD rO r] ro r_ rq r r r r r6 Y sJ
r6 c @ s e Zd ZdS )�InvalidResponseN)r2 r3 r4 r r r r ru < s ru 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 )N)rt �done�d�errbackru �
clientDone)r0 rs r r r rD D s zDictLookup.protocolErrorc C s: | j jdkr | j| j j� n| j jdkr6| j| j j� d S )NrT rZ )rt � queryTyperY �paramr\ )r0 r r r rO J s zDictLookup.dictConnectedc C s& | j j�g � | j �� | j�� d S )N)rt rx �callbackrz rE rL )r0 rs r r r r] Q s
zDictLookup.defineFailedc C s& | j j�|� | j �� | j�� d S )N)rt rx r} rz rE rL )r0 r% r r r ro W s
zDictLookup.defineDonec C s& | j j�g � | j �� | j�� d S )N)rt rx r} rz rE rL )r0 rs r r r r_ ] s
zDictLookup.matchFailedc C s& | j j�|� | j �� | j�� d S )N)rt rx r} rz rE rL )r0 r% r r r rq c s
zDictLookup.matchDoneN)
r2 r3 r4 r5 rD rO r] ro r_ rq r r r r rv A s rv 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| rx rw )r0 r{ r| rx r r r r1 o s zDictLookupFactory.__init__c C s d| _ | `dS )zCalled by client when done.r N)rw rx )r0 r r r rz v s z DictLookupFactory.clientDonec C s | j �|� d S )N)rx ry )r0 � connector�errorr r r �clientConnectionFailed| s z(DictLookupFactory.clientConnectionFailedc C s | j s| j�|� d S )N)rw rx ry )r0 r r� r r r �clientConnectionLost� s z&DictLookupFactory.clientConnectionLostc C s | � � }| |_|S )N)r rt )r0 �addr�pr r r �
buildProtocol� s z DictLookupFactory.buildProtocol) r2 r3 r4 r5 rv r rw r1 rz 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 serverrT r )�reactor)r �Deferredr~ �twisted.internetr� �
connectTCP)�host�portrW rX rx rt r� r r r rT � s
rT c C s8 t �� }td|||f|�}ddlm} |�| ||� |S )z Match a word using a dict serverrZ r )r� )r r� r~ r� r� r� )r� r� rW r[ rX rx rt r� r r r rZ � s
rZ N)r5 Ztwisted.protocolsr r� r r Ztwisted.pythonr r r r r r' r* r+ r= r6 � Exceptionru rv Z
ClientFactoryr~ rT rZ r r r r �<module> s "
d)"