File: //usr/lib/python3.9/site-packages/ansible/cli/__pycache__/inventory.cpython-39.pyc
a
�)g @ � @ s� d dl mZmZmZ eZd dlmZ d dlZd dl Z d dl
m Z d dl
m
Z
d dlmZ d dlmZmZ d dlmZmZmZ d d lmZ d d
lmZ d d l m Z m Z e� Z e g d ��Z!G d
d� de�Z"ddd�Z#e$dkr�e#� dS )� )�absolute_import�division�print_function)�CLIN)� constants)�context)�option_helpers)� AnsibleError�AnsibleOptionsError)�to_bytes� to_native�to_text)� combine_vars)�Display)� get_vars_from_inventory_sources�get_vars_from_path)Zansible_diff_modeZansible_config_fileZ
ansible_factsZ
ansible_forksZansible_inventory_sourcesZ
ansible_limitZansible_playbook_pythonZansible_run_tagsZansible_skip_tagsZansible_verbosityZansible_versionZ
inventory_dirZinventory_fileZinventory_hostnameZinventory_hostname_short�groupsZ group_namesZomitZ playbook_dirc s� e Zd ZdZdZddd�Z� fdd�Z� fdd �Z� fd
d �Z� fd d
�Z e
dd� �Z dd� Z dd� Z
dd� Ze
dd� �Ze
dd� �Ze
dd� �Ze
d)d d ��Zd*d d �Zd!d"� Zd#d$� Zd%d&� Zd'd(� Z� ZS )+� InventoryCLIzE used to display or dump the configured inventory as Ansible sees it zansible-inventoryzHThe name of a host to match in the inventory, relevant when using --listzAThe name of a group in the inventory, relevant when using --graph)�host�groupc s& t t| ��|� d | _d | _d | _d S �N)�superr �__init__�vm�loader� inventory)�self�args�� __class__� �9/usr/lib/python3.9/site-packages/ansible/cli/inventory.pyr 9 s zInventoryCLI.__init__c sP t t| �jddd� t�| j� t�| j� t�| j� t�| j� | jj ddt
j tj dd� | jj dt
j tj d � | jj d
d dd � | j�
d
d�}|j dddddd� |j ddd ddd� |j dddddd� | j�
|� | jj d d ddd d d� | jj d ddd!d"d� | jj d#ddd$d%d� | jj d&dtjd'd(d� | jj d)d d*d+d,� d S )-Nz#usage: %prog [options] [host|group]zWShow Ansible inventory information, by default it uses the inventory script JSON format)�usageZdescz-lz--limit�?)�help�action�nargsz --list-hosts)r$ r% r z
host|group)�metavarr& ZActionsz6One of following must be used on invocation, ONLY ONE!z--list�
store_trueF�listz0Output all hosts info, works as inventory script)r% �default�destr$ z--host�storer z4Output specific host info, works as inventory scriptz--graph�graphzJcreate inventory graph, if supplying pattern it must be a valid group namez-yz--yaml�yamlz<Use YAML format instead of default JSON, ignored for --graphz--toml�tomlz<Use TOML format instead of default JSON, ignored for --graphz--vars� show_varsz;Add vars to graph display, ignored unless used with --graphz--export�exportz�When doing an --list, represent in a way that is optimized for export,not as an accurate representation of how Ansible has processed itz--output� output_filezHWhen doing --list, send the inventory to a file instead of to the screen)r* r+ r$ )r r � init_parser�opt_helpZadd_inventory_options�parserZadd_vault_optionsZadd_basedir_optionsZadd_runtask_options� add_argument�argparse�SUPPRESSZUnrecognizedArgument�add_argument_group�CZINVENTORY_EXPORT)r � action_groupr r r! r3 @ sB
� � ���� �zInventoryCLI.init_parserc s� t t| ��|�}|jt_| �|� d}|j|j|jfD ]}|r6|d7 }q6|dkrZt d��n|dkrjt d��|j
rz|j
|_ nd|_ |S )Nr � zTNo action selected, at least one of --host, --graph or --list needs to be specified.z]Conflicting options used, only one of --host, --graph or --list can be used at the same time.�all) r r �post_process_args� verbosity�displayZvalidate_conflictsr) r r- r
r �pattern)r �options�used�optr r r! r> i s
z InventoryCLI.post_process_argsc
s� t t| ��� | �� \| _| _| _d }tjd rr| j� tjd �}t
|�dkrVt d��| j |d d�}| �
|�}nhtjd r�| �� }nTtjd r�| �d�}tjd r�| �|�}n tjd
r�| �|�}n
| �|�}| �
|�}|�r�tjd }|d u r�t�|� n�zBtt|�d �� }|�t|�� W d � n1 �s40 Y W nB ttf�y� } z$td
t|�t|�f ��W Y d }~n
d }~0 0 t� d� t� d� d S )Nr r< z5You must pass a single valid host to --host parameterr �r r- r) r= r. r/ r2 �wbz,Unable to write to destination file (%s): %s) r r �runZ
_play_prereqsr r r r �CLIARGS� get_hosts�lenr
�_get_host_variables�dump�inventory_graph�
_get_group�yaml_inventory�toml_inventory�json_inventoryr@ �openr �write�OSError�IOErrorr r �sys�exit)r �results�hosts�myvars�topZoutfile�f�er r r! rG � s<
2.
zInventoryCLI.runc C sD t jd r8dd l}ddlm} t|j| |ddd��}�nt jd r�ddlm} z || �}W nd t y� } zt
d | ��W Y d }~n>d }~0 t y� } z t
d
|j d ��W Y d }~n
d }~0 0 n~dd l
}dd lm} z|j| |dd ddd
�}W nN t �y> } z4|j| |dd ddd
�}t�dt|� � W Y d }~n
d }~0 0 |S )Nr. r )�
AnsibleDumperFT)ZDumperZdefault_flow_styleZ
allow_unicoder/ )�
toml_dumpszLThe source inventory contains a value that cannot be represented in TOML: %sa The source inventory contains a non-string key (%s) which cannot be represented in TOML. The specified key will need to be converted to a string. Be aware that if your playbooks expect this key to be non-string, your playbooks will need to be modified to support this change.)�AnsibleJSONEncoder� )�clsZ sort_keys�indentZpreprocess_unsafeZ ensure_asciiz?Could not sort JSON output due to issues while sorting keys: %s)r rH r. Zansible.parsing.yaml.dumperr^ r
rL Z ansible.plugins.inventory.tomlr_ � TypeErrorr �KeyErrorr �jsonZansible.parsing.ajsonr` �dumpsr@ Zwarningr )�stuffr. r^ rX r_ r] rf r` r r r! rL � s4
��� (zInventoryCLI.dumpc C sj |� � }t|t| j| jj|gd��}tjd rLt|t| jtjd |gd��}|j dkr`|j |d< | �
|�S )Nr= �basedirr< Zansible_group_priority) �get_varsr r r r �_sourcesr rH r �priority�_remove_internal)r r �resr r r! �_get_group_variables� s
z!InventoryCLI._get_group_variablesc C st t jd rX|�� }t|t| j| jj|gd��}t jd rjt|t| jt jd |gd��}n| j j|ddd�}| �
|�S )Nr1 r= ri F)r Zinclude_hostvarsZstage) r rH rj r r r r rk r r rm )r r �hostvarsr r r! rK � s
z InventoryCLI._get_host_variablesc C s | j j�|�}|S r )r r �get)r �gnamer r r r! rN � s zInventoryCLI._get_groupc C s t D ]}|| v r| |= q| S r )�
INTERNAL_VARS)rL Zinternalr r r! rm � s z InventoryCLI._remove_internalc C s$ dD ]}|| v r| | s| |= qd S )N)rY �vars�childrenr )rL �xr r r! �
_remove_empty� s zInventoryCLI._remove_emptyc C s8 g }t | �� �D ]"\}}|�t�d||f |�� q|S )Nz {%s = %s})�sorted�items�appendr � _graph_name)rL �depth�result�name�valr r r! �
_show_vars s zInventoryCLI._show_varsr c C s |rd| d| } | S )Nz |z--%sr )r~ r| r r r! r{ s zInventoryCLI._graph_namec C s� | � d|j |�g}|d }|jD ]}|�| �||�� q"|jdkr�|jD ]>}|�| � |j|�� tjd rJ|�| � | �
|�|d �� qJtjd r�|�| � | � |�|�� |S )Nz@%s:r< r= r0 ) r{ r~ � child_groups�extend� _graph_grouprY rz r rH r� rK ro )r r r| r} Zkidr r r r! r� s
zInventoryCLI._graph_groupc C s0 | � tjd �}|r$d�| �|��S td��d S )NrA �
z3Pattern must be valid group name when using --graph)rN r rH �joinr� r
)r Zstart_atr r r! rM ! s z InventoryCLI.inventory_graphc sb t � �� ��fdd�� � |�}di i|d< �j�� }|D ]$}��|�}|r8||d d |j< q8|S )Nc s� i }i || j < | j dkr2dd� | jD �|| j d< g || j d< | jD ]>}|| j d �|j � |j �vrF|�� |�� ��|j � qFtjd r���| �|| j d< �� || j � || j s�|| j = |S )Nr= c S s g | ]
}|j �qS r )r~ )�.0�hr r r! �
<listcomp>1 � zEInventoryCLI.json_inventory.<locals>.format_group.<locals>.<listcomp>rY ru r1 rt )
r~ rY r� rz �update�addr rH ro rw )r rX �subgroup�� format_group�seenr r r! r� - s
z1InventoryCLI.json_inventory.<locals>.format_grouprp Z_meta)�setr rI rK r~ )r r[ rX rY r Zhvarsr r� r! rQ ) s
zInventoryCLI.json_inventoryc s g �� ��fdd�� � |�S )Nc s� i }i || j < i || j d< | jD ]&}|j dkr"|| j d �� |�� q"i || j d< | j dkr�| jD ]>}i }|j �vr���|j � �j|d�}||| j d |j < qhtjd rΈ�| �}|r�||| j d< �� || j � |S )Nru r= rY rE r1 rt )
r~ r� r� rY rz rK r rH ro rw )r rX r� r� rZ �gvarsr� r r! r� Q s(
z1InventoryCLI.yaml_inventory.<locals>.format_groupr )r r[ r r� r! rO M s zInventoryCLI.yaml_inventoryc s< t � �ttdd� |jD ����� ���fdd�� � |�}|S )Nc s s | ]}|j d kr|jV qdS )� ungroupedN)r~ rY )r� �gr r r! � <genexpr>t r� z.InventoryCLI.toml_inventory.<locals>.<genexpr>c s, i }i || j < g || j d< | jD ]B}|j dkr6�s6q"| j dkrV|| j d �|j � |�� |�� q"| j dkr�| jD ]l}|j �vr���|j � �j|d�}ni }z||| j d |j < W qv ty� |j |i|| j d< Y qv0 qvtj d �r��
| �|| j d< �� || j � || j �s(|| j = |S )Nru r� r= rE rY r1 rt ) r~ r� rz r� rY r� rK re r rH ro rw )r rX r� r Z host_vars�r� Z
has_ungroupedr� r r r! r� v s2
z1InventoryCLI.toml_inventory.<locals>.format_group)r� �bool�nextr� )r r[ rX r r� r! rP r s
!zInventoryCLI.toml_inventory)r )r )�__name__�
__module__� __qualname__�__doc__r~ Z ARGUMENTSr r3 r> rG � staticmethodrL ro rK rN rm rw r� r{ r� rM rQ rO rP �
__classcell__r r r r! r 1 s6 � ) -
$%r c C s t �| � d S r )r Z cli_executor)r r r r! �main� s r� �__main__)N)%Z
__future__r r r �typeZ
__metaclass__Z ansible.clir rV r7 Zansibler r: r Zansible.cli.argumentsr r4 Zansible.errorsr r
Zansible.module_utils._textr r r
Zansible.utils.varsr Zansible.utils.displayr Zansible.vars.pluginsr r r@ � frozensetrs r r� r� r r r r! �<module> s( m