B 't\@sdZddlmZddlmZddlmZddlmZddl m Z dd l m Z dd l mZdd lmZmZdd lmZmZmZdd lZdddgZdZdZGdddeZefddZddZd S)z Win32 event loop. )unicode_literals)SECURITY_ATTRIBUTES) EventLoop)wrap_in_current_context)Future)InputHookContext)ThreadWithFuture)windllpointer)DWORDBOOLHANDLENWin32EventLoopwait_for_handlescreate_win32_eventicseZdZdZdfdd Zd ddZdd Zd d Zefd d Z ddZ d!ddZ d"ddZ ddZ ddZddZddZddZZS)#rz Event loop for Windows systems. :param recognize_paste: When True, try to discover paste actions and turn the event into a BracketedPaste. Tcs8tt|t|_g|_d|_d|_i|_d|_ dS)NF) superr__init__r_event_calls_from_executorclosed_running _read_fds_inputhook_context)selfZrecognize_paste) __class__=lib/python3.7/site-packages/prompt_toolkit/eventloop/win32.pyr#szWin32EventLoop.__init__NcCs~t|tst|dks"t|s"t|jr0td|jr>tdz2d|_x|s\||qHW|j rn||Wdd|_XdS)z Keep running the event loop until `future` has been set. :param future: :class:`prompt_toolkit.eventloop.future.Future` object. NzEvent loop is already runningzEvent loop already closed.TF) isinstancerAssertionErrorcallabler ExceptionrZdone _run_oncer)rfuture inputhookrrrrun_until_complete2s z!Win32EventLoop.run_until_completecs~|r0jdkrt_fdd}j||t}|jkr\tjj n|j krzj |} |dS)Ncst|rtndS)zF True when there is input ready. The inputhook should return control. r)bool_ready_for_readingINFINITE)wait)rrrreadyTsz'Win32EventLoop._run_once..ready) rr Zcall_inputhookr)r*rr kernel32Z ResetEvent#_process_queued_calls_from_executorr _run_task)rr&r,handlecallbackr)rrr$Ms       zWin32EventLoop._run_oncec Cs@y |Wn0tk r:}z|d|iWdd}~XYnXdS)NZ exception) BaseExceptionZcall_exception_handler)rterrrr/es  zWin32EventLoop._run_taskcCs"|jg}||jt||S)zS Return the handle that is ready for reading or `None` on timeout. )rextendrkeysr)rtimeouthandlesrrrr)msz!Win32EventLoop._ready_for_readingcCs(d|_tj|j|jr$|jdS)NT)rr r-Z CloseHandlerrclose)rrrrr9uszWin32EventLoop.closeFcCst||d}||j|jS)z Run a long running function in a background thread. (This is recommended for code that could block the event loop.) Similar to Twisted's ``deferToThread``. )Zdaemon)r call_from_executorstartr%)rr1Z_daemonZthrrrrun_in_executor~s  zWin32EventLoop.run_in_executorcCs&t|}|j|tj|jdS)zm Call this function in the main event loop. Similar to Twisted's ``callFromThread``. N)rrappendr r-ZSetEventr)rr1Z_max_postpone_untilrrrr:s z!Win32EventLoop.call_from_executorcCs6|jdd}|jdd=x|D]}||q WdS)N)rr/)rZcalls_from_executorcrrrr.s  z2Win32EventLoop._process_queued_calls_from_executorcCs"t|}t|}|||dS)z; Start watching the file descriptor for read availability. N)rmsvcrt get_osfhandleadd_win32_handle)rfdr1hrrr add_readers zWin32EventLoop.add_readercCst|}||dS)z: Stop watching the file descriptor for read availability. N)r?r@remove_win32_handle)rrBrCrrr remove_readers zWin32EventLoop.remove_readercCst|}||j|<dS)z' Add a Win32 handle to the event loop. N)rr)rr0r1rrrrAszWin32EventLoop.add_win32_handlecCs||jkr|j|=dS)z, Remove a Win32 handle from the event loop. N)r)rr0rrrrEs z"Win32EventLoop.remove_win32_handle)T)N)F)N)__name__ __module__ __qualname____doc__rr'r$r/r*r)r9r<r:r.rDrFrArE __classcell__rr)rrrs    cCsjt|tstt|tsttt|}||}tjt||t dt |}|t krZdS||}|SdS)z Waits for multiple handles. (Similar to 'select') Returns the handle which is ready. Returns `None` on timeout. http://msdn.microsoft.com/en-us/library/windows/desktop/ms687025(v=vs.85).aspx FN) r listr!intrlenr r-ZWaitForMultipleObjectsrr WAIT_TIMEOUT)r8r7ZarrtypeZ handle_arrayZretrCrrrrs cCs tjtttdtddS)z} Creates a Win32 unnamed Event . http://msdn.microsoft.com/en-us/library/windows/desktop/ms682396(v=vs.85).aspx TFN)r r-Z CreateEventAr rrrrrrrs )rJZ __future__rZ win32_typesrbasercontextrr%rr&r Zutilsr Zctypesr r Zctypes.wintypesr rrr?__all__rOr*rrrrrrrs&