1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
# Copyright 2016 Free Software Foundation, Inc.
# This file is part of GNU Radio
#
# SPDX-License-Identifier: GPL-2.0-or-later
#
from collections import OrderedDict
import yaml
from ..params.param import attributed_str
class GRCDumper(yaml.SafeDumper):
@classmethod
def add(cls, data_type):
def decorator(func):
cls.add_representer(data_type, func)
return func
return decorator
def represent_ordered_mapping(self, data):
value = []
node = yaml.MappingNode(u'tag:yaml.org,2002:map',
value, flow_style=False)
if self.alias_key is not None:
self.represented_objects[self.alias_key] = node
for item_key, item_value in data.items():
node_key = self.represent_data(item_key)
node_value = self.represent_data(item_value)
value.append((node_key, node_value))
return node
def represent_ordered_mapping_flowing(self, data):
node = self.represent_ordered_mapping(data)
node.flow_style = True
return node
def represent_list_flowing(self, data):
node = self.represent_list(data)
node.flow_style = True
return node
def represent_ml_string(self, data):
node = self.represent_str(data)
node.style = '|'
return node
class OrderedDictFlowing(OrderedDict):
pass
class ListFlowing(list):
pass
class MultiLineString(str):
pass
GRCDumper.add_representer(OrderedDict, GRCDumper.represent_ordered_mapping)
GRCDumper.add_representer(
OrderedDictFlowing, GRCDumper.represent_ordered_mapping_flowing)
GRCDumper.add_representer(ListFlowing, GRCDumper.represent_list_flowing)
GRCDumper.add_representer(tuple, GRCDumper.represent_list)
GRCDumper.add_representer(MultiLineString, GRCDumper.represent_ml_string)
GRCDumper.add_representer(yaml.nodes.ScalarNode, lambda r, n: n)
GRCDumper.add_representer(attributed_str, GRCDumper.represent_str)
def dump(data, stream=None, **kwargs):
config = dict(stream=stream, default_flow_style=False,
indent=4, Dumper=GRCDumper)
config.update(kwargs)
return yaml.dump_all([data], **config)
safe_load = yaml.safe_load
__with_libyaml__ = yaml.__with_libyaml__
|
