132 lines
3.6 KiB
Python
132 lines
3.6 KiB
Python
|
# coding: utf-8
|
||
|
|
|
||
|
|
from __future__ import print_function
|
||
|
|
|
||
|
|
import sys
|
||
|
|
from .compat import string_types
|
||
|
|
|
||
|
|
if False: # MYPY
|
||
|
|
from typing import Dict, Any, Text # NOQA
|
||
|
|
|
||
|
|
|
||
|
|
class Node(object):
|
||
|
|
__slots__ = 'tag', 'value', 'start_mark', 'end_mark', 'comment', 'anchor'
|
||
|
|
|
||
|
|
def __init__(self, tag, value, start_mark, end_mark, comment=None, anchor=None):
|
||
|
|
# type: (Any, Any, Any, Any, Any, Any) -> None
|
||
|
|
self.tag = tag
|
||
|
|
self.value = value
|
||
|
|
self.start_mark = start_mark
|
||
|
|
self.end_mark = end_mark
|
||
|
|
self.comment = comment
|
||
|
|
self.anchor = anchor
|
||
|
|
|
||
|
|
def __repr__(self):
|
||
|
|
# type: () -> str
|
||
|
|
value = self.value
|
||
|
|
# if isinstance(value, list):
|
||
|
|
# if len(value) == 0:
|
||
|
|
# value = '<empty>'
|
||
|
|
# elif len(value) == 1:
|
||
|
|
# value = '<1 item>'
|
||
|
|
# else:
|
||
|
|
# value = '<%d items>' % len(value)
|
||
|
|
# else:
|
||
|
|
# if len(value) > 75:
|
||
|
|
# value = repr(value[:70]+u' ... ')
|
||
|
|
# else:
|
||
|
|
# value = repr(value)
|
||
|
|
value = repr(value)
|
||
|
|
return '%s(tag=%r, value=%s)' % (self.__class__.__name__, self.tag, value)
|
||
|
|
|
||
|
|
def dump(self, indent=0):
|
||
|
|
# type: (int) -> None
|
||
|
|
if isinstance(self.value, string_types):
|
||
|
|
sys.stdout.write(
|
||
|
|
'{}{}(tag={!r}, value={!r})\n'.format(
|
||
|
|
' ' * indent, self.__class__.__name__, self.tag, self.value
|
||
|
|
)
|
||
|
|
)
|
||
|
|
if self.comment:
|
||
|
|
sys.stdout.write(' {}comment: {})\n'.format(' ' * indent, self.comment))
|
||
|
|
return
|
||
|
|
sys.stdout.write(
|
||
|
|
'{}{}(tag={!r})\n'.format(' ' * indent, self.__class__.__name__, self.tag)
|
||
|
|
)
|
||
|
|
if self.comment:
|
||
|
|
sys.stdout.write(' {}comment: {})\n'.format(' ' * indent, self.comment))
|
||
|
|
for v in self.value:
|
||
|
|
if isinstance(v, tuple):
|
||
|
|
for v1 in v:
|
||
|
|
v1.dump(indent + 1)
|
||
|
|
elif isinstance(v, Node):
|
||
|
|
v.dump(indent + 1)
|
||
|
|
else:
|
||
|
|
sys.stdout.write('Node value type? {}\n'.format(type(v)))
|
||
|
|
|
||
|
|
|
||
|
|
class ScalarNode(Node):
|
||
|
|
"""
|
||
|
|
styles:
|
||
|
|
? -> set() ? key, no value
|
||
|
|
" -> double quoted
|
||
|
|
' -> single quoted
|
||
|
|
| -> literal style
|
||
|
|
> -> folding style
|
||
|
|
"""
|
||
|
|
|
||
|
|
__slots__ = ('style',)
|
||
|
|
id = 'scalar'
|
||
|
|
|
||
|
|
def __init__(
|
||
|
|
self, tag, value, start_mark=None, end_mark=None, style=None, comment=None, anchor=None
|
||
|
|
):
|
||
|
|
# type: (Any, Any, Any, Any, Any, Any, Any) -> None
|
||
|
|
Node.__init__(self, tag, value, start_mark, end_mark, comment=comment, anchor=anchor)
|
||
|
|
self.style = style
|
||
|
|
|
||
|
|
|
||
|
|
class CollectionNode(Node):
|
||
|
|
__slots__ = ('flow_style',)
|
||
|
|
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
tag,
|
||
|
|
value,
|
||
|
|
start_mark=None,
|
||
|
|
end_mark=None,
|
||
|
|
flow_style=None,
|
||
|
|
comment=None,
|
||
|
|
anchor=None,
|
||
|
|
):
|
||
|
|
# type: (Any, Any, Any, Any, Any, Any, Any) -> None
|
||
|
|
Node.__init__(self, tag, value, start_mark, end_mark, comment=comment)
|
||
|
|
self.flow_style = flow_style
|
||
|
|
self.anchor = anchor
|
||
|
|
|
||
|
|
|
||
|
|
class SequenceNode(CollectionNode):
|
||
|
|
__slots__ = ()
|
||
|
|
id = 'sequence'
|
||
|
|
|
||
|
|
|
||
|
|
class MappingNode(CollectionNode):
|
||
|
|
__slots__ = ('merge',)
|
||
|
|
id = 'mapping'
|
||
|
|
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
tag,
|
||
|
|
value,
|
||
|
|
start_mark=None,
|
||
|
|
end_mark=None,
|
||
|
|
flow_style=None,
|
||
|
|
comment=None,
|
||
|
|
anchor=None,
|
||
|
|
):
|
||
|
|
# type: (Any, Any, Any, Any, Any, Any, Any) -> None
|
||
|
|
CollectionNode.__init__(
|
||
|
|
self, tag, value, start_mark, end_mark, flow_style, comment, anchor
|
||
|
|
)
|
||
|
|
self.merge = None
|