3
\�me#) � @ s� d dl mZ d dlmZ d dlmZmZmZmZ d dl Z e j
Z
G dd� d�Z G dd� de j �Z
d d
� Zee d �d d
�Ze ee ee d�dd�ZG dd� de j �Ze
jde
j� G dd� de jj�Zeeeeef f d�dd�Zdd� Zd d d �ZdS ) � )� defaultdict)�chain)�Any�Dict�List�TupleNc @ s2 e Zd Zeejejd�dd�Zdd� Zdd� ZdS ) �SchemaPathMarks)�path�
start_mark�end_markc C s || _ || _|| _d S )N)r r
r )�selfr r
r � r
� /usr/lib/python3.6/safeyaml.py�__init__ s zSchemaPathMarks.__init__c C sp |j j| j jk s |jj| jjkr$dS |j j| j jkrH|j j| j jk rHdS |jj| jjkrl|jj| jjkrldS dS )z;Return whether other start/end marks are within self marks.FT)r
�liner �column)r �otherr
r
r � __contains__ s z SchemaPathMarks.__contains__c C s@ | j j|j jko>| j j|j jko>| jj|jjko>| jj|jjkS )N)r
r r r )r r r
r
r �__eq__+ s zSchemaPathMarks.__eq__N) �__name__�
__module__� __qualname__�str�yamlZMarkr r r r
r
r
r r s r c s e Zd Z� fdd�Z� ZS )�_CustomSafeLoaderc s t � j|�S )N)�superZconstruct_scalar)r �node)� __class__r
r �construct_python_unicode5 s z*_CustomSafeLoader.construct_python_unicode)r r r r �
__classcell__r
r
)r r r 4 s r c C s2 x,|d d d� D ]}| |kr| |k r|S qW d S )N� ���r
)Z
child_mark�marks�markr
r
r �_find_closest_parent9 s r$ )�
line_marksc C sb x\| D ]T}t || �}|r|jjdd�\}}|j|jksF|jj|j� r|jj||jd�|_qW dS )zP
Update any SchemaPathMarks.path for items not under the proper parent.
�.r N)r$ r �rsplit�
startswith�replace)r% r# �parentZ path_prefixZ _path_idxr
r
r � _reparent_schema_mark_children@ s
r+ )�new_markr" �returnc C sf g }d}x@|D ]8}|| kr&|j |� q| |kr<d}|j | � |j |� qW |rXt|� n
|j | � |S )a= Insert new_mark into marks, ordering ancestors first.
Reparent existing SchemaPathMarks.path when new_mark is a parent of
an existing mark item.
Because schema processing is depth first, leaf/child mappings and
sequences may be processed for SchemaPathMarks before their parents.
This leads to SchemaPathMarks.path of 'grandchildren' being incorrectly
parented by the root dictionary instead of an intermediary parents below
root.
Walk through the list of existing marks and reparent marks that are
contained within the new_mark.
FT)�appendr+ )r, r" � new_marksZreparent_pathsr# r
r
r � _add_mark_and_reparent_marksO s
r0 c sN e Zd ZdZ� fdd�Zdd� Z� fdd�Zd
� fd d
� Z� fd d �Z� Z S )�_CustomSafeLoaderWithMarksa� A loader which provides line and column start and end marks for YAML.
If the YAML loaded represents a dictionary, get_single_data will inject
a top-level "schemamarks" key in that dictionary which can be used at
call-sites to process YAML paths schemamark metadata when annotating
YAML files for errors.
The schemamarks key is dictionary where each key is a dot-delimited path
into the YAML object. Each dot represents an element that is nested under
a parent and list items are represented with the format
`<parent>.<list-index>`.
The values in schemamarks will be the line number in the original content
where YAML element begins to aid in annotation when encountering schema
errors.
The example YAML shows expected schemamarks for both dicts and lists:
one: val1
two:
subtwo: val2
three: [val3, val4]
schemamarks == {
"one": 1, "two": 2, "two.subtwo": 3, "three": 4, "three.0": 4,
"three.1": 4
}
c s t � j|� tt�| _d S )N)r r r �list�schemamarks_by_line)r �stream)r r
r r � s z#_CustomSafeLoaderWithMarks.__init__c C s� |j j| jkrf| j|j j d }x2| j|j j dd � D ]}||kr8||kr8|}q8W ||krf|jd S xJt| jj� dd�D ]4\}}x*|d d d� D ]}||kr�|j� d�S q�W qzW dS )Nr r r& T)�reverse� r! )r
r r3 r �sorted�items)r r Zmost_specific_markZ path_markZ _line_num� schema_marksr# r
r
r �_get_nested_path_prefix� s
z2_CustomSafeLoaderWithMarks._get_nested_path_prefixc sr t � j|�}| j|�}xV|jD ]L\}}|� |j� �}|jj}t||j|j�}| j| } t || �}
|
| j|< q W |S )N)
r �construct_mappingr: �valuer
r r r r3 r0 ) r r �mapping�nested_path_prefixZkey_nodeZ
value_node�
node_key_path�line_numr, r9 r/ )r r
r r; � s
z,_CustomSafeLoaderWithMarks.construct_mappingFc
s t � j|dd�}| j|�}x�t|j�D ]�\}}|jj}|� |� �}t||j|j�} || j krj| g| j |< q&||jjkr�| j | }
t
| |
�} | | j |< q&xvt ||jj�D ]d} | | j kr�| j | }
t
| |
�} | |ko�|
d j |k�r | j
dt||
d j|
d j�� | | j | < q�W q&W |S )NT)�deepr r r! )r �construct_sequencer: � enumerater< r
r r r r3 r0 �ranger �insert)
r r rA Zsequencer> �indexZ
sequence_itemr@ r? r, r9 r/ Z
inner_line)r r
r rB � s<
z-_CustomSafeLoaderWithMarks.construct_sequencec s8 t � j� }t|t�r4tdd� t| jj� � D ��|d< |S )Nc S s g | ]}|j |jjd f�qS )r )r r
r )�.0�vr
r
r �
<listcomp>� s z>_CustomSafeLoaderWithMarks.get_single_data.<locals>.<listcomp>� schemamarks)r �get_single_data�
isinstance�dictr r3 �values)r �data)r r
r rK � s
z*_CustomSafeLoaderWithMarks.get_single_data)F)
r r r �__doc__r r: r; rB rK r r
r
)r r r1 r s *r1 z tag:yaml.org,2002:python/unicodec @ s e Zd ZdZdd� ZdS )�NoAliasSafeDumperz>A class which avoids constructing anchors/aliases on yaml dumpc C s dS )NTr
)r rO r
r
r �ignore_aliases� s z NoAliasSafeDumper.ignore_aliasesN)r r r rP rR r
r
r
r rQ � s rQ )r- c C s0 t j| td�}t|t�s i }n
|jd�}||fS )a� Perform YAML SafeLoad and track start and end marks during parse.
JSON schema errors come with an encoded object path such as:
<key1>.<key2>.<list_item_index>
YAML loader needs to preserve a mapping of schema path to line and column
marks to annotate original content with JSON schema error marks for the
command:
cloud-init devel schema --annotate
)�LoaderrJ )r �loadr1 rL rM �pop)�blob�resultrJ r
r
r �load_with_marks� s
rX c C s t j| td�S )N)rS )r rT r )rV r
r
r rT s rT TFc C s$ t j| dd||d|rtnt jjd�S )z%Return data in nicely formatted yaml.�
� F)Z
line_break�indent�explicit_start� explicit_endZdefault_flow_styleZDumper)r �dumprQ �dumper�
SafeDumper)�objr\ r] Znoaliasr
r
r �dumps s rb )TTF) � collectionsr � itertoolsr Ztypingr r r r r Z YAMLErrorr Z
SafeLoaderr r$ r+ r0 r1 Zadd_constructorr r_ r` rQ r �intrX rT rb r
r
r
r �<module> s$ "!}