B \0@sdZddlmZddlmZddlmZyddlmZWn ek rXddl mZYnXddl Z ddl Z ddl Z ddl mZddlmZdd lmZd d lmZy e jZWnek re j ZYnXyeWnek reZYnXd d ZGdddeZdS)zeImplements a fully blocking kernel client. Useful for test suites and blocking terminal interfaces. )print_function)partial)getpass)EmptyN)Type) HBChannel) KernelClient)ZMQSocketChannelcs^fdd}js|Sjdd\}}|g}d|krD|d|dd||_|S) Ncs>|dd}|dd}|f||}|s0|S|j||dS)NreplyFtimeout)r )pop _recv_reply)selfargskwargsr r msg_id)meth=lib/python3.7/site-packages/jupyter_client/blocking/client.pywrapped(s   zreqrep..wrappedzReturns r Z Parametersz/ Parameters ---------- a reply: bool (default: False) Whether to wait for and return reply timeout: float or None (default: None) Timeout to use when waiting for a reply Returns ------- msg_id: str The msg_id of the request sent, if reply=False (default) reply: dict The reply message for this request, if reply=True  )__doc__splitstripappendjoin)rrZbasedoc_partsr)rrreqrep's    rc@seZdZdZdddZeeZeeZeeZ ee Z dddZ e ejZe ejZe ejZe ejZe ejZe ejZe ejZddZd d Zd d ZdddZdS)BlockingKernelClientzA KernelClient with blocking APIs ``get_[channel]_msg()`` methods wait for and return messages on channels, raising :exc:`queue.Empty` if no message arrives within ``timeout`` seconds. NcCs|dkrtd}n t|}ddlm}t|j|sfx.|sdt|krXtd|tdq8Wxny|j j ddd }Wnt k rYnX|d d kr| |P|std t|krhtd |qhWx2y|j j ddd }Wqt k rPYqXqWdS)aHWaits for a response when a client is blocked - Sets future time for timeout - Blocks on shell channel until a message is received - Exit if the kernel has died - If client times out before receiving a message from the kernel, send RuntimeError - Flush the IOPub channel Ninf) KernelManagerz?Kernel didn't respond to heartbeats in %d seconds and timed outg?Tr )blockr msg_typeZkernel_info_replyz*Kernel died before replying to kernel_infoz#Kernel didn't respond in %d seconds)floattimeZmanagerr# isinstanceparentis_alive RuntimeErrorZsleep shell_channelget_msgrZ_handle_kernel_info_reply iopub_channel)rr Z abs_timeoutr#msgrrrwait_for_readyTs4           z#BlockingKernelClient.wait_for_readycCsv|dk rt|}x^|dk r,td|t}y|j|d}Wntk rXtdYnX|dd|krnq|SdS)z0Receive and return the reply for a given requestNr)r zTimeout waiting for reply parent_headerr) monotonicmaxZ get_shell_msgr TimeoutErrorget)rrr deadliner rrrrs z BlockingKernelClient._recv_replycCs|d}|ddrt}ntjdkr*t}nt}y||d}Wn6tk rVd}Yn tk rttj ddSX|j s|j s||dS) zHandle an input requestcontentZpasswordF)promptrN) r5rsys version_infoZ raw_inputinputEOFErrorKeyboardInterruptstdoutwrite stdin_channelZ msg_readyr,)rr/r7r9Zraw_datarrr_stdin_hook_defaults   z(BlockingKernelClient._stdin_hook_defaultcCs|dd}|d}|dkr:tt|d}||dnB|dkr\tj|dd d n |d kr|td |d tjddS)z/Default hook for redisplaying plain-text outputheaderr%r7streamnametext) display_dataexecute_resultdataz text/plainerrorr traceback)fileN)getattrr;rAr@r5printrstderr)rr/r%r7rErrr_output_hook_defaults z)BlockingKernelClient._output_hook_defaultcCs:|dd}|dkr,|j|||d|dn ||dS)z]Output hook when running inside an IPython kernel adds rich output support. rDr%)rHrIrLr7)r)N)sendrR)rsessionsocketr1r/r%rrr_output_hook_kernels z(BlockingKernelClient._output_hook_kernelFTc Cs|jstd|dkr |j}|r6|js6td|j||||||d} | dkrZ|j} |dkrdtjkrddl m } | } t | dd } | rt |j | jj| jj| jj}|dkr|j}|dk rt|}nd}t}|jj}||tj|r |jj}||tjnd}x|dk r4td|t}d |}t||}|sPtd ||krt|jjdd }| |q||krq|jjdd }|d d| krq|||dddkr|dddkrPqW|dk rtd|t}|j| |d S)a!Execute code in the kernel interactively Output will be redisplayed, and stdin prompts will be relayed as well. If an IPython kernel is detected, rich output will be displayed. You can pass a custom output_hook callable that will be called with every IOPub message that is produced instead of the default redisplay. .. versionadded:: 5.0 Parameters ---------- code : str A string of code in the kernel's language. silent : bool, optional (default False) If set, the kernel will execute the code as quietly possible, and will force store_history to be False. store_history : bool, optional (default True) If set, the kernel will store command history. This is forced to be False if silent is True. user_expressions : dict, optional A dict mapping names to expressions to be evaluated in the user's dict. The expression values are returned as strings formatted using :func:`repr`. allow_stdin : bool, optional (default self.allow_stdin) Flag for whether the kernel can send stdin requests to frontends. Some frontends (e.g. the Notebook) do not support stdin requests. If raw_input is called from code executed from such a frontend, a StdinNotImplementedError will be raised. stop_on_error: bool, optional (default True) Flag whether to abort the execution queue, if an exception is encountered. timeout: float or None (default: None) Timeout to use when waiting for a reply output_hook: callable(msg) Function to be called with output messages. If not specified, output will be redisplayed. stdin_hook: callable(msg) Function to be called with stdin_request messages. If not specified, input/getpass will be called. Returns ------- reply: dict The reply message for this request z/IOPub channel must be running to receive outputNz,stdin channel must be running to allow input)silent store_historyuser_expressions allow_stdin stop_on_errorIPythonr) get_ipythonZkernelFg@@zTimeout waiting for output)r r1rrDr%Zstatusr7Zexecution_stateZidle) r.r*r+rZrBexecuterCr;modulesr\r]rOrrVZ display_pubrTZ pub_socketr1rRr2zmqZPollerrUregisterZPOLLINr3dictZpollr4r-r5r)rcoderWrXrYrZr[r Z output_hookZ stdin_hookrr]ZipZ in_kernelr6Z timeout_msZpollerZ iopub_socketZ stdin_socketZeventsZreqr/rrrexecute_interactivesx:          z(BlockingKernelClient.execute_interactive)N)N)FTNNTNNN)__name__ __module__ __qualname__rr0rr Zshell_channel_classZiopub_channel_classZstdin_channel_classrZhb_channel_classrrrr^historyZcompleteinspectZ kernel_infoZ comm_infoZshutdownrCrRrVrdrrrrr Ms( 2          r )rZ __future__r functoolsrrZqueuer ImportErrorZQueuer;r'r`Z traitletsrZjupyter_client.channelsrZjupyter_client.clientrZchannelsr r2AttributeErrorr4 NameErrorr+rr rrrrs0          &