관리-도구
편집 파일: include_vars.cpython-39.pyc
a �)g�, � @ s� d dl mZmZmZ eZd dlmZmZ d dl Z d dl m Z d dl mZ d dlmZ d dlmZmZ d dlmZ d dlmZ G d d � d e�ZdS ) � )�absolute_import�division�print_function)�path�walkN)� AnsibleError)� string_types)� to_native�to_text)� ActionBase)� combine_varsc s� e Zd ZdZg d�Zg d�ZddgZddgZdd � Zd d � Z d � fd d� Z dd� Z dd� Z dd� Z dd� Zdd� Zd dd�Zdd � Z� ZS ) � ActionModuleF)ZyamlZymlZjson)�dir�depth�files_matching� ignore_files� extensions�ignore_unknown_extensions�file� _raw_params�name�hash_behaviourc C sx | j s d| _ | jr(t�d�| j��| _nd | _| js<t� | _t| jt �rV| j� � | _n t| jt �rtdd�| j�d�S d S )Nr z{0}Tz{0} must be a list)�failed�message) r r �re�compile�format�matcherr �list� isinstancer �split�dict��self� r$ �G/usr/lib/python3.9/site-packages/ansible/plugins/action/include_vars.py�_set_dir_defaults s �z ActionModule._set_dir_defaultsc C s | j j�dd�| _| j j�dd�| _| j j�dd�| _| j j�dd�| _| jsx| jsx| j j�d�| _| jrx| j�d�| _| j j�dd�| _| j j�d d�| _ | j j�d d �| _ | j j�d d�| _ | j j�d | j �| _ t| j t�r�t| j �| _ t| j t��std��dS )z@ Set instance variables based on the arguments that were passed r Nr r r r � r r r Fr r z7Invalid type for "extensions" option, it must be a list)�_task�args�getr �return_results_as_name� source_dir� source_file�rstripr r r r �VALID_FILE_EXTENSIONS�valid_extensionsr r r r r" r$ r$ r% � _set_args0 s" zActionModule._set_argsNc sZ ~|du rt � }d| _g | _d}d}| jjD ]J}|| jv rD|d7 }q,|| jv rX|d7 }q,|| jv rdq,td� t |����q,|r�|r�td��| � � t � }d}| j �r2| � � | �� t�| j �s�d}d� t | j ��}n\t�| j �s�d}d � t | j ��}n:| �� D ]0\} } | �| | �\}}} |�r$ �q�|�| � q�njz6| �d | j�| _| �| j�\}}} |�sf|�| � W n2 t�y� } zd}t | �}W Y d} ~ n d} ~ 0 0 | j�r�t � } || | j< | }tt| �j|d �}|�r�||d < ||d < nV| j du�r8| j t j k�r8| j dk}|� � D ](\}}|� |d�}t!|||d�||< �q| j|d< ||d< | j |d< |S )z6 Load yml files recursively from a directory. NTr � z){0} is not a valid option in include_varszGYou are mixing file only and dir only arguments, these are incompatibleFz {0} directory does not existz{0} is not a directory�vars)� task_varsr r �merge)r5 Zansible_included_var_filesZ ansible_factsZ_ansible_no_log)"r! � show_content�included_filesr( r) �VALID_DIR_ARGUMENTS�VALID_FILE_ARGUMENTS� VALID_ALLr r r r1 r, r& � _set_root_dirr �exists�isdir�_traverse_dir_depth�_load_files_in_dir�updateZ _find_needler- � _load_filesr+ �superr �runr �CZDEFAULT_HASH_BEHAVIOUR�itemsr* r )r# �tmpr4 �dirs�files�arg�resultsr �err_msg�root_dir� filenamesZupdated_results�eZscope�resultZ merge_hashes�key�value� old_value�� __class__r$ r% rC H sx � zActionModule.runc C s� | j jr`| j�d�d dkrBt�| j jj| j�}t�|�r^|| _q�t�| j jjd| j�}|| _n<t| j j d�r�d�| j j j �d�d d� �}t�|| j�| _d S )N�/r r3 � _data_source���) r( �_roler, r r �join� _role_pathr< �hasattrZ_dsrV )r# Z path_to_use� current_dirr$ r$ r% r; � s � �� �zActionModule._set_root_dirc C s" | j �dt|j�t|�f � d S )Nz#Issue with walking through "%s": %s)Z_displayZvvvr �filename)r# �errorr$ r$ r% � _log_walk� s zActionModule._log_walkc c sp d}t t| j| jd��}|jdd� d� |D ]>\}}}|d7 }|| jksR| jdkrf|�� ||fV q, qlq,dS )z� Recursively iterate over a directory and sort the files in alphabetical order. Do not iterate pass the set depth. The default depth is unlimited. r )�onerrorc S s | d S )Nr r$ )�xr$ r$ r% �<lambda>� � z2ActionModule._traverse_dir_depth.<locals>.<lambda>)rP r2 N)r r r, r_ �sortr )r# Z current_depthZ sorted_walkZ current_rootr\ Z current_filesr$ r$ r% r> � s z ActionModule._traverse_dir_depthc C sT | j D ]H}z t�d�|�|�r&W dS W q tyL d�|�}t|��Y q0 qdS )z� Return True if a file matches the list of ignore_files. Args: filename (str): The filename that is being matched against. Returns: Boolean z{0}$Tz Invalid regular expression: {0}F)r r �searchr � Exceptionr )r# r] Z file_typerK r$ r$ r% � _ignore_file� s zActionModule._ignore_filec C s0 t �|�}tt|�dko,|d dd� | jv �S )z� Verify if source file has a valid extension Args: source_file (str): The full path of source file or source file. Returns: Bool r2 rW N)r �splitext�bool�lenr0 )r# r- Zfile_extr$ r$ r% �_is_valid_file_ext� s z ActionModule._is_valid_file_extc C s� t � }d}d}|r:| �|�s:d}d�t|�d�| j��}nz| j�|�\}}t|dd�}| j |M _ | jj |||d�}|s�t � }t |t �s�d}d �t|��}n| j � |� |�|� |||fS ) z� Loads a file and converts the output into a valid Python dict. Args: filename (str): The source file. Returns: Tuple (bool, str, dict) F� Tz({0} does not have a valid extension: {1}z, Zsurrogate_or_strict)�errors)� file_namer6 z'{0} must be stored as a dictionary/hash)r! rk r r rY r0 Z_loaderZ_get_file_contentsr r6 �loadr r7 �appendr@ ) r# r] �validate_extensionsrJ r rK Zb_datar6 �datar$ r$ r% rA � s$ zActionModule._load_filesc C s� t � }d}d}|D ]�}d}| jjrHt�| jjj|�t�|dd�krHd}qt�||�}| jrj| j�|�sjd}|s|s| j r�t� |�r�| � |�s�| � |�r�| j |dd�\}}} |s�|�| � qt� |�r| � |�s| j |dd�\}}} |s|�| � q|||fS )a> Load the found yml files and update/overwrite the dictionary. Args: root_dir (str): The base directory of the list of files that is being passed. var_files: (list): List of files to iterate over and load into a dictionary. Returns: Tuple (bool, str, dict) Frl r3 zmain.ymlT)rq )r! r( rX r rY rZ r r re r r< rg rk rA r@ ) r# rL Z var_filesrJ r rK r] Z stop_iter�filepathZ loaded_datar$ r$ r% r? � s0 z ActionModule._load_files_in_dir)NN)F)�__name__� __module__� __qualname__ZTRANSFERS_FILESr/ r8 r9 r: r& r1 rC r; r_ r> rg rk rA r? � __classcell__r$ r$ rS r% r s R r )Z __future__r r r �typeZ __metaclass__�osr r r Zansible.constantsZ constantsrD Zansible.errorsr Zansible.module_utils.sixr Zansible.module_utils._textr r Zansible.plugins.actionr Zansible.utils.varsr r r$ r$ r$ r% �<module> s