File: //proc/thread-self/root/lib/python3/dist-packages/twisted/spread/__pycache__/util.cpython-38.pyc
U
��W[� � @ 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
�5/usr/lib/python3/dist-packages/twisted/spread/util.py�__init__ s zLocalMethod.__init__c O s | j j| jf|�|�S r )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 sr t | d| �r"t| d| �||�S z t| d| �}t�|||��W S t� }| jr^|�� t�|� Y S X 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 r )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 r )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 r )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 r0 r, r r r
r
r r Z s zPager.__init__c C s6 | j s0| jjddd� | jdk r0| j| j| j� | j S )�5
(internal) Method called by Broker.
Z endedPagingF�ZpbanswerN)r/ r0 r r, r- r. �r r
r
r � stillPagingi s
zPager.stillPagingc C s | j jd| �� dd� dS )r1 �gotPageFr2 N)r0 r �nextPager3 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( r3 r
r
r r6 y s zPager.nextPagec C s
d| _ dS )z4
Call this when you're done paging.
r N)r/ r3 r
r
r �
stopPaging s zPager.stopPaging)N) r r r r! r r4 r7 r6 r8 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 r0 �str= r, r r r
r
r r � s zStringPager.__init__c C sD | j | j| j| j � }| j| j7 _| jt| j �kr@| �� |S r )r; r< r= �lenr8 �r �valr
r
r r6 � s
zStringPager.nextPage)r: N)r r r r! r r6 r
r
r
r r9 � s
r9 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 r )�chunksr* r �startProducing)r r0 �fdr, r r r
r
r r � s zFilePager.__init__c s* t �� �|� �� _� j�� fdd�� d S )Nc s � � � S r )r8 )�xr3 r
r �<lambda>� � z*FilePager.startProducing.<locals>.<lambda>)r Z
FileSenderZbeginFileTransferZdeferredZaddBoth)r rE r
r3 r rD � s zFilePager.startProducingc C s || _ |s| j �� d S r )�producer�resumeProducing)r rI Z streamingr
r
r �registerProducer� s zFilePager.registerProducerc C s
d | _ d S r )rI r3 r
r
r �unregisterProducer� s z FilePager.unregisterProducerc C s | j �|� d S r )rC �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 r5 Fr2 )rC �poprI rJ r0 r r@ r
r
r r7 � s
zFilePager.sendNextPage)N)
r r r r! r rD rK rL rO r7 r
r
r
r rB � s
rB 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 r )�pagesr, )r r, r
r
r r � s z CallbackPageCollector.__init__c C s | j �|� d S r )rR rM )r Zpager
r
r �remote_gotPage� s z$CallbackPageCollector.remote_gotPagec C s | � | j� d S r )r, rR r3 r
r
r �remote_endedPaging� s z(CallbackPageCollector.remote_endedPagingN)r r r r! r rS rT r
r
r
r rQ � s rQ 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 rQ r, )Z
referenceableZ
methodNamer r �dr
r
r � getAllPages� s rV N)r! Ztwisted.internetr Ztwisted.python.failurer Ztwisted.spreadr Ztwisted.protocolsr r Zzope.interfacer r r r" r* r9 Z IConsumerrB Z
ReferenceablerQ rV r
r
r
r �<module> s 0%