B '˜t\ù ã@s„dZddlmZddlmZddlmZddddd gZeƒZda d d„Z Gd d„de ƒZ Gd d„de ƒZ d d„ZGdd „d eƒZdS)a¸ Task local storage for coroutines. Example:: # Create a new task local. my_local = TaskLocal() # Set/retrieve/erase: my_local.set(value) value = my_local.get() my_local.delete() A new scope can be created as follows:: with context(): ... Within this scope, a new value can be assigned, which is only visible within the scope. The scope as passed along when code is sent to an executor and back. é)Úunicode_literals)Úlocal)ÚwrapsÚcontextÚget_context_idÚwrap_in_current_contextÚ TaskLocalÚTaskLocalNotSetErrorcCs ytjStk rdSXdS)z( Return the current context ID or None. rN)Ú_storageÚ context_idÚAttributeError©r r ú?lib/python3.7/site-packages/prompt_toolkit/eventloop/context.pyr)sc@s*eZdZdZd dd„Zdd„Zdd„ZdS) rz5 Context manager that activates a new scope. NcCs"|dk r||_ntd7at|_dS)Né)ÚidÚ_last_context_id)Úselfr r r rÚ__init__5szcontext.__init__cCs6y tj|_Wntk r&d|_YnX|jt_|jS)N)r r Ú _previous_idr r)rr r rÚ __enter__>s   zcontext.__enter__cGs|jdkrt`n|jt_dS)N)rr r )rÚar r rÚ__exit__Gs zcontext.__exit__)N)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrr r r rr1s  c@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) rz< Like a thread local, but tied to the current task. cCs i|_dS)N)r )rr r rrRszTaskLocal.__init__cCs.ytƒ}|j|Stk r(t‚YnXdS)N)rr ÚKeyErrorr )rÚctxr r rÚgetUs  z TaskLocal.getcCstƒ}||j|<dS)N)rr )rÚvaluerr r rÚset\sz TaskLocal.setcCs,tƒ}y |j|=Wntk r&YnXdS)N)rr r)rrr r rÚdelete`s  zTaskLocal.deleteN)rrrrrrr r!r r r rrNs cs,tˆƒs t‚tƒ‰tˆƒ‡‡fdd„ƒ}|S)zq Decorator that takes a function, and ensures that when it's called, the current context will apply. c stˆƒ ˆ||ŽSQRXdS)N)r)rÚkw)Úctx_idÚfuncr rÚnew_funcps z)wrap_in_current_context..new_func)ÚcallableÚAssertionErrorrr)r$r%r )r#r$rrhs c@s eZdZdS)r N)rrrr r r rr wsN)rZ __future__rZ threadingrÚ functoolsrÚ__all__r rrÚobjectrrrÚ Exceptionr r r r rÚs