Current File : //lib/python3.9/site-packages/ansible/plugins/action/__pycache__/include_vars.cpython-39.opt-1.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
Mini Shell