V+ddlZddlZddlZddlmZddlmZddlmZGdde Z Gdde Z Gdde Z dd Z Gd d e ZGd d e Zy)N)SSLErrorc.eZdZdZdZdZdZdZdZy) WINCHHandlerz? WINCH Signal handler to keep the PTY correctly sized. c ||_d|_y)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)selfrs //usr/lib/python3/dist-packages/dockerpty/pty.py__init__zWINCHHandler.__init__s $c&|j|S)z5 Invoked on entering a `with` block. )startr s r __enter__zWINCHHandler.__enter__*s  r c$|jy)z4 Invoked on exiting a `with` block. N)stop)r _s r __exit__zWINCHHandler.__exit__2s r cbfd}tjtj|_y)z Start trapping WINCH signals and resizing the PTY. This method saves the previous WINCH handler so it can be restored on `stop()`. cb|tjk(rjjyyN)signalSIGWINCHrresize)signumframer s r handlez"WINCHHandler.start..handleAs"(!)r N)rrr)r rs` r rzWINCHHandler.start9s! "!' foov Fr cz|j/tjtj|jyy)zU Stop trapping WINCH signals and restore the previous WINCH handler. N)rrrrs r rzWINCHHandler.stopGs-  , MM&//4+@+@ A -r N) __name__ __module__ __qualname____doc__r rrrrr r rrs" % GBr rc$eZdZdZdZdZdZy) Operationc t)z3 are we dealing with a tty or not? NotImplementedErrorr kwargss r israwzOperation.israwR "##r c t)! start execution r'r)s r rzOperation.startXr,r c t)z0 if we have terminal, resize it r'r heightwidthr*s r rzOperation.resize^r,r ct)zReturn sockets for streams.r'rs r socketszOperation.socketsds !##r N)rr r!r+rrr4r#r r r%r%Ps$ $ $ $r r%c8eZdZdZd dZd dZdZdZdZdZ y) RunOperationz6 class for handling `docker run`-like command Nc|tjdtd}||_||_d|_||_|xstj|_|xstj|_ |xstj|_ ||_ y)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)r r=r>r@rBrCrDrEs r r zRunOperation.__init__ns| < MMZ\n oD "&*  *  'cii  r c p|xs|j\}}}g}|rP|jrD|jtjtj |j |d|rD|jtj|tj |jd|rD|jtj|tj |jd|jdds'|jj|jfi||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) r4r@appendioPumpStreamrDrBrC_container_infor=rr>)r r4r* pty_stdin pty_stdout pty_stderrpumpss r rzRunOperation.starts-4,Et||~) :z )) LL4::!6 SXY Z  LLRYYt{{-CUZ[ \  LLRYYt{{-CUZ[ \##%g.y9 DKK  dnn 7 7 r c |j9|j}|jjxr|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. ConfigTty)r?rQrBisatty)r r*infos r r+zRunOperation.israwsG 88 '')D{{))+EXu0EDHxxr cJjfd}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 $ddj|jrjjjj|ddddj i}t j|}ddr|St j|Sy)NrWz Attach{0}r9streamrErX) format capitalizer= attach_socketr>rErNrPDemuxer)keysocketr]rZr s r r`z+RunOperation.sockets..attach_socketsH~k001ABC22NN!Xq&$))<6*>%(!M::f--r )rDrBrC)rQmap)r r`rZs` @r r4zRunOperation.socketss(##% ="?@@r c T|jj|j||y)z- resize pty within container r1r2N)r=rr>r0s r rzRunOperation.resizes! 4>>&Fr cL|jj|jS)zA Thin wrapper around client.inspect_container(). )r=inspect_containerr>rs r rQzRunOperation._container_infos {{,,T^^<commandr@exec_ids r rkrks   GK XG Nr c>eZdZdZd dZd dZdZdZdZdZ d Z y) ExecOperationz7 class for handling `docker exec`-like command Nc||_||_d|_||_|xstj |_|xstj |_|xstj|_d|_yr) rmr=r?r@rArBrCrD_info)r r=rmr@rBrCrDs r r zExecOperation.__init__sV  &*  *  'cii  r c Z|xs|j}g}|jrD|jtjtj |j |d|jtj|tj |jd|S)r.FrGrI)r4r@rMrNrOrPrDrB)r r4r*r]rUs r rzExecOperation.startsv*DLLN    LL4::!6PUV W RWWVRYYt{{%;UST  r c |j1|jjxr|j|_|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. )r?rBrYis_process_ttyr)s r r+zExecOperation.israws9 88 {{))+E0C0C0EDHxxr c|jj|jd|j}t j |}|j r|St j|S)zL Return a single socket which is processing all I/O to exec T)rcrj)r= exec_startrmr@rNrPrtra)r rcr]s r r4zExecOperation.socketssW'' TtGWGW'X6"    M::f% %r c T|jj|j||y)z1 resize pty of an execed process rfN)r= exec_resizermr0s r rzExecOperation.resize s!  V5Ir c.|jddS)z9 does execed process have allocated tty? ProcessConfigrj) _exec_infors r rtzExecOperation.is_process_ttys 1%88r c|j*|jj|j|_|jS)z< Caching wrapper around client.exec_inspect )rqr= exec_inspectrmrs r r{zExecOperation._exec_infos2 :: 11$,,?DJzzr )TNNNr) rr r!r"r rr+r4rrtr{r#r r roros+"  &J 9 r roc2eZdZdZdZdZddZddZdZy) 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. c ||_||_y)r8N)r= operation)r r=rs r r zPseudoTerminal.__init__<s  "r c6|jjSr)rr4rs r r4zPseudoTerminal.socketsDs~~%%''r Nc|jj|}|Dcgc]}|jd}} t|5|j |ddd|r+t ||D]\}}t j||yycc}w#1swY   sA00 A<;A<ctj|jj|jj 5|j  |Dcgc]}|j r|}}|Dcgc])}|jjs|j+}}tj||d\}} |D]}|j|D]}|jt|Dcgc]}|jc}rn dddycc}wcc}wcc}w#t$r} d| j vr| Yd} ~ 8d} ~ wwxYw#1swYyxYw)N)r?<)timeoutzThe operation did not complete)rjTerminalrrDr+reof to_stream needs_writerNselectdo_writeflushallis_donerstrerror) r rUr read_pumps write_streams read_ready write_ready write_streamres r rzPseudoTerminal._hijack_ttyhsA \\$....DNN4H4H4J K KKM).rse" 3B63Bl$$2c=9c=L IIIX\ V\ r