Current File : //lib/python2.7/site-packages/salt/modules/container_resource.pyo
�
���^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
d d l Z
d d l Z
d d l m Z m Z e j e � Z d Z d � Z d � Z d � Z d � Z e d d d e d e d
e d e d d � � Z e d d d e e d � � Z d S(
ua
Common resources for LXC and systemd-nspawn containers
.. versionadded:: 2015.8.0
These functions are not designed to be called directly, but instead from the
:mod:`lxc <salt.modules.lxc>`, :mod:`nspawn <salt.modules.nspawn>`, and
:mod:`docker <salt.modules.docker>` execution modules. They provide for
common logic to be re-used for common actions.
i ( t absolute_importt print_functiont unicode_literalsN( t CommandExecutionErrort SaltInvocationErroruJ PATH=/bin:/usr/bin:/sbin:/usr/sbin:/opt/bin:/usr/local/bin:/usr/local/sbinc @ s" t j � � � f d � � } | S( u;
Decorator for common function argument validation
c @ s� | j d � } | j d � } i d
d 6d d 6d d 6} | | k ro t d j | d
j t | � � � � � n | | | k r� t d j d
j | | � � � � n | d k r� t j j j d � r� t d � � n � | t j j j
| � � S( Nu container_typeu exec_driveru
lxc-attachu nsenteru docker-execu dockeru lxcu nspawnu2 Invalid container type '{0}'. Valid types are: {1}u , u8 Invalid command execution driver. Valid drivers are: {0}um The 'lxc-attach' execution driver has been chosen, but lxc-attach is not available. LXC may not be installed.( u
lxc-attachu nsenteru docker-exec( u
lxc-attach( u nsenter( t getR t formatt joint sortedt saltt utilst patht whicht argst clean_kwargs( R
t kwargst container_typet exec_drivert valid_driver( t wrapped( sC /usr/lib/python2.7/site-packages/salt/modules/container_resource.pyt wrapper' s$
! " ( t functoolst wraps( R R ( ( R sC /usr/lib/python2.7/site-packages/salt/modules/container_resource.pyt _validate# s c C@ s
d j | � S( uE
Return the nsenter command to attach to the named container
u4 nsenter --target {0} --mount --uts --ipc --net --pid( R ( t pid( ( sC /usr/lib/python2.7/site-packages/salt/modules/container_resource.pyt _nsenterC s c C@ sV | | d j t j | � � d t �d } y | j � d SWn t k
rQ d SXd S( u9
Get the MD5 checksum of a file from a container
u
md5sum {0}t ignore_retcodeu stdouti N( R t pipest quotet Truet splitt
IndexErrort None( t nameR t run_funct output( ( sC /usr/lib/python2.7/site-packages/salt/modules/container_resource.pyt _get_md5M s
c C@ sr yE | j d � rD t d | � } | s@ t d j | � � � n | SWn& t k
rm t d j | � � � n X| S( u�
Wrapper for cp.cache_file which raises an error if the file was unable to
be cached.
CLI Example:
.. code-block:: bash
salt myminion container_resource.cache_file salt://foo/bar/baz.txt
u salt://u
cp.cache_fileu Unable to cache {0}u Invalid source file {0}( t
startswitht __salt__R R t AttributeErrorR ( t sourcet
cached_source( ( sC /usr/lib/python2.7/site-packages/salt/modules/container_resource.pyt
cache_file[ s
u debugc
C@ s5 d2 }
| d1 k r d } n3 | |
k rH t d j d j |
� � � � n d } | d1 k si t | t � rr g } nX t | t t f � s� y | j d � } Wq� t k
r� t
j d
� g } q� Xn | } | d k r�d } |
r| d
j t j
|
� � 7} n | t k r<| d 7} d | k r<| d j t � 7} q<n | d j g | D]7 } | t j k rLd j | t j
t j | � � ^ qL� 7} | d j t j
| � | � 7} n�| d k r�t d j | � | � } d j | � } | t k r | d 7} d | k r | d j t � 7} q n | d j g | D]7 } | t j k r.d j | t j
t j | � � ^ q.� 7} | d j | � 7} n� | d k r_d } | r�| d 7} n | d j | � 7} | t k r�| d 7} d | k r�| d j t � 7} q�n | d j g | D]7 } | t j k rd j | t j
t j | � � ^ q� 7} | d j | � 7} n | s�t | | d | d | d | d! | �} n�d3 \ } } t j j j | d# | d$ | d% k r�d% n d& d' | d( | d) t d* t d+ t d, t �} z0y� x� | j ryb | j � \ } } | r-| | 7} n | rY| d1 k rL| | 7} qY| | 7} n t j d- � Wq�t k
r{Pq�Xq�W| d1 k r�| n" i | j d 6d. d/ 6| d 6| d 6} Wng t j j j k
r!t j � } t
j! | � | d1 k r�| n i d0 d 6d. d/ 6| d 6| d 6} n XWd1 | j" � X| S(4 uS
Common logic for running shell commands in containers
path
path to the container parent (for LXC only)
default: /var/lib/lxc (system default)
CLI Example:
.. code-block:: bash
salt myminion container_resource.run mycontainer 'ps aux' container_type=docker exec_driver=nsenter output=stdout
u stdoutu stderru retcodeu allu cmd.runu0 'output' param must be one of the following: {0}u , u cmd.run_allu ,u Invalid keep_env value, ignoringu
lxc-attachu lxc-attach u -P {0} u --clear-env u PATHu --set-var {0} u u --set-var {0}={1}u -n {0} -- {1}u nsenteru {0}.pidu8 nsenter --target {0} --mount --uts --ipc --net --pid -- u env -i u {0} u {0}={1}u {0}u docker-execu docker exec u -i t stdint python_shellt output_loglevelR u t shellt log_stdin_levelu quietu infot log_stdout_levelt log_stderr_levelt
log_stdoutt
log_stderrt
stream_stdoutt
stream_stderrg �?i u pidi N( u stdoutu stderru retcodeu all( u u (# R R R R t
isinstancet boolt listt tupleR R' t logt warningR R R t PATHt ost environR& R R
t vtt Terminalt Falset has_unread_datat recvt timet sleept KeyboardInterruptt
exitstatust TerminalExceptiont tracebackt
format_exct errort terminate( R! t cmdR R R# t no_startR+ R, R- R R t use_vtt keep_envt valid_outputt cmd_funct to_keept full_cmdt xR t rett stdoutt stderrt proct cstdoutt cstderrt trace( ( sC /usr/lib/python2.7/site-packages/salt/modules/container_resource.pyt runu s�
;"
;
;
c @ s# t d j � � � � � f d � } i � i t d 6} � d' k r[ � | d <� � d <n � � f d � }
|
| � } | d k r� t d j | � � � n t | � } t j j | � \ }
} t j j | � s� t d � � nK t j j
| � s t d
j | � � � n! t j j | � s-t d � � n t j j | � sNt d � � n | | d
j t j
| � � | � d d k r�t j j | | � } n� t j j | � \ } } | | d
j t j
| � � | � d d k rz| r\| | d j t j
| � � | � } | d d k rwd j | | � } | d rJ| d j | d � 7} n t | � � qwqzt d j | � | � � � n | r�| | d j t j
| � � | � d d k r�t d j | � � � n t d | d � } | t | | | � k rt j d j | | | � � t St j d j | � | | � � | d k r�d } � rh| d j t j
� � � 7} n d j | t | | | � } nm | d k r�t d j � � | � } d j | t | � t | � } n' | d! k r�d" j | | t | � } n t d# | d$ t d% d& �| t | | | � k S(( u]
Common logic for copying files to containers
path
path to the container parent (for LXC only)
default: /var/lib/lxc (system default)
CLI Example:
.. code-block:: bash
salt myminion container_resource.copy_to mycontainer /local/file/path /container/file/path container_type=docker exec_driver=nsenter
u {0}.statec @ sN t j | � } � d k r4 d | k r4 � | d <n t d j � � | | � S( Nu lxcu pathu {0}.run_all( u lxc( t copyt deepcopyR& R ( R
t akwargs( R R ( sC /usr/lib/python2.7/site-packages/salt/modules/container_resource.pyt run_all& s
u ignore_retcodeu lxcu pathc @ s! � r � | � � S� | � Sd S( N( ( R! ( t statet state_kwargs( sC /usr/lib/python2.7/site-packages/salt/modules/container_resource.pyt _state3 s
u runningu Container '{0}' is not runningu Source path must be absoluteu Source file {0} does not existu Source must be a regular fileu! Destination path must be absoluteu test -d {0}u retcodei u mkdir -p {0}u= Unable to create destination directory {0} in container '{1}'u stderru : {0}u3 Directory {0} does not exist on {1} container '{2}'u test -e {0}uG Destination path {0} already exists. Use overwrite=True to overwrite itu file.get_sumu md5u0 {0} and {1}:{2} are the same file, skipping copyu) Copying {0} to {1} container '{2}' as {3}u
lxc-attachu -P {0}u= cat "{0}" | {4} --clear-env --set-var {1} -n {2} -- tee "{3}"u nsenteru {0}.pidu$ cat "{0}" | {1} env -i {2} tee "{3}"u docker-execu3 cat "{0}" | docker exec -i {1} env -i {2} tee "{3}"u cmd.runR, R- u quiet( u lxc( R& R R R R* R= R R t isabsR t existst isfileR R R R$ R: t debugR< R ( R! R( t destR R R t overwritet makedirsRa t
cmd_kwargsRd t c_statet
local_filet
source_dirt source_namet dest_dirt dest_namet resultRK t
source_md5t lxcattacht copy_cmdR ( ( R R Rb Rc sC /usr/lib/python2.7/site-packages/salt/modules/container_resource.pyt copy_to
s�
( t __doc__t
__future__R R R R R^ t loggingR= R RD RI t salt.utils.argsR t salt.utils.patht
salt.utils.vtt salt.exceptionsR R t getLoggert __name__R: R< R R R$ R* R RA R R] Rw ( ( ( sC /usr/lib/python2.7/site-packages/salt/modules/container_resource.pyt <module> sH
�
Mini Shell