MhdZddlmZddlZddlZddlmZddlmZmZm Z m Z m Z ddl Z ddl mZmZddlmZddlmZmZdd lmZd d gZGd d e jZGd d e jZdS)zI Utility for running a prompt_toolkit application in an asyncssh server. ) annotationsN)get_running_loop)AnyCallable CoroutineTextIOcast) AppSessioncreate_app_session)Size) PipeInputcreate_pipe_input) Vt100_OutputPromptToolkitSSHSessionPromptToolkitSSHServercNeZdZddZdd Zdd Zdd Zd dZd dZd!dZ d"dZ dS)#rinteract>Callable[[PromptToolkitSSHSession], Coroutine[Any, Any, None]] enable_cprboolreturnNonec|_|_d_d_d_d_d_Gfdd}tt|_ dS)NcHeZdZd fd Zd dZd dZedfd Zd S)0PromptToolkitSSHSession.__init__..Stdoutdatastrrrc j0j|dddSdS#t$rYdSwxYw)N z )_chanwritereplaceBrokenPipeError)srselfs a/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/prompt_toolkit/contrib/ssh/server.pyr!z6PromptToolkitSSHSession.__init__..Stdout.write,sdz- ((dF)C)CDDDDD.-&DDs5< A  A rcdSNTr$s r&isattyz7PromptToolkitSSHSession.__init__..Stdout.isatty3stcdSNr)r*s r&flushz6PromptToolkitSSHSession.__init__..Stdout.flush6sr,cxjJtjjdS)Nr)r r _orig_chan get_encoding)r$r%s r&encodingz9PromptToolkitSSHSession.__init__..Stdout.encoding9s5z---4:0==??BCCCr,N)rrrrrrrr)rr)__name__ __module__ __qualname__r!r+r/propertyr3r%sr&Stdoutr+s               D D D D DX D D Dr,r;) rr interact_taskr app_session_input_outputr rstdout)r%rrr;s` r&__init__z PromptToolkitSSHSession.__init__s ! $8<!% .2 )- ,0  D D D D D D D D D D&66688,, r,r c|jtddS|j\}}}}t||S)zU Callable that returns the current `Size`, required by Vt100_Output. NO)rowscolumns)r r get_terminal_sizer%widthheightpixwidth pixheights r& _get_sizez!PromptToolkitSSHSession._get_size@sN : R,,, ,151M1M1O1O .E68YVU333 3r,chanrc||_dSr.)r )r%rNs r&connection_madez'PromptToolkitSSHSession.connection_madeJs  r,cdSr(r)r:s r&shell_requestedz'PromptToolkitSSHSession.shell_requestedMstr,cvt||_dSr.)r create_task _interactr<r:s r&session_startedz'PromptToolkitSSHSession.session_startedPs.-//;;DNNr r=r BaseException traceback print_excclose)r%rYsessions r&rUz!PromptToolkitSSHSession._interactSs* : JKK K 4: / / ,DJ4F4R J $ $U + + +z++--# Kdt     (DK#$+dlKKK (w#* (------------$***')))))*J$$&&&K%%''''J$$&&&K%%''''' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (sl"!FF C('D>(DD>DD> 3F>4E22F6 FF F F FF!FrIintrJrKobjectrLcp|jr,|jjr"|jjdSdSdSr.)r=app _on_resizerHs r&terminal_size_changedz-PromptToolkitSSHSession.terminal_size_changedosN   . 0 4 .   + + - - - - - . . . .r,rrdatatypecL|jdS|j|dSr.)r> send_text)r%rrks r& data_receivedz%PromptToolkitSSHSession.data_receivedvs+ ;  F d#####r,Nrrrrrr)rr )rNrrrr4r5) rIrerJrerKrfrLrfrr)rrrkrfrr) r6r7r8rArMrPrRrVrUrjrnr)r,r&rrs'-'-'-'-R4444NNNN((((8....$$$$$$r,c0eZdZdZdddd Zdd ZddZdS)raZ Run a prompt_toolkit application over an asyncssh server. This takes one argument, an `interact` function, which is called for each connection. This should be an asynchronous function that runs the prompt_toolkit applications. This function runs in an `AppSession`, which means that we can have multiple UI interactions concurrently. Example usage: .. code:: python async def interact(ssh_session: PromptToolkitSSHSession) -> None: await yes_no_dialog("my title", "my text").run_async() prompt_session = PromptSession() text = await prompt_session.prompt_async("Type something: ") print_formatted_text('You said: ', text) server = PromptToolkitSSHServer(interact=interact) loop = get_running_loop() loop.run_until_complete( asyncssh.create_server( lambda: MySSHServer(interact), "", port, server_host_keys=["/etc/ssh/..."], ) ) loop.run_forever() :param enable_cpr: When `True`, the default, try to detect whether the SSH client runs in a terminal that responds to "cursor position requests". That way, we can properly determine how much space there is available for the UI (especially for drop down menus) to render. Trrrrrrrc"||_||_dSr.)rr)r%rrs r&rAzPromptToolkitSSHServer.__init__s ! $r,usernamercdS)NFr))r%rss r& begin_authz!PromptToolkitSSHServer.begin_authsur,rc8t|j|jS)Nrq)rrrr:s r&session_requestedz(PromptToolkitSSHServer.session_requesteds&t}QQQQr,Nro)rsrrr)rr)r6r7r8__doc__rArurwr)r,r&rr~sp##R %%%%%%RRRRRRr,)rx __future__rasynciorartypingrrrrr asyncssh"prompt_toolkit.application.currentr r prompt_toolkit.data_structuresr prompt_toolkit.inputr rprompt_toolkit.output.vt100r__all__SSHServerSessionr SSHServerrr)r,r&rsS#"""""$$$$$$99999999999999MMMMMMMM//////========444444 $&> ?e$e$e$e$e$h7e$e$e$P4R4R4R4R4RX/4R4R4R4R4Rr,