Current File : //usr/lib/python3.6/site-packages/pyzor/__pycache__/message.cpython-36.pyc
3
�2�^� � @ s d Z ddlZddlZddlZG dd� dejj�ZG dd� de�ZG dd� de�ZG d d
� d
e�Z G d d � d e �Z
G d
d� de
�Z G dd� de �Z G dd� de
�Z
G dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd � d e �ZG d d � d e�ZdS ) zZThis modules contains the various messages used in the pyzor client server
communication.
� Nc @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �Messagec C s t jjj| � | j� d S )N)�email�messager �__init__�setup)�self� r � /usr/lib/python3.6/message.pyr s zMessage.__init__c C s d S )Nr )r r r r r s z
Message.setupc C s | j � d S )N)�ensure_complete)r r r r �init_for_sending s zMessage.init_for_sendingc C s | j � S )N)Z as_string)r r r r �__str__ s zMessage.__str__c C s d S )Nr )r r r r r
s zMessage.ensure_completeN)�__name__�
__module__� __qualname__r r r r r
r r r r r s
r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �ThreadedMessagec C s> d| kr| j tj� � d| ks"t�ttj�| d< tj| � d S )N�Thread�PV) �
set_thread�ThreadId�generate�AssertionError�str�pyzorZ
proto_versionr r )r r r r r s
z ThreadedMessage.init_for_sendingc C s( d| ksd| krt jd��tj| � d S )Nr r z*Doesn't have fields for a ThreadedMessage.)r �IncompleteMessageErrorr r
)r r r r r
& s
z ThreadedMessage.ensure_completec C s t | d �S )Nr )�float)r r r r �get_protocol_version, s z$ThreadedMessage.get_protocol_versionc C s t | d �S )Nr )r )r r r r �
get_thread/ s zThreadedMessage.get_threadc C s t |�| d< d S )Nr )r )r �ir r r r 2 s zThreadedMessage.set_threadN)r
r r r r
r r r r r r r r s
r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�Response�� c C s( d| ksd| krt jd��tj| � d S )N�Code�Diagz"doesn't have fields for a Response)r r r r
)r r r r r
9 s
zResponse.ensure_completec C s | j � | jkS )N)�get_code�ok_code)r r r r �is_ok? s zResponse.is_okc C s t | d �S )Nr )�int)r r r r r" B s zResponse.get_codec C s | d S )Nr! r )r r r r �get_diagE s zResponse.get_diagc C s | j � | j� fS )N)r" r& )r r r r �
head_tupleH s zResponse.head_tupleN) r
r r r# r
r$ r" r& r' r r r r r 6 s r c @ s e Zd ZdZdd� Zdd� ZdS )�Requestz�This is the class that should be used to read in Requests of any type.
Subclasses are responsible for setting 'Op' if they are generating a
message,c C s | d S )N�Opr )r r r r �get_opQ s zRequest.get_opc C s d| krt jd��tj| � d S )Nr) z!doesn't have fields for a Request)r r r r
)r r r r r
T s
zRequest.ensure_completeN)r
r r �__doc__r* r
r r r r r( L s r( c @ s e Zd ZdZdd� ZdS )�ClientSideRequestNc C s t j| � | j| d< d S )Nr) )r( r �op)r r r r r ^ s
zClientSideRequest.setup)r
r r r- r r r r r r, [ s r, c @ s e Zd Zddd�Zdd� ZdS )�SimpleDigestBasedRequestNc C s" t j| � d| _|r | j|� d S )Nr )r, r � digest_count�
add_digest)r �digestr r r r d s
z!SimpleDigestBasedRequest.__init__c C s | j d|� | jd7 _d S )Nz Op-Digest� )Z
add_headerr/ )r r1 r r r r0 j s z#SimpleDigestBasedRequest.add_digest)N)r
r r r r0 r r r r r. c s
r. c @ s e Zd Zddd�ZdS )� SimpleDigestSpecBasedRequestNc C s: t j| |� |r6dd� |D �}djdd� |D ��| d< d S )Nc S s g | ]}|D ]}|�q qS r r )�.0Zsublist�itemr r r �
<listcomp>s s z9SimpleDigestSpecBasedRequest.__init__.<locals>.<listcomp>�,c s s | ]}t |�V qd S )N)r )r4 �partr r r � <genexpr>t s z8SimpleDigestSpecBasedRequest.__init__.<locals>.<genexpr>zOp-Spec)r. r �join)r r1 �specZ flat_specr r r r p s z%SimpleDigestSpecBasedRequest.__init__)NN)r
r r r r r r r r3 o s r3 c @ s e Zd ZdZdS )� PingRequestZpingN)r
r r r- r r r r r< w s r< c @ s e Zd ZdZdS )� PongRequestZpongN)r
r r r- r r r r r= { s r= c @ s e Zd ZdZdS )� CheckRequestZcheckN)r
r r r- r r r r r> s r> c @ s e Zd ZdZdS )� InfoRequest�infoN)r
r r r- r r r r r? � s r? c @ s e Zd ZdZdS )�
ReportRequestZreportN)r
r r r- r r r r rA � s rA c @ s e Zd ZdZdS )�WhitelistRequestZ whitelistN)r
r r r- r r r r rB � s rB c @ s< e Zd ZdZded fZdZdd� Zedd � �Zd
d � Z d S )r r � � i r2 c C s< t j| |�}| jd | ko*| jd k n s8td��|S )Nr r2 zvalue outside of range)r% �__new__�
full_range�
ValueError)�clsr r r r r rE � s $zThreadId.__new__c C s | t j| j� �S )N)�randomZ randrange�ok_range)rH r r r r � s zThreadId.generatec C s$ | j d | ko | j d k S S )Nr r2 )rJ )r r r r � in_ok_range� s zThreadId.in_ok_rangeN� )r rL )
r
r r rF rJ Z error_valuerE � classmethodr rK r r r r r � s r )r+ rI Z
email.messager r r r r r r( r, r. r3 r< r= r> r? rA rB r% r r r r r �<module> s"
Mini Shell