File: //lib/python3/dist-packages/twisted/spread/__pycache__/util.cpython-310.pyc
o
�b� � @ s� d Z ddlmZ ddlmZmZ ddlmZ ddlm Z ddl
m Z G dd� d�Z G d d
� d
�Z
G d d � d �ZG d
d� d�ZG dd� de�Zeej�G dd� de��ZG dd� de j�Zdd� ZdS )z
Utility classes for spread.
� )� implementer)�defer�
interfaces)�basic)�Failure)�pbc @ 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__ �
zLocalMethod.__init__c O s | j j| jg|�R i |��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 s| t | d| �rt| d| �|i |��S zt| d| �}t�||i |���W S ty= t� }| jr6|�� t�|� Y S w )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.
�sync_�async_) �hasattr�getattrr �succeed�
BaseExceptionr �reportAllTracebacks�printTraceback�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 �
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$ |� |�sJ �|| _|| _|| _d S r )�
providedBy� forwarded�interfaceClass�failWhenNotImplemented)r r* r+ r, r
r
r r A s
z LocalAsyncForwarder.__init__c O s t | j|�|i |��S r )r r* )r r# r r r
r
r � _callMethodG s z LocalAsyncForwarder._callMethodc O sP | j �|�rtj| j|g|�R i |��}|S | jr#t�ttd| ��S t� d �S )Nz No Such Method in Interface: %s)
r+ �queryDescriptionForr �
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� collector�broker�registerPageProducer)r r8 r4 r r r
r
r r [ s zPager.__init__c C s: | j s| jjddd� | jdur| j| ji | j�� | j S )�5
(internal) Method called by Broker.
� endedPagingF��pbanswerN)r7 r8 r r4 r5 r6 �r r
r
r � stillPagingj s
zPager.stillPagingc C s | j jd| �� dd� dS )r; �gotPageFr= N)r8 r �nextPager? r
r
r � sendNextPaget s zPager.sendNextPagec C s t � �)zO
Override this to return an object to be sent to my collector.
)r0 r? r
r
r rB z s zPager.nextPagec C s
d| _ dS )z4
Call this when you're done paging.
r N)r7 r? r
r
r �
stopPaging� s
zPager.stopPagingr ) r r r r' r r@ rC rB rD r
r
r
r r2 V s
r2 c @ s" e Zd ZdZddd�Zdd� ZdS ) � StringPagerz:
A simple pager that splits a string into chunks.
� Nc O s2 || _ d| _|| _tj| ||g|�R i |�� d S )Nr )�string�pointer� chunkSizer2 r )r r8 �strI r4 r r r
r
r r � s zStringPager.__init__c C sD | j | j| j| j � }| j| j7 _| jt| j �kr | �� |S r )rG rH rI �lenrD �r �valr
r
r rB � s
zStringPager.nextPage)rF N)r r r r' r rB r
r
r
r rE � s
rE 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 s0 g | _ tj| ||g|�R i |�� | �|� d S r )�chunksr2 r �startProducing)r r8 �fdr4 r r r
r
r r � s zFilePager.__init__c s* t �� �|� �� _� j�� fdd�� d S )Nc s � � � S r )rD )�xr? r
r �<lambda>� s z*FilePager.startProducing.<locals>.<lambda>)r �
FileSender�beginFileTransfer�deferred�addBoth)r rQ r
r? r rP � s zFilePager.startProducingc C s || _ |s | j �� d S d S r )�producer�resumeProducing)r rX � streamingr
r
r �registerProducer� s �zFilePager.registerProducerc C s
d | _ d S r )rX r? r
r
r �unregisterProducer� r& z FilePager.unregisterProducerc C � | j �|� d S r )rO �append)r �chunkr
r
r �write� � zFilePager.writec C s6 | j sdS | j �d�}| j�� | jjd|dd� dS )zD
Get the first chunk read and send it to collector.
Nr rA Fr= )rO �poprX rY r8 r rL r
r
r rC � s
zFilePager.sendNextPager )
r r r r' r rP r[ r\ r` rC r
r
r
r rN � s
rN 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 )�pagesr4 )r r4 r
r
r r � r z CallbackPageCollector.__init__c C r] r )rd r^ )r �pager
r
r �remote_gotPage� ra z$CallbackPageCollector.remote_gotPagec C s | � | j� d S r )r4 rd r? r
r
r �remote_endedPaging� ra z(CallbackPageCollector.remote_endedPagingN)r r r r' r rf rg r
r
r
r rc � s
rc c O s, t �� }| j|t|j�g|�R i |�� |S )zr
A utility method that will call a remote method which expects a
PageCollector as the first argument.
)r �Deferredr rc r4 )�
referenceable�
methodNamer r �dr
r
r � getAllPages� s rl N)r' �zope.interfacer �twisted.internetr r �twisted.protocolsr �twisted.python.failurer �twisted.spreadr r r r( r2 rE � IConsumerrN �
Referenceablerc rl r
r
r
r �<module> s 1%