guangzong/stm32-data
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

cleanup yaml

Browse Source
This commit is contained in:
Dario Nieuwenhuis 2022-02-06 22:26:53 +01:00
parent f79e304d07
commit 689c9080ee
1 changed files with 10 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
stm32data/yaml.py
View File

@ -12,23 +12,24 @@ class DecimalInt:
self.val = val
def represent_decimal_int(dumper, data):
def represent_decimalint(dumper, data):
return dumper.represent_int(data.val)
yaml.add_representer(DecimalInt, represent_decimal_int)
class HexInt:
def __init__(self, val):
self.val = val
def represent_hex_int(dumper, data):
def represent_hexint(dumper, data):
return dumper.represent_int(hex(data.val))
yaml.add_representer(HexInt, represent_hex_int)
def represent_int(dumper, data):
if data > 0x10000:
return dumper.represent_int(hex(data))
else:
return dumper.represent_int(data)
def represent_ordereddict(dumper, data):
@ -43,19 +44,12 @@ def represent_ordereddict(dumper, data):
return yaml.nodes.MappingNode(u'tag:yaml.org,2002:map', value)
yaml.add_representer(DecimalInt, represent_decimalint)
yaml.add_representer(HexInt, represent_hexint)
yaml.add_representer(int, represent_int)
yaml.add_representer(OrderedDict, represent_ordereddict)
def hexint_presenter(dumper, data):
if data > 0x10000:
return dumper.represent_int(hex(data))
else:
return dumper.represent_int(data)
yaml.add_representer(int, hexint_presenter)
def load(*args, **kwargs):
return yaml.load(*args, Loader=SafeLoader, **kwargs)