3
\me#) @ s d dl mZ d dlmZ d dlmZmZmZmZ d dl Z e j
Z
G dd dZ G dd de j Z
d d
Zee d d d
Ze ee ee dddZG dd de j Ze
jde
j G dd de jjZeeeeef f dddZdd Zd d d ZdS ) ) defaultdict)chain)AnyDictListTupleNc @ s2 e Zd ZeejejdddZdd Zdd ZdS ) SchemaPathMarks)path
start_markend_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__stryamlZMarkr r r r
r
r
r r s r c s e Zd Z fddZ 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_markmarksmarkr
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
startswithreplace)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.`.
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 listschemamarks_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 ddD ]4\}}x*|d d d D ]}||kr|j dS qW qzW dS )Nr r r& T)reverse r! )r
r r3 r sorteditems)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 mappingnested_path_prefixZkey_nodeZ
value_node
node_key_pathline_numr, r9 r/ )r r
r r; s
z,_CustomSafeLoaderWithMarks.construct_mappingFc
s t j|dd}| j|}xt|jD ]\}}|jj}| | }t||j|j} || j krj| g| j |< q&||jjkr| j | }
t
| |
} | | j |< q&xvt ||jjD ]d} | | j kr| j | }
t
| |
} | |ko|
d j |kr | j
dt||
d j|
d j | | j | < qW 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|tr4tdd t| jj D |d< |S )Nc S s g | ]}|j |jjd fqS )r )r r
r ).0vr
r
r
s z>_CustomSafeLoaderWithMarks.get_single_data.. schemamarks)r get_single_data
isinstancedictr 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|ts 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:
..
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)blobresultrJ r
r
r load_with_marks s
rX c C s t j| tdS )N)rS )r rT r )rV r
r
r rT s rT TFc C s$ t j| dd||d|rtnt jjdS )z%Return data in nicely formatted yaml.
F)Z
line_breakindentexplicit_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 s$ "!}