Current File : //lib/python2.7/site-packages/salt/cli/caller.pyc
�
���^c @@ sj 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 Z
d d l Z
d d l Z
d d l Z
d d l Z
d d l j j Z d d l Z
d d l Z
d d l Z
d d l Z
d d l m Z d d l m Z d d l m Z d d l! m" Z" yX d d l# m$ Z$ m% Z% d d l& m' Z' d d
l( m) Z) m* Z* e � r�d d l+ Z+ n Wn e, k
r�n Xd d l- m. Z. d d l/ m0 Z0 m1 Z1 m2 Z2 m3 Z3 e j4 e5 � Z6 d
e7 f d � � YZ8 d e7 f d � � YZ9 d e9 f d � � YZ: d � Z; d e9 f d � � YZ< d S( u]
The caller module is used as a front-end to manage direct calls to the salt
minion modules.
i ( t absolute_importt print_functiont unicode_literalsN( t daemons( t
LOG_LEVELS( t
is_windows( t MultiprocessingProcess( t raetingt nacling( t LaneStack( t
RemoteYardt Yard( t six( t SaltClientErrort CommandNotFoundErrort CommandExecutionErrort SaltInvocationErrort Callerc B@ s e Z d Z e d � � Z RS( uK
Factory class to create salt-call callers for different transport
c K@ s� d } d | k r | d } n9 d | j d i � j d i � k rX | d d d } n | d k rq t | | � S| d k r� t | | � St d � � d S(
Nu zeromqu transportu pillaru masteru tcpu detectu raetu, Callers are only defined for ZeroMQ and raet( u zeromqu tcpu detect( t gett ZeroMQCallert
RAETCallert Exception( t optst kwargst ttype( ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyt factoryC s
$
( t __name__t
__module__t __doc__t staticmethodR ( ( ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyR ? s t
BaseCallerc B@ s; e Z d Z d � Z d � Z d � Z d � Z d � Z RS( u*
Base class for caller transports
c C@ s� | | _ t | j d <t j j | j � | _ yC | j j d � r[ t j j | � | _ n t j j | � | _ Wn( t
k
r� } t t j
| � � � n Xd S( u+
Pass in command line opts
u calleru proxyidN( R t Truet saltt payloadt Serialt serialR t miniont SProxyMiniont SMinionR
t
SystemExitR t text_type( t selfR t exc( ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyt __init__\ s
c C@ s� i } xK t j | j j � D]4 \ } } | | k r | j rP | j | | <qP q q WxX t | � D]J } | j | j j d d � � ra t j
j j d j
| | | � � qa qa Wd S( uT
Pick up the documentation for all of the modules and print it out.
u funu u {0}:
{1}
N( R t iteritemsR$ t functionsR t sortedt
startswithR R R t utilst stringutilst print_clit format( R) t docst namet func( ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyt
print_docsn s " c C@ sN | j j j d � p' t j j | j � } t j j i | d 6d | j � d S( u&
Print out the grains
u grainsu localN( R$ R R R t loadert grainst outputt display_output( R) R9 ( ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyt print_grains{ s *c
C@ si | j j d t � } y.t j j j | � } z | j � } Wd t j j j | d | j j d d � d t �X| j d d � } | j d r� | } d } n | j d
i � } t j
j i | d 6d | d
| j d | j d d � �| j j d t � rt j
| d � n2 | d t j j j k rBt j
t j j j � n Wn t k
rd} t | � � n Xd S( u-
Execute the salt call logic
u profiling_enabledNt
stats_pathu profiling_pathu
/tmp/statst stopu outu nestedu print_metadatau returnu localt outR t _retcodeu retcodei u retcode_passthrough( R R t FalseR R0 t profilet activate_profilet callt output_profileR R: R; t syst exitt defaultst exitcodest EX_OKt
EX_GENERICR R' ( R) t profiling_enabledt prt retR? t print_rett err( ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyt run� s2
c @ s& i � | j d } t j j j | j � � d <t j j t j j | j d � � d � } | | j j
k rM| j j
d d j | � � } | r� | j j
j | � | | <d � d <| � d <� St
j j | j j
j | � � | j d � d
} | | j j k r-t
j j d j | j j | � � n t
j j d � t
j d
� n | j j d � } | d k r�t j j j | � } n y�i | d 6t j � d 6� d d 6d d 6} | d k r�| | d <n t j j | j j
| t j j j | j d d | j j d g � �d | �\ } } y> t j j j | d � � } | j | j j | � � Wd QXWn: t k
rrn* t k
r�t
j j d j | � � n X| j j
| }
i | d 6| d 6} | j | � t! | j d g � pt j j j | j j d d � � } | j j d d � r>t j j j | j d � | d <n t" | t# j$ � r\| g } n y� x� | D]| }
d j |
� } | | j j% k r�t& d j |
� � � n | j j% | | j | |
| | � � d <� d d k rfPqfqfWWn� t' k
r�} t
j j d j | � � t j j( j) |
j* � t+ j | j d j, � t- j. � } | t- j/ k rxt0 j1 � } t
j j | � n t
j t j2 j3 j4 � n Xy# t
j5 |
j6 j7 j d d
� } Wn t8 k
r�t j2 j3 j4 } n X| d
k r9y t9 � f d! � d2 D� � } Wn t: k
r t; } n X| s9t j2 j3 j4 } q9n | � d <Wn� t< k
r�} d$ } t+ j | j d j, � t- j. � } | t- j/ k r�t
j j t0 j1 � � n t
j j | j | | � � t
j t j2 j3 j4 � nK t= k
r%} d% } t
j j | j | | � � t
j t j2 j3 j4 � n Xy t j> | � Wn t t? f k
rPn Xt@ | j j
| d& � r�| j j
| jA } t" | t# j$ � r�| � d <q�n | j d' p�| j j d( tB � d' k p�| j j d) � d* k } | j j d d+ � j d, � } | s
| rR| j d- � d- <| � d <| j d � d. <| d k rR| � d <qRn xY | D]Q } | skqYn y+ t; � d# <| j jC d/ j | � � � WqYt: k
r�qYXqYW| s�y' � jD � } d0 | d <| jE | � Wq"t: k
r�q"Xn0 | j d1 r"t j j jF | j � d � � n � S(3 u!
Call the module
u funu jidu cachediru sys.docu {0}*u nestedu outu returnu .i u Possible reasons: {0}
u
i����u metadatau pidu salt-callu tgtu argt no_parseu no_parset datau w+bNuM Cannot write to process directory. Do you have permissions to write to {0} ?
u module_executorsu
[direct_call]u
executor_optsu {0}.executeu Executor '{0}' is not availableu(
Passed invalid arguments: {0}.
Usage:
u log_levelu retcodec 3@ s% | ] } � d j | t � Vq d S( u returnN( R R ( t .0t x( RN ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pys <genexpr> s u resultu successu Error running '{0}': {1}
u* Command required for '{0}' not found: {1}
u
__outputter__u localu file_clientu master_typeu disableu u ,u idu fun_argsu {0}.returneru requ
cache_jobs( u resultu success(G R R R0 t jidt gen_jidt ost patht joinR$ t get_proc_dirR- R3 t missing_fun_stringRF t stderrt writet splitt function_errorsRG R t Nonet argst yamlify_argt getpidt load_args_and_kwargst parse_inputt filest fopenR# t dumpst NameErrort IOErrort updatet getattrt
isinstanceR t string_typest executorsR t TypeErrorR1 R2 R R t lowert loggingt ERRORt DEBUGt tracebackt
format_excRH RI RK t modulesR t __context__t AttributeErrort allR R R R t removet OSErrort hasattrt
__outputter__RA t returnerst copyt
return_pubt
cache_jobs( R) t funt proc_fnR4 t mod_namet metadatat sdataRb R t fp_R6 RS Rp R5 t fnameR* t active_levelt tracet retcodet func_resultt msgt oputt is_localR� t returnert mret( ( RN s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyRD � s
&
#
)
!
#( R R R R+ R7 R< RQ RD ( ( ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyR X s
!R c B@ s e Z d Z d � Z d � Z RS( uT
Object to wrap the calling of local salt modules for the salt-call command
c C@ s t t | � j | � d S( u2
Pass in the command line options
N( t superR R+ ( R) R ( ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyR+ H s c C@ s� t j j j j | j d d �} i d d 6| j d d 6} x' t j | � D] \ } } | | | <qL Wz | j | � Wd | j � Xd S( u2
Return the data up to the master
t usageu salt_callu _returnu cmdu idN(
R t transportt clientt
ReqChannelR R R R, t sendt close( R) RN t channelt loadt keyt value( ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyR� N s ! ( R R R R+ R� ( ( ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyR D s c C@ s t j � } | j d | � d S( u�
Set up the minion caller. Should be run in its own process.
This function is intentionally left out of RAETCaller. This will avoid
needing to pickle the RAETCaller object on Windows.
t cleanup_protectedsN( R t MinionRD ( R� R$ ( ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyt raet_minion_run\ s R c B@ s2 e Z d Z d � Z d � Z d � Z d � Z RS( u�
Object to wrap the calling of local salt modules for the salt-call command
when transport is raet
There are two operation modes.
1) Use a preexisting minion
2) Set up a special caller minion if no preexisting minion
The special caller minion is a subset whose only function is to perform
Salt-calls with raet as the transport
The essentials:
A RoadStack whose local estate name is of the form "role_kind" where:
role is the minion id opts['id']
kind is opts['__role'] which should be 'caller' APPL_KIND_NAMES
The RoadStack if for communication to/from a master
A LaneStack with manor yard so that RaetChannels created by the func Jobbers
can communicate through this manor yard then through the
RoadStack to/from a master
A Router to route between the stacks (Road and Lane)
These are all managed via a FloScript named caller.flo
c C@ s� d | _ | d s� | j | � | _ | j t j _ | j d � t j t j
j k r� t d t
d i | j j g d 6� | _ | j j � | j | � q� n t t | � j | � d S( u2
Pass in the command line options
u localu __rolet targetR u cleanup_protectedsN( Ra t processt _setup_caller_stackt stackR R� t jobber_stackR t kindst APPL_KIND_NAMESt applKindst callerR R� t hat startt _wait_callerR� R R+ ( R) R ( ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyR+ s
c
C@ sK y%| j � } | j d s; | j j j � d t j _ n | j d rQ | } n | j d i � } | j
r| | j
j � n t j j
i | d 6d | j d d � d | j d | j d t j j j � �| j j d
t � r� t j | d � n2 | d t j j j k r$t j t j j j � n Wn t k
rF} t | � � n Xd S( u-
Execute the salt call logic
u localu print_metadatau returnR? u outu nestedR R@ u retcodeu retcode_passthroughN( RD R R� t serverR� Ra R R� R� R R� t terminateR: R; RH RI RJ RA RF RG RK R R' ( R) RN RO RP ( ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyRQ � s*
c
C@ s� | j d � } | s; d } t j | d � t | � � n | j d � } | t j k r� d j | � } t j | d � t | � � n | t j t j j t j t j j
g k r� d j | | � } n, d j | � } t j | d � t | � � | d } d t j d
d � } t
d | d
| d | � } t j j j | _ | j t d | d d d
| d | � � t j d | j � | S( u�
Setup and return the LaneStack and Yard used by by channel when global
not already setup such as in salt-call to communicate to-from the minion
u idu+ Missing role required to setup RAETChannel.u
u __roleu1 Invalid application kind = '{0}' for RAETChannel.u {0}_{1}u3 Unsupported application kind '{0}' for RAETChannel.u sock_diru callert sizei R5 t lanenamet sockdirpathR� u manort dirpathu Created Caller Jobber Stack %s
( R t logt errort
ValueErrorR� t
APPL_KINDSR3 R� R� R$ R� R t uuidR R t PackKindt packR� t Pkt addRemoteR
t debugR5 ( R) R t rolet emsgt kindR� R� t stacknameR� ( ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyR� � s8
c C@ s d } | d } | j d � } | sK d } t j | d � t | � � n | j d � } | t j k r� d j | � } t j | d � t | � � n | t j t j j t j t j j
g k r� d j | | � } n, d j | � } t j | d � t | � � t j | | | � \ } } t
� r�t } x� | s�yH t j | t j t j Bt j d
t j d
d
� }
t j |
� t } Wq2t j k
r�t j d � q2Xq2WnL xI t j j | � o�t j j | � o�t j j | � s�t j d � q�Wt j d � d
S( u<
Returns when RAET Minion Yard is available
u manoru sock_diru idu/ Missing role required to setup RAET SaltCaller.u
u __roleu5 Invalid application kind = '{0}' for RAET SaltCaller.u {0}_{1}u7 Unsupported application kind '{0}' for RAET SaltCaller.i g�������?g �?N( R R� R� R� R� R� R3 R� R� R$ R� R t computeHaR RA t win32filet
CreateFilet
GENERIC_WRITEt GENERIC_READt FILE_SHARE_READRa t
OPEN_EXISTINGt CloseHandleR t timet sleepRX RY t existst isfilet isdir( R) R t yardnameR� R� R� R� R� R� R� t f( ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyR� � sP
( R R R R+ RQ R� R� ( ( ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyR f s
((= R t
__future__R R R RX RF R� Rs Rv R t salt.loadert salt.miniont salt.outputt salt.payloadt salt.transportt salt.transport.clientt salt.utils.argst salt.utils.filest salt.utils.jidt salt.utils.kindsR0 R� t salt.utils.miniont salt.utils.profilet salt.utils.stringutilst salt.defaults.exitcodest salt.cliR t salt.logR t salt.utils.platformR t salt.utils.processR t raetR R t raet.lane.stackingR t raet.lane.yardingR
R R� t ImportErrort salt.extR t salt.exceptionsR
R R R t getLoggerR R� t objectR R R R� R ( ( ( s3 /usr/lib/python2.7/site-packages/salt/cli/caller.pyt <module> sR
"�
Mini Shell