File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/extend.cpython-310.pyc
Back
o ;j0# � @ s d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z ddl m Z ddl Z ddlZ ddlmZ e�e�ZzddlZdZW n eyW Z zdZW Y dZ[ndZ[ww d Zd d � Zd d � Zdd� Zdd� Z dd� Z dd� Z dd� Z ddd�Z edkr�e � dS dS ) a* 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> � N)� OrderedDict)�date)�Template)� deserializeTFz template.ymlc C sP t jj�| d��}t|�}||�dd�|f}W d � |S 1 s!w Y |S )a- 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`` �r� description� N)�salt�utils�files�fopenr �get)�pathZ option_keyZ template_f�template�info� r �E/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/extend.py� _get_template) s ��r c C sx g }t �d| � t�| �D ],}tj�| |�}tj�|�r9tj�|t�}tj�|�r2|� t ||�� q t �d|t� q |S )z� 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') zListing contents of %sz Directory does not contain %s %s) �log�debug�os�listdirr �join�isdir�TEMPLATE_FILE_NAME�isfile�appendr )�src� templates�item�sZ template_pathr r r �_fetch_templates< s ��r! c C s� t �| �D ]=}t j�| |�}t j�||�}t j�|�r5t�d||� t j�|�r.t||� qt � ||� qt�d||� t � ||� qdS )z� 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`` �Copying folder %s to %s�Copying file %s to %sN) r r r r r r r �exists� _mergetree�shutil�copytree�copy2)r �dstr r �dr r r r% W s �r% c C s4 t �| �D ]�}t j�| |�}t j�||�}t j�|�r;t�d||� t j�|�r/t|||� qt � |� t|||� q|t kr�t |�� |�}t�d||� t jj�|d��}t jj�|�� �}t |�� |�}W d � n1 spw Y t jj�|d��} | �t jj�|�� W d � n1 s�w Y qdS )ak 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`` r" r# r N�w)r r r r r r r r$ �_mergetreejinja�mkdirr r �renderr r r r Z stringutilsZ to_unicode�read�writeZto_str) r r) �contextr r r* Z source_fileZ src_contentsZ dest_contentsZ dest_filer r r r, o s* � ���r, c C s t j| |d�S )a 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 )�default)�click�prompt)�var_nameZ default_valuer r r �_prompt_user_variable� s r6 c C s~ t dd� t|d�D ��}|�� }d}dd� |�� D �}d�d| � d �d�|�d �d �|��f�}tj|t�|�|d �}|| S ) a� 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* � | ]\}}|d dkr|� |fV qdS )r �testNr )�.0�i�valuer r r � <genexpr>� s � �z!_prompt_choice.<locals>.<genexpr>� �1c S s4 g | ]}|d � d|d d � d|d d � ��qS )r z - r< r )r8 �cr r r � <listcomp>� s 4 z"_prompt_choice.<locals>.<listcomp>� zSelect �:zChoose from {}z, )�typer2 ) r � enumerate�keys�itemsr �formatr3 r4 ZChoice)r5 �optionsZ choice_map�choicesr2 Z choice_linesr4 Z user_choicer r r �_prompt_choice� s � ��rI c C s t | ||� dS )aP 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, )� template_dirZ output_dirr1 r r r �apply_template� s rK c s� t s td� t�d� |du rd}ttj�|d��}� du r*td� d}t||�}n � t t |� �d vr=td � t�d� � fd d �|D �d }|d }|d } |du r]td � t dd�}|du rft dd�}d|� �} |} t j j�� j| |t�� �d�t�� �d�d�} i } | �di ��� D ]\}}t|�dd���| �}t |d |�}|| |< q�| �� }|�| � |�| � |du r�t�� }t | ||� |s�|}nt ||� |}t � d|� |S )a� 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`` z0click is not installed, please install using pipr< N�.r z?Choose which type of extension you are developing for SaltStackzExtension typer z!Module extension option not validc s g | ] }|d � kr|�qS )r r )r8 �m�� extensionr r r? s zrun.<locals>.<listcomp>� z3Enter the short name for the module (e.g. mymodule)z Module namer z Short description of the modulez templates/z%Y-%m-%dz%Y)�version� module_nameZshort_descriptionZ release_date�yearZ questionsr2 ZquestionzNew module stored in %s) � HAS_CLICK�print�sys�exitr! r r r rI �list�zipr6 r rQ ZSaltStackVersionZ next_release�namer �today�strftimer rE r r. �copy�update�tempfileZmkdtemprK r% r r )rO rZ r Zsalt_dir�mergeZtemp_dirZMODULE_OPTIONSZextension_typeZchosen_extensionZextension_contextrJ rR Z param_dictZadditional_context�key�valr2 Z prompt_varr1 r r rN r �run� s\ � rc �__main__)NNNNFN)!�__doc__�loggingr r& rV r_ � collectionsr �datetimer Zjinja2r Zsalt.utils.filesr Z salt.versionZsalt.serializers.yamlr � getLogger�__name__r r3 rT � ImportErrorZier r r! r% r, r6 rI rK rc r r r r �<module> sL ��#! �b �
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings