File: //usr/lib/python3.9/site-packages/ansible/inventory/__pycache__/group.cpython-39.pyc
a
�)g�% � @ s� d dl mZmZmZ eZd dlmZmZ d dl m
Z
d dl m Z
d dlmZ d dlmZmZ d dlmZ d dlmZ e� Zdd d �ZG d
d� d�ZdS )� )�absolute_import�division�print_function)�Mapping�MutableMapping)�chain)� constants)� AnsibleError)� to_native�to_text)�Display)� combine_vars�_Fc C s� d}| r�t j�| �}|r�dtt|��t| �f }t jdvs>|rnt j�|| �} |s�t jdks�t�d| � d}n t jdkr�t�d| � d }|r�t� |� | S )
N� z,invalid character(s) "%s" in group name (%s))�never�ignoreZsilentlyz
Replacing zaInvalid characters were found in group names and automatically replaced, use -vvvv to see detailsr zNot replacing %szWInvalid characters were found in group names but not replaced, use -vvvv to see details)
�CZINVALID_VARIABLE_NAMES�findallr �setZ TRANSFORM_INVALID_GROUP_CHARS�sub�displayZvvvvZwarning)�nameZreplacer�forceZsilent�warnZ
invalid_chars�msg� r �;/usr/lib/python3.9/site-packages/ansible/inventory/group.py�to_safe_group_name s
r c @ s� e Zd ZdZd0dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
d1dd�Z dd� Z dd� Z
edd� �Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� ZdS )2�Groupz a group of ansible hosts Nc C s> d| _ t|�| _g | _d | _i | _g | _g | _d | _d| _ d S )Nr � )
�depthr r �hosts�_hosts�vars� child_groups�
parent_groups� _hosts_cache�priority)�selfr r r r �__init__= s
zGroup.__init__c C s | � � S �N��get_name�r( r r r �__repr__I s zGroup.__repr__c C s | � � S r* r+ r- r r r �__str__L s z
Group.__str__c C s | � � S r* )� serializer- r r r � __getstate__O s zGroup.__getstate__c C s
| � |�S r* )� deserialize)r( �datar r r � __setstate__R s zGroup.__setstate__c C sF g }| j D ]}|�|�� � q
d | _t| j| j�� || j| j d�}|S )N)r r# r% r r! )
r% �appendr0 r" �dictr r# �copyr r! )r( r% �parent�resultr r r r0 U s
�zGroup.serializec C s| | � � |�d�| _|�dt� �| _|�dd�| _|�dg �| _d | _|�dg �}|D ] }t� }|� |� | j
� |� qVd S )Nr r# r r r! r% ) r) �getr r6 r# r r! r" r r2 r% r5 )r( r3 r% Z parent_data�gr r r r2 f s
zGroup.deserializeFc s� t g �}t t| � ��}|r$|�| � |rF|r2| gng }|�t| � �� |r�|�|� t g �}t�� fdd�|D ��D ]$}|�|� |rt||vrt|�|� qt|�|� |}qF|r�|S |S )a�
Given `rel` that is an iterable property of Group,
consitituting a directed acyclic graph among all groups,
Returns a set of all groups in full tree
A B C
| / | /
| / | /
D -> E
| / vertical connections
| / are directed upward
F
Called on F, returns set of (A, B, C, D, E)
c 3 s | ]}t |� �V qd S r* )�getattr��.0r; ��relr r � <genexpr>� � z+Group._walk_relationship.<locals>.<genexpr>) r r< �add�extend�updater �
from_iterabler5 �difference_update) r( r@ � include_self�preserve_ordering�seen� unprocessedZorderedZnew_unprocessedZnew_itemr r? r �_walk_relationshipt s(
zGroup._walk_relationshipc C s
| � d�S )Nr% �rL r- r r r �
get_ancestors� s zGroup.get_ancestorsc K s | j di |��S )Nr$ )r$ rM )r( �kwargsr r r �get_descendants� s zGroup.get_descendantsc C s | j d u rt| j�| _ | j S r* )r" r r! r- r r r �
host_names� s
zGroup.host_namesc C s | j S r* �r r- r r r r, � s zGroup.get_namec C s� d}| |krt d��|| jvr�|�� }| �� }||v rRtdt|j�t| j�f ��|�| � |�|� d}| j�|� t | j
d |j
g�|_
|� � | jdd� |j D �vr�|j �| � |�
� D ]}|j|d� q�| �� |S ) NFzcan't add group to itselfzGAdding group '%s' as child to '%s' creates a recursive dependency loop.Tr c S s g | ]
}|j �qS r rR r= r r r �
<listcomp>� rB z)Group.add_child_group.<locals>.<listcomp>)Z additions)� Exceptionr$ rN r r
r rC rG r5 �maxr �_check_children_depthr% � get_hostsZpopulate_ancestors�clear_hosts_cache)r( �group�addedZstart_ancestorsZ
new_ancestors�hr r r �add_child_group� s(
zGroup.add_child_groupc C s� | j }| j }tg �}t| j�}|r�|�|� |d7 }|�� }tg �}|D ] }|j |k rH||_ |�|j� qH|| t|�kr tdt| j� ��q d S )Nr z5The group named '%s' has a recursive dependency loop.) r r r$ rE r7 �lenr r
r )r( r Z start_depthrJ rK Z
to_processr; r r r rV � s
zGroup._check_children_depthc C sD d}|j | jvr@| j�|� | j�|j � |�| � | �� d}|S �NFT)r rQ r! r5 r" rC � add_grouprX )r( �hostrZ r r r �add_host� s
zGroup.add_hostc C sD d}|j | jv r@| j�|� | j�|j � |�| � | �� d}|S r^ )r rQ r! �remover" Z remove_grouprX )r( r` Zremovedr r r � remove_host� s
zGroup.remove_hostc C s^ |dkr| � t|�� nB|| jv rPt| j| t�rPt|t�rPt| j||i�| _n
|| j|< d S )NZansible_group_priority)� set_priority�intr# �
isinstancer r r
)r( �key�valuer r r � set_variable� s
$zGroup.set_variablec C s d | _ | �� D ]
}d |_ qd S r* )r& rN )r( r; r r r rX � s zGroup.clear_hosts_cachec C s | j d u r| �� | _ | j S r* )r& �
_get_hostsr- r r r rW s
zGroup.get_hostsc C s\ g }i }| j ddd�D ]@}|j}|D ]0}||vr$d||< | jdkrJ|jrJq$|�|� q$q|S )NT)rH rI r �all)rP r! r Zimplicitr5 )r( r! rJ ZkidZ kid_hostsZkkr r r rj
s zGroup._get_hostsc C s
| j �� S r* )r# r7 r- r r r �get_vars s zGroup.get_varsc C s& zt |�| _W n ty Y n0 d S r* )re r' � TypeError)r( r' r r r rd s zGroup.set_priority)N)FF)�__name__�
__module__� __qualname__�__doc__r) r. r/ r1 r4 r0 r2 rL rN rP �propertyrQ r, r\ rV ra rc ri rX rW rj rl rd r r r r r 8 s0
'
#
r N)r FF)Z
__future__r r r �typeZ
__metaclass__�collections.abcr r � itertoolsr Zansibler r Zansible.errorsr Zansible.module_utils._textr
r Zansible.utils.displayr Zansible.utils.varsr
r r r r r r r �<module> s