o V+@sddlZddlZddlZddlmZddlmZddlmZGddde Z Gddde Z Gddde Z dd d Z Gd d d e ZGddde ZdS)N)SSLErrorc@s8eZdZdZddZddZddZdd Zd d Zd S) WINCHHandlerz? WINCH Signal handler to keep the PTY correctly sized. cCs||_d|_dS)z Initialize a new WINCH handler for the given PTY. Initializing a handler has no immediate side-effects. The `start()` method must be invoked for the signals to be trapped. N)ptyoriginal_handler)selfrr//usr/lib/python3/dist-packages/dockerpty/pty.py__init__s zWINCHHandler.__init__cCs ||S)z5 Invoked on entering a `with` block. )startrrrr __enter__*szWINCHHandler.__enter__cGs |dS)z4 Invoked on exiting a `with` block. N)stop)r_rrr__exit__2s zWINCHHandler.__exit__cs fdd}ttj|_dS)z Start trapping WINCH signals and resizing the PTY. This method saves the previous WINCH handler so it can be restored on `stop()`. cs|tjkr jdSdSN)signalSIGWINCHrresize)signumframer rrhandleAs z"WINCHHandler.start..handleN)rrr)rrrr rr 9s zWINCHHandler.startcCs"|jdurttj|jdSdS)zU Stop trapping WINCH signals and restore the previous WINCH handler. N)rrrr rrrr Gs zWINCHHandler.stopN) __name__ __module__ __qualname____doc__r r rr r rrrrrs  rc@s,eZdZddZddZddZddZd S) OperationcKt)z3 are we dealing with a tty or not? NotImplementedErrorrkwargsrrrisrawRzOperation.israwcKr)! start execution rrrrrr Xr"zOperation.startcKr)z0 if we have terminal, resize it rrheightwidthr rrrr^r"zOperation.resizecCr)zReturn sockets for streams.rr rrrsocketsdszOperation.socketsN)rrrr!r rr'rrrrrPs  rc@sDeZdZdZdddZdddZdd Zd d Zd d ZddZ dS) RunOperationz6 class for handling `docker run`-like command TNcCs^|dur tdtd}||_||_d|_||_|ptj|_|p"tj |_ |p(tj |_ ||_ dS)Y Initialize the PTY using the docker.Client instance and container dict. NzThe default behaviour of dockerpty is changing. Please add logs=1 to your dockerpty.start call to maintain existing behaviour. See https://github.com/d11wtq/dockerpty/issues/51 for details.) warningswarnDeprecationWarningclient containerraw interactivesysstdoutstderrstdinlogs)rr.r/r1r3r4r5r6rrrr ns     zRunOperation.__init__cKs|p|\}}}g}|r|jr|tjt|j|dd|r0|tj|t|jdd|rA|tj|t|jdd| ddsT|j j |j fi||S)z Present the PTY of the container inside the current process. This will take over the current process' TTY until the container's PTY is closed. Fwait_for_outputpropagate_closeStateRunning) r'r1appendioPumpStreamr5r3r4_container_infor.r r/)rr'r pty_stdin pty_stdout pty_stderrpumpsrrrr s zRunOperation.startcKs0|jdur|}|jo|dd|_|jS)z Returns True if the PTY should operate in raw mode. If the container was not started with tty=True, this will return False. NConfigTty)r0rAr3isatty)rr inforrrr!s zRunOperation.israwcs fdd}t|dS)z Returns a tuple of sockets connected to the pty (stdin,stdout,stderr). If any of the sockets are not attached in the container, `None` is returned in the tuple. c s\dd|r,jj|ddddji}t|}ddr'|St|SdS)NrFz Attach{0}r*streamr6rG) format capitalizer. attach_socketr/r6r>r@Demuxer)keysocketrJrIrrrrMs   z+RunOperation.sockets..attach_socket)r5r3r4)rAmap)rrMrrQrr's zRunOperation.socketscK|jj|j||ddS)z- resize pty within container r%r&N)r.rr/r$rrrrzRunOperation.resizecCs|j|jS)zA Thin wrapper around client.inspect_container(). )r.inspect_containerr/r rrrrAszRunOperation._container_info)TNNNNr) rrrrr r r!r'rrArrrrr(is    r(TcCs|j||||d}|S)N)ttyr5) exec_create)r.r/commandr1exec_idrrrrXsrXc@sLeZdZdZdddZdddZdd Zd d Zd d ZddZ ddZ dS) ExecOperationz7 class for handling `docker exec`-like command TNcCsF||_||_d|_||_|ptj|_|ptj|_|ptj|_d|_dSr) rZr.r0r1r2r3r4r5_info)rr.rZr1r3r4r5rrrr s    zExecOperation.__init__cKsV|p|}g}|jr|tjt|j|dd|tj|t|jdd|S)r#Fr7r9)r'r1r=r>r?r@r5r3)rr'r rJrErrrr s zExecOperation.startcKs$|jdur|jo ||_|jS)z Returns True if the PTY should operate in raw mode. If the exec was not started with tty=True, this will return False. N)r0r3rHis_process_ttyrrrrr!s zExecOperation.israwcCs6|jj|jd|jd}t|}|r|St|S)zL Return a single socket which is processing all I/O to exec T)rPrW)r. exec_startrZr1r>r@r]rN)rrPrJrrrr's   zExecOperation.socketscKrS)z1 resize pty of an execed process rTN)r. exec_resizerZr$rrrr rUzExecOperation.resizecCs|ddS)z9 does execed process have allocated tty? ProcessConfigrW) _exec_infor rrrr]szExecOperation.is_process_ttycCs |jdur |j|j|_|jS)z< Caching wrapper around client.exec_inspect N)r\r. exec_inspectrZr rrrras zExecOperation._exec_info)TNNNr) rrrrr r r!r'rr]rarrrrr[s     r[c@s<eZdZdZddZddZd ddZd d d Zd d ZdS)PseudoTerminala! Wraps the pseudo-TTY (PTY) allocated to a docker container. The PTY is managed via the current process' TTY until it is closed. Example: import docker from dockerpty import PseudoTerminal client = docker.Client() container = client.create_container( image='busybox:latest', stdin_open=True, tty=True, command='/bin/sh', ) # hijacks the current tty until the pty is closed PseudoTerminal(client, container).start() Care is taken to ensure all file descriptors are restored on exit. For example, you can attach to a running container from within a Python REPL and when the container exits, the user will be returned to the Python REPL without adverse effects. cCs||_||_dS)r)N)r. operation)rr.rdrrrr <s zPseudoTerminal.__init__cCs |jSr)rdr'r rrrr'Ds zPseudoTerminal.socketsNc Cs|jj|d}dd|D}z0t| ||Wdn1s#wYW|r;t||D] \}}t||q0dSdS|rQt||D] \}}t||qFww)N)r'cSsg|]}|dqS)F) set_blocking.0prrr Jz(PseudoTerminal.start..)rdr r _hijack_ttyzipr>re)rr'rEflagspumpflagrrrr Gs"  zPseudoTerminal.startcCs`|jsdS|pt|jj}|dur.|\}}z |jj||dWdSty-YdSwdS)z Resize the container's PTY. If `size` is not None, it must be a tuple of (height,width), otherwise it will be determined by the size of the current TTY. NrT)rdr!rWsizer3rIOError)rrprowscolsrrrrTs  zPseudoTerminal.resizec Cstj|jj|jd`| dd|D}dd|D}tj||dd\}}z|D]}|q-|D]}| q6t dd|DrHWnWnt yb}z d |j vrX|WYd}~nd}~wwqWddS1sowYdS) N)r0TcSsg|]}|js|qSr)eofrfrrrrilrjz.PseudoTerminal._hijack_tty..cSsg|] }|jr|jqSr) to_stream needs_writerfrrrrims<)timeoutcSsg|]}|qSr)is_donerfrrrriwszThe operation did not complete) rWTerminalrdr5r!rr>selectdo_writeflushallrstrerror) rrE read_pumps write_streams read_ready write_ready write_streamrnerrrrkhs.     "zPseudoTerminal._hijack_ttyr) rrrrr r'r rrkrrrrrc s  rc)T)r2rr+sslr dockerpty.ior> dockerpty.ttyrWobjectrrr(rXr[rcrrrrs   6 fL