a xd@sdZddlmZddlZddlZddlmZddlmZmZm Z m Z m Z m Z ddl Z ddlmZmZddlmZddlmZmZdd lmZd d gZGd d d e jZGd d d e jZdS)zI Utility for running a prompt_toolkit application in an asyncssh server. ) annotationsN)get_running_loop)Any AwaitableCallableOptionalTextIOcast) AppSessioncreate_app_session)Size) PipeInputcreate_pipe_input) Vt100_OutputPromptToolkitSSHSessionPromptToolkitSSHServerc@seZdZddddddZddd d Zd dd d dZddddZddddZddddZddddddddZ ddddddZ dS)r4Callable[[PromptToolkitSSHSession], Awaitable[None]]boolNoneinteract enable_cprreturncsN|_|_d_d_d_d_d_Gfddd}tt|_ dS)NcsReZdZdddfdd Zdddd Zddd d Zeddfd d ZdS)z0PromptToolkitSSHSession.__init__..Stdoutstrr)datarcs:z"jdur j|ddWnty4Yn0dS)N z )_chanwritereplaceBrokenPipeError)srselfY/var/www/html/Ranjet/env/lib/python3.9/site-packages/prompt_toolkit/contrib/ssh/server.pyr+s   z6PromptToolkitSSHSession.__init__..Stdout.writerrcSsdSNTr#r r#r#r$isatty2sz7PromptToolkitSSHSession.__init__..Stdout.isattycSsdSNr#r'r#r#r$flush5sz6PromptToolkitSSHSession.__init__..Stdout.flushcs"jdusJtjjdS)Nr)rrZ _orig_chanZ get_encodingr'r!r#r$encoding8sz9PromptToolkitSSHSession.__init__..Stdout.encodingN)__name__ __module__ __qualname__rr(r*propertyr+r#r!r#r$Stdout*s r0) rr interact_taskr app_session_input_outputr rstdout)r"rrr0r#r!r$__init__sz PromptToolkitSSHSession.__init__r r%cCs8|jdurtdddS|j\}}}}t||dSdS)zU Callable that returns the current `Size`, required by Vt100_Output. NO)rowscolumns)rr get_terminal_sizer"widthheightpixwidth pixheightr#r#r$ _get_size?s  z!PromptToolkitSSHSession._get_sizer)chanrcCs ||_dSr))r)r"rBr#r#r$connection_madeIsz'PromptToolkitSSHSession.connection_madecCsdSr&r#r!r#r#r$shell_requestedLsz'PromptToolkitSSHSession.shell_requestedcCst||_dSr))rZ create_task _interactr1r!r#r#r$session_startedOsz'PromptToolkitSSHSession.session_startedc s|jdurtdt|jdr6|jjdur6|jd|j}t|j|j||j d|_ t |_ t |j |j dv}||_zHz||IdHWntytYn0W|j|j n|j|j 0Wdn1s0YWdn1s0YdS)Nz,`_interact` called before `connection_made`. set_line_modeF)termr)inputoutput)r ExceptionhasattrZ_editorrGZget_terminal_typerr5rArr4rr3r r2r BaseException traceback print_excclose)r"rHsessionr#r#r$rERs&       z!PromptToolkitSSHSession._interactintobject)r=r>r?r@rcCs|jr|jjr|jjdSr))r2ZappZ _on_resizer<r#r#r$terminal_size_changednsz-PromptToolkitSSHSession.terminal_size_changedr)rdatatypercCs|jdurdS|j|dSr))r3Z send_text)r"rrUr#r#r$ data_receivedus z%PromptToolkitSSHSession.data_receivedN) r,r-r.r6rArCrDrFrErTrVr#r#r#r$rs) c@sFeZdZdZdddddddd Zd dd d d ZddddZdS)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. TrrrrrcCs||_||_dSr))rr)r"rrr#r#r$r6szPromptToolkitSSHServer.__init__r)usernamercCsdS)NFr#)r"rXr#r#r$ begin_authsz!PromptToolkitSSHServer.begin_authrr%cCst|j|jdS)NrW)rrrr!r#r#r$session_requestedsz(PromptToolkitSSHServer.session_requestedN)r,r-r.__doc__r6rYrZr#r#r#r$r}s ) )r[ __future__rZasynciorNrtypingrrrrrr ZasyncsshZ"prompt_toolkit.application.currentr r Zprompt_toolkit.data_structuresr Zprompt_toolkit.inputr rZprompt_toolkit.output.vt100r__all__ZSSHServerSessionrZ SSHServerrr#r#r#r$s     h