← Back
Editing: 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%
Save File
Cancel