comparison env/lib/python3.7/site-packages/yaml/serializer.py @ 0:26e78fe6e8c4 draft

"planemo upload commit c699937486c35866861690329de38ec1a5d9f783"
author shellac
date Sat, 02 May 2020 07:14:21 -0400
parents
children
comparison
equal deleted inserted replaced
-1:000000000000 0:26e78fe6e8c4
1
2 __all__ = ['Serializer', 'SerializerError']
3
4 from .error import YAMLError
5 from .events import *
6 from .nodes import *
7
8 class SerializerError(YAMLError):
9 pass
10
11 class Serializer:
12
13 ANCHOR_TEMPLATE = 'id%03d'
14
15 def __init__(self, encoding=None,
16 explicit_start=None, explicit_end=None, version=None, tags=None):
17 self.use_encoding = encoding
18 self.use_explicit_start = explicit_start
19 self.use_explicit_end = explicit_end
20 self.use_version = version
21 self.use_tags = tags
22 self.serialized_nodes = {}
23 self.anchors = {}
24 self.last_anchor_id = 0
25 self.closed = None
26
27 def open(self):
28 if self.closed is None:
29 self.emit(StreamStartEvent(encoding=self.use_encoding))
30 self.closed = False
31 elif self.closed:
32 raise SerializerError("serializer is closed")
33 else:
34 raise SerializerError("serializer is already opened")
35
36 def close(self):
37 if self.closed is None:
38 raise SerializerError("serializer is not opened")
39 elif not self.closed:
40 self.emit(StreamEndEvent())
41 self.closed = True
42
43 #def __del__(self):
44 # self.close()
45
46 def serialize(self, node):
47 if self.closed is None:
48 raise SerializerError("serializer is not opened")
49 elif self.closed:
50 raise SerializerError("serializer is closed")
51 self.emit(DocumentStartEvent(explicit=self.use_explicit_start,
52 version=self.use_version, tags=self.use_tags))
53 self.anchor_node(node)
54 self.serialize_node(node, None, None)
55 self.emit(DocumentEndEvent(explicit=self.use_explicit_end))
56 self.serialized_nodes = {}
57 self.anchors = {}
58 self.last_anchor_id = 0
59
60 def anchor_node(self, node):
61 if node in self.anchors:
62 if self.anchors[node] is None:
63 self.anchors[node] = self.generate_anchor(node)
64 else:
65 self.anchors[node] = None
66 if isinstance(node, SequenceNode):
67 for item in node.value:
68 self.anchor_node(item)
69 elif isinstance(node, MappingNode):
70 for key, value in node.value:
71 self.anchor_node(key)
72 self.anchor_node(value)
73
74 def generate_anchor(self, node):
75 self.last_anchor_id += 1
76 return self.ANCHOR_TEMPLATE % self.last_anchor_id
77
78 def serialize_node(self, node, parent, index):
79 alias = self.anchors[node]
80 if node in self.serialized_nodes:
81 self.emit(AliasEvent(alias))
82 else:
83 self.serialized_nodes[node] = True
84 self.descend_resolver(parent, index)
85 if isinstance(node, ScalarNode):
86 detected_tag = self.resolve(ScalarNode, node.value, (True, False))
87 default_tag = self.resolve(ScalarNode, node.value, (False, True))
88 implicit = (node.tag == detected_tag), (node.tag == default_tag)
89 self.emit(ScalarEvent(alias, node.tag, implicit, node.value,
90 style=node.style))
91 elif isinstance(node, SequenceNode):
92 implicit = (node.tag
93 == self.resolve(SequenceNode, node.value, True))
94 self.emit(SequenceStartEvent(alias, node.tag, implicit,
95 flow_style=node.flow_style))
96 index = 0
97 for item in node.value:
98 self.serialize_node(item, node, index)
99 index += 1
100 self.emit(SequenceEndEvent())
101 elif isinstance(node, MappingNode):
102 implicit = (node.tag
103 == self.resolve(MappingNode, node.value, True))
104 self.emit(MappingStartEvent(alias, node.tag, implicit,
105 flow_style=node.flow_style))
106 for key, value in node.value:
107 self.serialize_node(key, node, None)
108 self.serialize_node(value, node, key)
109 self.emit(MappingEndEvent())
110 self.ascend_resolver()
111