U W[@sdZddlmZddlmZddlmZddlmZddlm Z ddl m Z Gdd d Z Gd d d Z Gd d d ZGdddZGdddeZe e jGdddeZGdddejZddZdS)z Utility classes for spread. )defer)Failure)pb)basic) interfaces) implementerc@seZdZddZddZdS) LocalMethodcCs||_||_dSN)localname)selfr r r 5/usr/lib/python3/dist-packages/twisted/spread/util.py__init__szLocalMethod.__init__cOs|jj|jf||Sr )r callRemoter )r argskwr r r__call__szLocalMethod.__call__N)__name__ __module__ __qualname__rrr r r rrsrc@s$eZdZdZdZddZddZdS) LocalAsRemotezN A class useful for emulating the effects of remote behavior locally. cOsrt|d|r"t|d|||Sz t|d|}t|||WSt}|jr^|t|YSXdS)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)hasattrgetattrrsucceedrreportAllTracebacksZprintTracebackfail)r r rrmethodfr r rr#s zLocalAsRemote.callRemotecCs t||Sr )r)r r r r r remoteMethod8szLocalAsRemote.remoteMethodN)rrr__doc__rrr r r r rrsrc@s*eZdZdZd ddZddZddZd S) LocalAsyncForwarderzD A class useful for forwarding a locally-defined interface. rcCs$||st||_||_||_dSr )Z providedByAssertionError forwardedinterfaceClassfailWhenNotImplemented)r r$r%r&r r rrAszLocalAsyncForwarder.__init__cOst|j|||Sr )rr$)r rrrr r r _callMethodGszLocalAsyncForwarder._callMethodcOsN|j|r&tj|j|f||}|S|jr@tttd|St dSdS)NzNo Such Method in Interface: %s) r%ZqueryDescriptionForrZ maybeDeferredr'r&rrNotImplementedErrorr)r rrrresultr r rrJs zLocalAsyncForwarder.callRemoteN)r)rrrr!rr'rr r r rr"<s r"c@s:eZdZdZd ddZddZddZd d Zd d ZdS)Pagerz5 I am an object which pages out information. NcOs>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)callablecallback callbackArgscallbackKeyword _stillPaging collectorZbrokerZregisterPageProducer)r r0r,rrr r rrZszPager.__init__cCs6|js0|jjddd|jdk r0|j|j|j|jS)5 (internal) Method called by Broker. Z endedPagingFZpbanswerN)r/r0rr,r-r.r r r r stillPagingis  zPager.stillPagingcCs|jjd|dddS)r1gotPageFr2N)r0rnextPager3r r r sendNextPagesszPager.sendNextPagecCs tdS)zO Override this to return an object to be sent to my collector. N)r(r3r r rr6yszPager.nextPagecCs d|_dS)z4 Call this when you're done paging. rN)r/r3r r r stopPagingszPager.stopPaging)N) rrrr!rr4r7r6r8r r r rr*Vs   r*c@s"eZdZdZdddZddZdS) StringPagerz: A simple pager that splits a string into chunks. NcOs,||_d|_||_tj|||f||dS)Nr)stringpointer chunkSizer*r)r r0str=r,rrr r rrszStringPager.__init__cCsD|j|j|j|j}|j|j7_|jt|jkr@||Sr )r;r<r=lenr8r valr r rr6s zStringPager.nextPage)r:N)rrrr!rr6r r r rr9s r9c@sBeZdZdZdddZddZddZd d Zd d Zd dZ dS) FilePagerzC Reads a file in chunks and sends the chunks as they come. NcOs*g|_tj|||f||||dSr )chunksr*rstartProducing)r r0fdr,rrr r rrszFilePager.__init__cs*t|_jfdddS)NcsSr )r8)xr3r rz*FilePager.startProducing..)rZ FileSenderZbeginFileTransferZdeferredZaddBoth)r rEr r3rrDszFilePager.startProducingcCs||_|s|jdSr )producerresumeProducing)r rIZ streamingr r rregisterProducerszFilePager.registerProducercCs d|_dSr )rIr3r r runregisterProducerszFilePager.unregisterProducercCs|j|dSr )rCappend)r chunkr r rwriteszFilePager.writecCs6|js dS|jd}|j|jjd|dddS)zD Get the first chunk read and send it to collector. Nrr5Fr2)rCpoprIrJr0rr@r r rr7s   zFilePager.sendNextPage)N) rrrr!rrDrKrLrOr7r r r rrBs rBc@s(eZdZdZddZddZddZdS) 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. cCsg|_||_dSr )pagesr,)r r,r r rrszCallbackPageCollector.__init__cCs|j|dSr )rRrM)r Zpager r rremote_gotPagesz$CallbackPageCollector.remote_gotPagecCs||jdSr )r,rRr3r r rremote_endedPagingsz(CallbackPageCollector.remote_endedPagingN)rrrr!rrSrTr r r rrQsrQcOs&t}|j|t|jf|||S)zr A utility method that will call a remote method which expects a PageCollector as the first argument. )rZDeferredrrQr,)Z referenceableZ methodNamerrdr r r getAllPagessrVN)r!Ztwisted.internetrZtwisted.python.failurerZtwisted.spreadrZtwisted.protocolsrrZzope.interfacerrrr"r*r9Z IConsumerrBZ ReferenceablerQrVr r r rs       0%