B 't\@spdZddlmZmZddlmZddlmZmZddlm Z ddl Z dd gZ Gd d d e Z Gd ddeZdS) z9 Future implementation for the prompt_toolkit eventloop. )unicode_literalsprint_function) EventLoop)get_context_idcontext)get_event_loopNFutureInvalidStateErrorc@seZdZdZdS)r z- The operation is not allowed in this state. N)__name__ __module__ __qualname____doc__rr>lib/python3.7/site-packages/prompt_toolkit/eventloop/future.pyr sc@seZdZdZd ddZejdkr(ddZedd Z ed d Z d d Z ddZ ddZ ddZddZddZddZddZed!ddZddZeZdS)"r z `Future` object for use with the prompt_toolkit event loops. (Not by accident very similar to asyncio -- but much more limited in functionality. They are however not meant to be used interchangeable.) NcCsL|dkst|tst|pt|_g|_d|_d|_d|_d|_ t |_ dS)NF) isinstancerAssertionErrorrloopdone_callbacks_result _exception_done_retrieved_resultr_ctx_id)selfrrrr__init__s zFuture.__init__)cCs6|jr2|js2|j}d|jj||d}|j|dS)Nz %s exception was never retrieved)message exceptionZfuture)rr __class__r rZcall_exception_handler)rexcrrrr__del__/s zFuture.__del__cCs|}|||S)z Returns a Future for which the result has been set to the given result. Similar to Twisted's `Deferred.succeed()`. ) set_result)clsresultfrrrsucceed:s zFuture.succeedcCs|}|||S)z Returns a Future for which the error has been set to the given result. Similar to Twisted's `Deferred.fail()`. ) set_exception)r$r%r&rrrfailDs z Future.failcs0jjr,fdd}j|dS)z Add a callback to be run when the future becomes done. (This callback will be called with one argument only: this future object.) csd_dS)NT)rr)callbackrrrcall_cbXsz)Future.add_done_callback..call_cbN)rappendrrcall_from_executor)rr*r+r)r*rradd_done_callbackNs zFuture.add_done_callbackcCs&|jrtd||_d|_|dS)z* Mark the future done and set its result. z#Future result has been set already.TN)rr r_call_callbacks)rr%rrrr#^s zFuture.set_resultcCs.|jrtd||_d|_|jr*|ndS)z, Mark the future done and set an exception. z#Future result has been set already.TN)rr rrr/)rrrrrr(gs zFuture.set_exceptioncs6jddrd_fdd}j|dS)NTc s0tjxD] }|qWWdQRXdS)N)rr)cb)rrrr call_them_alls  z-Future._call_callbacks..call_them_all)rrrr-)rr1r)rrrr/{s zFuture._call_callbackscCs(|js td|_|jr|jn|jSdS)z+ Return the result this future represents. TN)rr rrr)rrrrr%s z Future.resultcCs|js td|_|jS)z3 Return the exception that was set on this future. T)rr rr)rrrrrszFuture.exceptioncCs|jS)z Return True if the future is done. Done means either that a result / exception are available, or that the future was cancelled. )r)rrrrdonesz Future.donecs(ddlm}||jfdd}S)zE Turn this `Future` into an asyncio `Future` object. r)r cs*|r|n|dS)N)rr(r#r%)r&) asyncio_frr_sz#Future.to_asyncio_future.._)Zasyncior r.)rr r4r)r3rto_asyncio_futures zFuture.to_asyncio_futurecs ||d|jfdd}S)zQ Return a prompt_toolkit `Future` from the given asyncio Future. )rcs*|r|n|dS)N)rr(r#r%)r3)r&rrr4sz%Future.from_asyncio_future.._)r.)r$r3rr4r)r&rfrom_asyncio_futures zFuture.from_asyncio_futurecCs |S)z! For compatibility with asyncio. )r5__iter__)rrrrr7szFuture.__iter__)N)N)r r r rrsys version_infor" classmethodr'r)r.r#r(r/r%rr2r5r6r7 __await__rrrrr s$      )rZ __future__rrbaserrrdefaultsrr8__all__ Exceptionr objectr rrrrs