o i@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 YC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/spread/util.py__init__ zLocalMethod.__init__cOs|jj|jg|Ri|Sr )r callRemoter )r argskwr r r__call__szLocalMethod.__call__N)__name__ __module__ __qualname__rrr r r rrs rc@s$eZdZdZdZddZddZdS) LocalAsRemotezN A class useful for emulating the effects of remote behavior locally. cOs|t|d|rt|d||i|Szt|d|}t||i|WSty=t}|jr6|t|YSw)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 remoteMethod8 zLocalAsRemote.remoteMethodN)rrr__doc__rrr"r r r rrs  rc@s*eZdZdZd ddZddZddZd S) LocalAsyncForwarderzD A class useful for forwarding a locally-defined interface. rcCs$||sJ||_||_||_dSr )Z providedBy forwardedinterfaceClassfailWhenNotImplemented)r r&r'r(r r rrAs zLocalAsyncForwarder.__init__cOst|j||i|Sr )rr&)r r rrr r r _callMethodGszLocalAsyncForwarder._callMethodcOsP|j|rtj|j|g|Ri|}|S|jr#tttd|St dS)NzNo Such Method in Interface: %s) r'ZqueryDescriptionForrZ maybeDeferredr)r(rrNotImplementedErrorr)r r rrresultr 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 r2r.rrr r rr[szPager.__init__cCs:|js|jjddd|jdur|j|ji|j|jS)5 (internal) Method called by Broker. Z endedPagingFZpbanswerN)r1r2rr.r/r0r r r r stillPagingjs  zPager.stillPagingcCs|jjd|dddS)r3gotPageFr4N)r2rnextPager5r r r sendNextPagetszPager.sendNextPagecCst)zO Override this to return an object to be sent to my collector. N)r*r5r r rr8zszPager.nextPagecCs d|_dS)z4 Call this when you're done paging. rN)r1r5r r r stopPagings zPager.stopPagingr ) rrrr$rr6r9r8r:r r r rr,Vs   r,c@s"eZdZdZdddZddZdS) StringPagerz: A simple pager that splits a string into chunks. NcOs2||_d|_||_tj|||g|Ri|dS)Nr)stringpointer chunkSizer,r)r r2str?r.rrr r rrs zStringPager.__init__cCsD|j|j|j|j}|j|j7_|jt|jkr ||Sr )r=r>r?lenr:r valr r rr8s zStringPager.nextPage)r<N)rrrr$rr8r 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. NcOs0g|_tj|||g|Ri|||dSr )chunksr,rstartProducing)r r2fdr.rrr r rrszFilePager.__init__cs*t|_jfdddS)NcsSr )r:)xr5r rsz*FilePager.startProducing..)rZ FileSenderZbeginFileTransferdeferredZaddBoth)r rGr r5rrFszFilePager.startProducingcCs||_|s |jdSdSr )producerresumeProducing)r rKZ streamingr r rregisterProducerszFilePager.registerProducercCs d|_dSr )rKr5r r runregisterProducerr#zFilePager.unregisterProducercC|j|dSr )rEappend)r chunkr r rwritezFilePager.writecCs6|jsdS|jd}|j|jjd|dddS)zD Get the first chunk read and send it to collector. Nrr7Fr4)rEpoprKrLr2rrBr r rr9s   zFilePager.sendNextPager ) rrrr$rrFrMrNrRr9r r r rrDs  rDc@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 rrrzCallbackPageCollector.__init__cCrOr )rVrP)r pager r rremote_gotPagerSz$CallbackPageCollector.remote_gotPagecCs||jdSr )r.rVr5r r rremote_endedPagingrSz(CallbackPageCollector.remote_endedPagingN)rrrr$rrXrYr r r rrUs  rUcOs,t}|j|t|jg|Ri||S)zr A utility method that will call a remote method which expects a PageCollector as the first argument. N)rZDeferredrrUr.)Z referenceable methodNamerrdr r r getAllPagess r\N)r$Zzope.interfacerZtwisted.internetrrZtwisted.protocolsrZtwisted.python.failurerZtwisted.spreadrrrr%r,r;Z IConsumerrDZ ReferenceablerUr\r r r rs      1%