U ~Hj@sdZddlmZddlmZmZddlmZddlm Z ddl m Z GdddZ Gd d d Z Gd d d ZGd ddZGdddeZeejGdddeZGddde jZddZdS)z Utility classes for spread. ) implementer)defer interfaces)basic)Failure)pbc@seZdZddZddZdS) LocalMethodcCs||_||_dSN)localname)selfr r r 7/tmp/pip-unpacked-wheel-ewod1tgb/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. cOszt|d|r"t|d|||Sz t|d|}t|||WStk rtt}|jrf|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) hasattrgetattrrsucceed BaseExceptionrreportAllTracebacksZprintTracebackfail)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 r1r-rrr r rr[szPager.__init__cCs6|js0|jjddd|jdk r0|j|j|j|jS)5 (internal) Method called by Broker. Z endedPagingFZpbanswerN)r0r1rr-r.r/r r r r stillPagingjs  zPager.stillPagingcCs|jjd|dddS)r2gotPageFr3N)r1rnextPager4r r r sendNextPagetszPager.sendNextPagecCs tdS)zO Override this to return an object to be sent to my collector. N)r)r4r r rr7zszPager.nextPagecCs d|_dS)z4 Call this when you're done paging. rN)r0r4r r r stopPagingszPager.stopPaging)N) rrrr"rr5r8r7r9r 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 r1str>r-rrr r rrszStringPager.__init__cCsD|j|j|j|j}|j|j7_|jt|jkr@||Sr )r<r=r>lenr9r valr r rr7s zStringPager.nextPage)r;N)rrrr"rr7r r r rr:s r:c@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 r1fdr-rrr r rrszFilePager.__init__cs*t|_jfdddS)NcsSr )r9)xr4r rz*FilePager.startProducing..)rZ FileSenderZbeginFileTransferdeferredZaddBoth)r rFr r4rrEszFilePager.startProducingcCs||_|s|jdSr )producerresumeProducing)r rKZ streamingr r rregisterProducerszFilePager.registerProducercCs d|_dSr )rKr4r r runregisterProducerszFilePager.unregisterProducercCs|j|dSr )rDappend)r chunkr r rwriteszFilePager.writecCs6|js dS|jd}|j|jjd|dddS)zD Get the first chunk read and send it to collector. Nrr6Fr3)rDpoprKrLr1rrAr r rr8s   zFilePager.sendNextPage)N) rrrr"rrErMrNrQr8r r r rrCs rCc@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 )rTrO)r pager r rremote_gotPagesz$CallbackPageCollector.remote_gotPagecCs||jdSr )r-rTr4r r rremote_endedPagingsz(CallbackPageCollector.remote_endedPagingN)rrrr"rrVrWr r r rrSsrScOs&t}|j|t|jf|||S)zr A utility method that will call a remote method which expects a PageCollector as the first argument. )rZDeferredrrSr-)Z referenceable methodNamerrdr r r getAllPagessrZN)r"Zzope.interfacerZtwisted.internetrrZtwisted.protocolsrZtwisted.python.failurerZtwisted.spreadrrrr#r+r:Z IConsumerrCZ ReferenceablerSrZr r r rs      1%