U 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__szWINCHHandler.__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__2szWINCHHandler.__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|tjkrjdSN)signalSIGWINCHrresize)Zsignumframer rrhandleAs z"WINCHHandler.start..handleN)rrr)rrrr rr 9s zWINCHHandler.startcCs|jdk rttj|jdS)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) OperationcKs tdS)z3 are we dealing with a tty or not? NNotImplementedErrorrkwargsrrrisrawRszOperation.israwcKs tdS)! start execution Nrrrrrr XszOperation.startcKs tdS)z0 if we have terminal, resize it Nrrheightwidthrrrrr^szOperation.resizecCs tdS)zReturn sockets for streams.Nrr rrrsocketsdszOperation.socketsN)rrrrr rr$rrrrrPsrc@sDeZdZdZdddZdddZdd Zd d Zd d ZddZ dS) RunOperationz6 class for handling `docker run`-like command TNcCs^|dkrtdtd}||_||_d|_||_|p8tj|_|pDtj |_ |pPtj |_ ||_ 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,r.r0r1r2r3rrrr ns    zRunOperation.__init__cKs|p |\}}}g}|r>|jr>|tjt|j|dd|r`|tj|t|jdd|r|tj|t|jdd| dds|j j |j f||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. FZwait_for_outputZpropagate_closeStateZRunning) r$r.appendioPumpStreamr2r0r1_container_infor+r r,)rr$rZ pty_stdinZ pty_stdoutZ pty_stderrpumpsrrrr s zRunOperation.startcKs0|jdkr*|}|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)r-r;r0isatty)rrinforrrrs 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 sbdd|rZjj|ddddji}t|}ddrN|St|SndSdS)Nr=z Attach{0}r'streamr3r>) format capitalizer+ attach_socketr,r3r8r:Demuxer)keysocketrAr@rrrrDs   z+RunOperation.sockets..attach_socket)r2r0r1)r;map)rrDrrHrr$szRunOperation.socketscKs|jj|j||ddS)z- resize pty within container r"r#N)r+rr,r!rrrrszRunOperation.resizecCs|j|jS)zA Thin wrapper around client.inspect_container(). )r+Zinspect_containerr,r rrrr;szRunOperation._container_info)TNNNN)N) rrrrr r rr$rr;rrrrr%is   r%TcCs|j||||d}|S)N)ttyr2) exec_create)r+r,Zcommandr.exec_idrrrrLsrLc@sLeZdZdZdddZdddZdd Zd d Zd d ZddZ ddZ dS) ExecOperationz7 class for handling `docker exec`-like command TNcCsF||_||_d|_||_|p tj|_|p,tj|_|p8tj|_d|_dSr) rMr+r-r.r/r0r1r2_info)rr+rMr.r0r1r2rrrr s   zExecOperation.__init__cKsV|p |}g}|jr4|tjt|j|dd|tj|t|jdd|S)r Fr4r5)r$r.r7r8r9r:r2r0)rr$rrAr<rrrr s  zExecOperation.startcKs$|jdkr|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)r-r0r?is_process_ttyrrrrrs zExecOperation.israwcCs:|jj|jd|jd}t|}|r,|St|SdS)zL Return a single socket which is processing all I/O to exec T)rGrKN)r+Z exec_startrMr.r8r:rPrE)rrGrArrrr$s  zExecOperation.socketscKs|jj|j||ddS)z1 resize pty of an execed process rJN)r+Z exec_resizerMr!rrrr szExecOperation.resizecCs|ddS)z9 does execed process have allocated tty? Z ProcessConfigrK) _exec_infor rrrrPszExecOperation.is_process_ttycCs |jdkr|j|j|_|jS)z< Caching wrapper around client.exec_inspect N)rOr+Z exec_inspectrMr rrrrQs zExecOperation._exec_info)TNNN)N) rrrrr r rr$rrPrQrrrrrNs   rNc@s<eZdZdZddZddZd ddZdd 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+rSrrrr <szPseudoTerminal.__init__cCs |jSr)rSr$r rrrr$DszPseudoTerminal.socketsNc Csj|jj|d}dd|D}z"t|||W5QRXW5|rdt||D]\}}t||qNXdS)N)r$cSsg|]}|dqS)F) set_blocking.0prrr Jsz(PseudoTerminal.start..)rSr zipr8rTr _hijack_tty)rr$r<flagspumpflagrrrr Gs zPseudoTerminal.startcCs^|jsdS|pt|jj}|dk rZ|\}}z|jj||dWntk rXYnXdS)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. NrJ)rSrrKsizer0rIOError)rr^ZrowsZcolsrrrrTs zPseudoTerminal.resizec Cstj|jj|jd|dd|D}dd|D}tj||dd\}}z>|D] }|qX|D] }| qjt dd|DrWqWq"t k r}zd|j kr|W5d}~XYq"Xq"W5QRXdS) N)r-cSsg|]}|js|qSr)eofrUrrrrXlsz.PseudoTerminal._hijack_tty..cSsg|]}|jr|jqSr)Z to_streamZ needs_writerUrrrrXms <)ZtimeoutcSsg|] }|qSr)Zis_donerUrrrrXwszThe operation did not complete) rKZTerminalrSr2rrr8ZselectZdo_writeflushallrstrerror) rr<Z read_pumpsZ write_streamsZ read_readyZ write_readyZ write_streamr\errrrZhs   zPseudoTerminal._hijack_tty)N)N) rrrrr r$r rrZrrrrrR s  rR)T)r/rr(ZsslrZ dockerpty.ior8Z dockerpty.ttyrKobjectrrr%rLrNrRrrrrs   6f L