Current File : //lib/python2.7/site-packages/salt/utils/extend.pyc
�
���^c @@ s� d Z d d l m Z m Z m Z d d l 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 m
Z
d d l m Z d d l m Z d d l m Z d d l Z d d l Z e j e � Z y d d l Z e Z Wn e k
rZ e Z n Xd Z d
� Z! d � Z" d � Z# d
� Z$ d � Z% d � Z& d � Z' d d d d e d d � Z) e d k r}e) � n d S( u*
SaltStack Extend
~~~~~~~~~~~~~~~~
A templating tool for extending SaltStack.
Takes a template directory and merges it into a SaltStack source code
directory. This tool uses Jinja2 for templating.
This tool is accessed using `salt-extend`
:codeauthor: Anthony Shaw <anthonyshaw@apache.org>
i ( t absolute_importt unicode_literalst print_function( t dateN( t Template( t deserialize( t zip( t OrderedDictu template.ymlc C@ sL t j j j | d � �. } t | � } | | j d d � | f } Wd QX| S( u-
Get the contents of a template file and provide it as a module type
:param path: path to the template.yml file
:type path: ``str``
:param option_key: The unique key of this template
:type option_key: ``str``
:returns: Details about the template
:rtype: ``tuple``
u ru descriptionu N( t saltt utilst filest fopenR t get( t patht
option_keyt
template_ft templatet info( ( s5 /usr/lib/python2.7/site-packages/salt/utils/extend.pyt
_get_template- s
!c C@ s� g } t j d | � x� t j | � D]� } t j j | | � } t j j | � r& t j j | t � } t j j | � r� | j t
| | � � q� t j d | t � q& q& W| S( u�
Fetch all of the templates in the src directory
:param src: The source path
:type src: ``str``
:rtype: ``list`` of ``tuple``
:returns: ``list`` of ('key', 'description')
u Listing contents of %su Directory does not contain %s %s( t logt debugt ost listdirR
t joint isdirt TEMPLATE_FILE_NAMEt isfilet appendR ( t srct templatest itemt st
template_path( ( s5 /usr/lib/python2.7/site-packages/salt/utils/extend.pyt _fetch_templates@ s
c C@ s� x� t j | � D]� } t j j | | � } t j j | | � } t j j | � r� t j d | | � t j j | � r� t | | � q� t j
| | � q t j d | | � t j | | � q Wd S( u�
Akin to shutils.copytree but over existing directories, does a recursive merge copy.
:param src: The source path
:type src: ``str``
:param dst: The destination path
:type dst: ``str``
u Copying folder %s to %su Copying file %s to %sN( R R R
R R R R t existst
_mergetreet shutilt copytreet copy2( R t dstR R t d( ( s5 /usr/lib/python2.7/site-packages/salt/utils/extend.pyR# X s
c
C@ sw xpt j | � D]_} t j j | | � } t j j | | � } t j j | � r� t j d | | � t j j | � r� t | | | � qot j | � t | | | � q | t
k r t | � j | � } t j d | | � t
j j j | d � �7 } t
j j j | j � � } t | � j | � } Wd QXt
j j j | d � �# } | j t
j j j | � � Wd QXq q Wd S( uk
Merge directory A to directory B, apply Jinja2 templating to both
the file/folder names AND to the contents of the files
:param src: The source path
:type src: ``str``
:param dst: The destination path
:type dst: ``str``
:param context: The dictionary to inject into the Jinja template as context
:type context: ``dict``
u Copying folder %s to %su Copying file %s to %su rNu w( R R R
R R R R R" t _mergetreejinjat mkdirR R t renderR R R
R t stringutilst
to_unicodet readt writet to_str(
R R' t contextR R R( t source_filet src_contentst
dest_contentst dest_file( ( s5 /usr/lib/python2.7/site-packages/salt/utils/extend.pyR) p s"
c C@ s t j | d | �S( u
Prompt the user to enter the value of a variable
:param var_name: The question to ask the user
:type var_name: ``str``
:param default_value: The default value
:type default_value: ``str``
:rtype: ``str``
:returns: the value from the user
t default( t clickt prompt( t var_namet
default_value( ( s5 /usr/lib/python2.7/site-packages/salt/utils/extend.pyt _prompt_user_variable� s
c C@ s� t d � t | d � D� � } | j � } d } g | j � D]/ } d j | d | d d | d d � ^ q> } d j d j | � d j | � d j d j | � � f � } t j | d
t j | � d | �} | | S( u�
Prompt the user to choose between a list of options, index each one by adding an enumerator
based on https://github.com/audreyr/cookiecutter/blob/master/cookiecutter/prompt.py#L51
:param var_name: The question to ask the user
:type var_name: ``str``
:param options: A list of options
:type options: ``list`` of ``tupple``
:rtype: ``tuple``
:returns: The selected user
c s@ s: | ]0 \ } } | d d k r d j | � | f Vq d S( i u testu {0}N( t format( t .0t it value( ( s5 /usr/lib/python2.7/site-packages/salt/utils/extend.pys <genexpr>� s i u 1u {0} - {1} - {2}i u
u Select {0}:u Choose from {0}u , t typeR6 ( R t enumeratet keyst itemsR< R R7 R8 t Choice( R9 t optionst
choice_mapt choicesR6 t ct choice_linesR8 t user_choice( ( s5 /usr/lib/python2.7/site-packages/salt/utils/extend.pyt _prompt_choice� s B c C@ s t | | | � d S( uP
Apply the template from the template directory to the output
using the supplied context dict.
:param src: The source path
:type src: ``str``
:param dst: The destination path
:type dst: ``str``
:param context: The dictionary to inject into the Jinja template as context
:type context: ``dict``
N( R) ( t template_dirt
output_dirR1 ( ( s5 /usr/lib/python2.7/site-packages/salt/utils/extend.pyt apply_template� s c C@ s� t s t d � t j d � n | d k r5 d } n t t j j | d � � } | d k r~ t d � d } t | | � } nc | t
t | � � d k r� t d � t j d � n g | D] } | d | k r� | ^ q� d } | d } | d }
| d k r t d
� t d d � } n | d k r;t d
d � } n d j
| � } | } i t j j j � j d 6| d 6| d 6t j � j d � d 6t j � j d � d 6}
i } xd |
j d i � j � D]J \ } } t | j d d � � j |
� } t | d | � } | | | <q�W|
j � } | j |
� | j | � | d k rTt j � } n t | | | � | ss| } n t | | � | } t j! d | � | S( u�
A template factory for extending the salt ecosystem
:param extension: The extension type, e.g. 'module', 'state', if omitted, user will be prompted
:type extension: ``str``
:param name: Python-friendly name for the module, if omitted, user will be prompted
:type name: ``str``
:param description: A description of the extension, if omitted, user will be prompted
:type description: ``str``
:param salt_dir: The targeted Salt source directory
:type salt_dir: ``str``
:param merge: Merge with salt directory, `False` to keep separate, `True` to merge trees.
:type merge: ``bool``
:param temp_dir: The directory for generated code, if omitted, system temp will be used
:type temp_dir: ``str``
u0 click is not installed, please install using pipi u .u templatesu? Choose which type of extension you are developing for SaltStacku Extension typei u! Module extension option not validi u3 Enter the short name for the module (e.g. mymodule)u Module nameu u Short description of the moduleu
templates/{0}u versionu module_nameu short_descriptionu %Y-%m-%du release_dateu %Yu yearu questionsu defaultu questionu New module stored in %sN(" t HAS_CLICKt printt syst exitt NoneR! R R
R RK t listR R; R< R t versiont SaltStackVersiont next_releaset nameR t todayt strftimeR RC R R+ t copyt updatet tempfilet mkdtempRN R# R R ( t extensionRX t descriptiont salt_dirt merget temp_dirt MODULE_OPTIONSt extension_typet chosen_extensiont mt extension_contextRL t module_namet
param_dictt additional_contextt keyt valR6 t
prompt_varR1 R
( ( s5 /usr/lib/python2.7/site-packages/salt/utils/extend.pyt run� sb
-
%!
u __main__(* t __doc__t
__future__R R R t datetimeR t loggingR] R RQ R$ t jinja2R t salt.serializers.yamlR t salt.ext.six.movesR t salt.utils.odictR t salt.utils.filesR t salt.versiont getLoggert __name__R R7 t TrueRO t ImportErrort iet FalseR R R! R# R) R; RK RN RS Ro ( ( ( s5 /usr/lib/python2.7/site-packages/salt/utils/extend.pyt <module> s<
# ! ^
Mini Shell