Current File : //usr/lib/python2.7/site-packages/salt/utils/nb_popen.pyc
�
���^c @@ s d Z d d l m Z m Z m Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z
d d l Z d d l Z e j
d k Z y6 d d l m Z m Z d d l m Z d d l Z Wn e k
r� d d l Z n Xe
j e � Z d e j f d � � YZ d S( u
:codeauthor: Pedro Algarvio (pedro@algarvio.me)
salt.utils.nb_popen
~~~~~~~~~~~~~~~~~~~
Non blocking subprocess Popen.
This functionality has been adapted to work on windows following the recipe
found on:
http://code.activestate.com/recipes/440554/
i ( t absolute_importt unicode_literalst print_functionNu win32( t ReadFilet WriteFile( t
PeekNamedPipet NonBlockingPopenc B@ s� e Z d Z d Z d � Z d d � Z d d � Z d d d � Z d � Z d � Z
e ro d � Z d
� Z
n d � Z d � Z
d
d � Z d d � Z RS( u$ salt.utils.nb_popen.STDOUT.PID-{pid}u$ salt.utils.nb_popen.STDERR.PID-{pid}c O@ s_ | j d t � | _ | j d d � | _ | j d | j � | _ | j d | j � | _ | j d d � } | j d d � } t t | � j
| | � t j | j � | _
t j | j j d | j � � | _ | t j k r� | j
| _ | j | _ n9 t j | j � | _ t j | j j d | j � � | _ t j d | j | d k rT| n | � d S(
Nu stream_stdsu max_size_in_memi � u stdout_logger_nameu stderr_logger_nameu logging_commandu stderrt pidu" Running command under pid %s: '%s'( t popt Falset stream_stdst max_size_in_memt _stdout_logger_name_t _stderr_logger_name_t Nonet gett superR t __init__t tempfilet SpooledTemporaryFilet stdout_bufft loggingt getLoggert formatR t _stdout_loggert
subprocesst STDOUTt stderr_bufft _stderr_loggert logt info( t selft argst kwargst logging_commandt stderr( ( s7 /usr/lib/python2.7/site-packages/salt/utils/nb_popen.pyR . s. c C@ s | j d | � S( Nu stdout( t _recv( R t maxsize( ( s7 /usr/lib/python2.7/site-packages/salt/utils/nb_popen.pyt recv_ s c C@ s | j d | � S( Nu stderr( R$ ( R R% ( ( s7 /usr/lib/python2.7/site-packages/salt/utils/nb_popen.pyt recv_errb s u c C@ s( | j | � | j | � | j | � f S( N( t sendR& R' ( R t inputR% ( ( s7 /usr/lib/python2.7/site-packages/salt/utils/nb_popen.pyt send_recve s c C@ s= | d k r d } n | d k r* d } n t | | � | f S( Ni i ( R t getattr( R t whichR% ( ( s7 /usr/lib/python2.7/site-packages/salt/utils/nb_popen.pyt get_conn_maxsizeh s
c C@ s' t | | � j � t | | d � d S( N( R+ t closet setattrR ( R R, ( ( s7 /usr/lib/python2.7/site-packages/salt/utils/nb_popen.pyt _closeo s c C@ s� | j s
d Sy1 t j | j j � � } t | | � \ } } Wne t k
r[ | j d � St j j
t f k
r� } | j d d t
j f k r� | j d � S� n X| S( Nu stdini im ( t stdinR t msvcrtt
get_osfhandlet filenoR t
ValueErrorR0 R t
pywintypest errort ExceptionR t errnot ESHUTDOWN( R R) t xt errCodet writtent why( ( s7 /usr/lib/python2.7/site-packages/salt/utils/nb_popen.pyR( t s
c
C@ s� | j | | � \ } } | d k r( d Sym t j | j � � } t | d � \ } } } | | k rm | } n | d k r� t | | d � \ } } n Wne t k
r� | j | � St j
j t f k
r� } | j
d d t j f k r� | j | � S� n Xt | d j | � � j | � t | d j | � � j | j � � | j rct t | � j | � n | j r~| j | � } n | S( Ni im u {0}_buffu _{0}_logger( R- R R2 R3 R4 R R R5 R0 R R6 R7 R8 R R9 R: R+ R t writet debugt rstripR
t syst universal_newlinest _translate_newlines(
R R, R% t connR; t readt nAvailt nMessageR< R> ( ( s7 /usr/lib/python2.7/site-packages/salt/utils/nb_popen.pyR$ � s.
% c C@ s� | j s
d St j g | j g g d � d s3 d Sy t j | j j � | � } Wn9 t k
r� } | j d t j k r� | j
d � S� n X| S( Ni i u stdin( R1 R t selectt osR? R4 t OSErrorR R9 t EPIPER0 ( R R) R= R> ( ( s7 /usr/lib/python2.7/site-packages/salt/utils/nb_popen.pyR( � s "
c C@ s^ | j | | � \ } } | d k r( d St j | t j � } | j sf t j | t j | t j B� n z� t j | g g g d � d s� d S| j | � } | s� | j
| � S| j r� | j | � } n t
| d j | � � j | � t
| d j | � � j | j � � | j r/t
t | � j | � n | SWd | j sYt j | t j | � n Xd S( Ni u u {0}_buffu _{0}_logger( R- R t fcntlt F_GETFLt closedt F_SETFLRJ t
O_NONBLOCKRI RF R0 RC RD R+ R R? R@ RA R
RB ( R R, R% RE t flagst buff( ( s7 /usr/lib/python2.7/site-packages/salt/utils/nb_popen.pyR$ � s*
% g{�G�z�?c C@ s� d } x� | j � d k r� | j d k r= d } | j � n | j d k r_ d } | j � n | d 7} | d k r� d } | j � \ } } | r� t j | � n | r� t j | � q� n t
j | � q Wd S( Ni i id ( t pollR t stdoutR& R# R' t communicateR R@ R7 t timet sleep( R t intervalt silent_iterationst
stdoutdatat
stderrdata( ( s7 /usr/lib/python2.7/site-packages/salt/utils/nb_popen.pyt poll_and_read_until_finish� s"
c C@ sl t t | � j | � | j j � | j j d � | j j � | j j d � | j j � | j j � f S( Ni ( R R RV R t flusht seekR RF ( R R) ( ( s7 /usr/lib/python2.7/site-packages/salt/utils/nb_popen.pyRV � s
N( t __name__t
__module__R R
R R R& R' R* R- R0 t mswindowsR( R$ R] RV ( ( ( s7 /usr/lib/python2.7/site-packages/salt/utils/nb_popen.pyR ( s 1 ( t __doc__t
__future__R R R RJ RB RW R9 RI R R R t platformRb t win32fileR R t win32pipeR R2 t ImportErrorRM R R` R t PopenR ( ( ( s7 /usr/lib/python2.7/site-packages/salt/utils/nb_popen.pyt <module> s$
Mini Shell