Your IP : 216.73.216.213
B
;��]� � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddl
m Z G dd � d �Z G d
d � d �Z
G d d
� d
�ZG dd� d�ZG dd� de�Ze e j�G dd� de��ZG dd� dej�Zdd� ZdS )z
Utility classes for spread.
� )�defer)�Failure)�pb)�basic)�
interfaces)� implementerc @ s e Zd Zdd� Zdd� ZdS )� LocalMethodc C s || _ || _d S )N)�local�name)�selfr r
� r �F/opt/alt/python37/lib64/python3.7/site-packages/twisted/spread/util.py�__init__ s zLocalMethod.__init__c O s | j j| jf|�|�S )N)r �
callRemoter
)r �args�kwr r r
�__call__ s zLocalMethod.__call__N)�__name__�
__module__� __qualname__r r r r r r
r s r c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�
LocalAsRemotezN
A class useful for emulating the effects of remote behavior locally.
� c O sj t | d| �r"t| d| �||�S y t| d| �}t�|||��S t� }| jr\|�� t�|�S dS )ax
Call a specially-designated local method.
self.callRemote('x') will first try to invoke a method named
sync_x and return its result (which should probably be a
Deferred). Second, it will look for a method called async_x,
which will be called and then have its result (or Failure)
automatically wrapped in a Deferred.
Zsync_Zasync_N)�hasattr�getattrr �succeedr �reportAllTracebacksZprintTraceback�fail)r r
r r �method�fr r r
r # s
zLocalAsRemote.callRemotec C s
t | |�S )N)r )r r
r r r
� remoteMethod8 s zLocalAsRemote.remoteMethodN)r r r �__doc__r r r r r r r
r s r c @ s* e Zd ZdZd
dd�Zdd� Zdd� Zd S ) �LocalAsyncForwarderzD
A class useful for forwarding a locally-defined interface.
r c C s$ |� |�st�|| _|| _|| _d S )N)Z
providedBy�AssertionError� forwarded�interfaceClass�failWhenNotImplemented)r r# r$ r% r r r
r A s z LocalAsyncForwarder.__init__c O s t | j|�||�S )N)r r# )r r r r r r r
� _callMethodG s z LocalAsyncForwarder._callMethodc O sN | j �|�r&tj| j|f|�|�}|S | jr@t�ttd| ��S t� d �S d S )Nz No Such Method in Interface: %s)
r$ ZqueryDescriptionForr Z
maybeDeferredr& r% r r �NotImplementedErrorr )r r r r �resultr r r
r J s z LocalAsyncForwarder.callRemoteN)r )r r r r r r& r r r r r
r! < s
r! c @ s: e Zd ZdZd
dd�Zdd� Zdd� Zd d
� Zd d � ZdS )�Pagerz5
I am an object which pages out information.
Nc O s> t |�r || _|| _|| _nd| _d| _|| _|j�| � dS )z�
Create a pager with a Reference to a remote collector and
an optional callable to invoke upon completion.
Nr )�callable�callback� callbackArgs�callbackKeyword� _stillPaging� collectorZbrokerZregisterPageProducer)r r/ r+ r r r r r
r Z s zPager.__init__c C s6 | j s0| jjddd� | jdk r0| j| j| j� | j S )z5
(internal) Method called by Broker.
Z endedPagingF)�pbanswerN)r. r/ r r+ r, r- )r r r r
� stillPagingi s
zPager.stillPagingc C s | j jd| �� dd� dS )z5
(internal) Method called by Broker.
�gotPageF)r0 N)r/ r �nextPage)r r r r
� sendNextPages s zPager.sendNextPagec C s
t � �dS )zO
Override this to return an object to be sent to my collector.
N)r' )r r r r
r3 y s zPager.nextPagec C s
d| _ dS )z4
Call this when you're done paging.
r N)r. )r r r r
�
stopPaging s zPager.stopPaging)N) r r r r r r1 r4 r3 r5 r r r r
r) V s
r) c @ s" e Zd ZdZddd�Zdd� ZdS ) � StringPagerz:
A simple pager that splits a string into chunks.
� Nc O s, || _ d| _|| _tj| ||f|�|� d S )Nr )�string�pointer� chunkSizer) r )r r/ �str: r+ r r r r r
r � s zStringPager.__init__c C sD | j | j| j| j � }| j| j7 _| jt| j �kr@| �� |S )N)r8 r9 r: �lenr5 )r �valr r r
r3 � s
zStringPager.nextPage)r7 N)r r r r r r3 r r r r
r6 � s
r6 c @ sB e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )� FilePagerzC
Reads a file in chunks and sends the chunks as they come.
Nc O s* g | _ tj| ||f|�|� | �|� d S )N)�chunksr) r �startProducing)r r/ �fdr+ r r r r r
r � s zFilePager.__init__c s* t �� �|� �� _� j�� fdd�� d S )Nc s � � � S )N)r5 )�x)r r r
�<lambda>� � z*FilePager.startProducing.<locals>.<lambda>)r Z
FileSenderZbeginFileTransferZdeferredZaddBoth)r rA r )r r
r@ � s zFilePager.startProducingc C s || _ |s| j �� d S )N)�producer�resumeProducing)r rE Z streamingr r r
�registerProducer� s zFilePager.registerProducerc C s
d | _ d S )N)rE )r r r r
�unregisterProducer� s z FilePager.unregisterProducerc C s | j �|� d S )N)r? �append)r �chunkr r r
�write� s zFilePager.writec C s6 | j s
dS | j �d�}| j�� | jjd|dd� dS )zD
Get the first chunk read and send it to collector.
Nr r2 F)r0 )r? �poprE rF r/ r )r r= r r r
r4 � s
zFilePager.sendNextPage)N)
r r r r r r@ rG rH rK r4 r r r r
r> � s
r> c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �CallbackPageCollectorz�
I receive pages from the peer. You may instantiate a Pager with a
remote reference to me. I will call the callback with a list of pages
once they are all received.
c C s g | _ || _d S )N)�pagesr+ )r r+ r r r
r � s z CallbackPageCollector.__init__c C s | j �|� d S )N)rN rI )r Zpager r r
�remote_gotPage� s z$CallbackPageCollector.remote_gotPagec C s | � | j� d S )N)r+ rN )r r r r
�remote_endedPaging� s z(CallbackPageCollector.remote_endedPagingN)r r r r r rO rP r r r r
rM � s rM c O s& t �� }| j|t|j�f|�|� |S )zr
A utility method that will call a remote method which expects a
PageCollector as the first argument.
)r ZDeferredr rM r+ )Z
referenceableZ
methodNamer r �dr r r
� getAllPages� s rR N)r Ztwisted.internetr Ztwisted.python.failurer Ztwisted.spreadr Ztwisted.protocolsr r Zzope.interfacer r r r! r) r6 Z IConsumerr> Z
ReferenceablerM rR r r r r
�<module> s 0%