File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/yamlloader.cpython-310.pyc
o
�N�g � @ sr d Z ddlZddlmZ ddlmZmZ ddlZe edej
�Z g d�Z G dd� de �Z
e
fd d
�Ze
fd d �ZdS )
z
Custom YAML loading in Salt
� N)�ConstructorError)� MappingNode� SequenceNodeZ CSafeLoader)�SaltYamlSafeLoader�load� safe_loadc sZ e Zd ZdZef� fdd� Zdd� Zdd� Zdd d
�Z� fd d �Z d
d� Z
dd� Z � Z S )r z�
Create a custom YAML loader that uses the custom constructor. This allows
for the YAML loading defaults to be manipulated based on needs within salt
to make things like sls file more intuitive.
c sx t � �|� |tur | �dt| �j� | �dt| �j� | �dt| �j� | �dt| �j� | �dt| �j� || _ d S )Nztag:yaml.org,2002:mapztag:yaml.org,2002:omap�tag:yaml.org,2002:strz tag:yaml.org,2002:python/unicodeztag:yaml.org,2002:timestamp)
�super�__init__�dictZadd_constructor�type�construct_yaml_map�construct_yaml_str�construct_unicode�construct_scalar� dictclass)�self�streamr �� __class__� �I/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/yamlloader.pyr
s
�
�
zSaltYamlSafeLoader.__init__c c s( � | � � }|V | �|�}|�|� d S �N)r �construct_mapping�update)r �node�data�valuer r r r
) s
�
z%SaltYamlSafeLoader.construct_yaml_mapc C s |j S r �r �r r r r r r / s z$SaltYamlSafeLoader.construct_unicodeFc C s� t |t�stddd|j� �|j��| �|� d}| �� }|jD ]B\}}| j||d�}zt |� W n t
yD t||jd|j� �|j��w | j||d�}||v r]t||jd|� d�|j��|||< q |S )z,
Build the mapping for YAML
Nz#expected a mapping node, but found � while constructing a mapping)�deepzfound unacceptable key zfound conflicting ID '�') �
isinstancer r �id�
start_mark�flatten_mappingr r Zconstruct_object�hash� TypeError) r r r! �context�mapping�key_node�
value_node�keyr r r r r 2 s@
�
��
�
z$SaltYamlSafeLoader.construct_mappingc sX |j dkr&|jdkr n|j�d�r&|j�d�s&|j�d�|_|jdkr&d|_t� �|�S )zb
Verify integers and pass them in correctly is they are declared
as octal
ztag:yaml.org,2002:int�0)Z0b�0x� )�tagr �
startswith�lstripr r r r r r r X s
z#SaltYamlSafeLoader.construct_scalarc C s | � |�}tjj�|�S r )r �salt�utilsZ stringutilsZ
to_unicode)r r r r r r r h s
z%SaltYamlSafeLoader.construct_yaml_strc
sN g }d}|t |j�k r�|j| \}}|jdkrr|j|= t|t�r,| �|� |�|j� nWt|t�reg }|jD ] }t|t�sJtd|j d�
|j �|j ��| �|� |� |j� q6|�
� |D ]}|�|� q\n td|j d�
|j �|j ��|jdkrd|_|d7 }n|d7 }|t |j�k s |r�d d
� |jD �� � fd d
�|D �} | |j |_d S d S ) Nr ztag:yaml.org,2002:merger z,expected a mapping for merging, but found {}z@expected a mapping or list of mappings for merging, but found {}ztag:yaml.org,2002:valuer � c S s g | ]\}}|j �qS r r )�.0Z name_noder, r r r �
<listcomp>� s z6SaltYamlSafeLoader.flatten_mapping.<locals>.<listcomp>c s g | ] }|d j � vr|�qS )r r )r7 �x�Zexisting_nodesr r r8 � s )�lenr r1 r# r r& �extendr r r% �formatr$ �append�reverse)
r r �merge�indexr+ r, ZsubmergeZsubnoder Zmergeable_itemsr r: r r& l sX
��
��
�&�z"SaltYamlSafeLoader.flatten_mapping)F)
�__name__�
__module__� __qualname__�__doc__r r
r
r r r r r& �
__classcell__r r r r r s
&r c C � t j| |d�S )N��Loader��yamlr �r rI r r r r � s r c C rG )ze
.. versionadded:: 2018.3.0
Helper function which automagically uses our custom loader.
rH rJ rL r r r r � s r )rE rK Zyaml.constructorr Z
yaml.nodesr r Zsalt.utils.stringutilsr4 �getattrZ
SafeLoaderZ
BaseLoader�__all__r r r r r r r �<module> s