Ben Gamari
12 years ago
committed by
Piotr Esden-Tempski
1 changed files with 70 additions and 0 deletions
@ -0,0 +1,70 @@ |
|||||
|
#!/usr/bin/env python |
||||
|
|
||||
|
import sys |
||||
|
import yaml |
||||
|
import csv |
||||
|
from collections import OrderedDict |
||||
|
|
||||
|
def repr_odict(dumper, data): |
||||
|
""" |
||||
|
>>> data = OrderedDict([('foo', 'bar'), ('mumble', 'quux'), ('baz', 'gorp')]) |
||||
|
>>> yaml.dump(data, default_flow_style=False) |
||||
|
'!!omap\\n- foo: bar\\n- mumble: quux\\n- baz: gorp\\n' |
||||
|
>>> yaml.dump(data, default_flow_style=True) |
||||
|
'!!omap [foo: bar, mumble: quux, baz: gorp]\\n' |
||||
|
""" |
||||
|
return repr_pairs(dumper, u'tag:yaml.org,2002:omap', data.iteritems()) |
||||
|
|
||||
|
yaml.add_representer(OrderedDict, repr_odict) |
||||
|
|
||||
|
def repr_pairs(dump, tag, sequence, flow_style=None): |
||||
|
"""This is the same code as BaseRepresenter.represent_sequence(), |
||||
|
but the value passed to dump.represent_data() in the loop is a |
||||
|
dictionary instead of a tuple.""" |
||||
|
|
||||
|
value = [] |
||||
|
node = yaml.SequenceNode(tag, value, flow_style=flow_style) |
||||
|
if dump.alias_key is not None: |
||||
|
dump.represented_objects[dump.alias_key] = node |
||||
|
best_style = True |
||||
|
for (key, val) in sequence: |
||||
|
item = dump.represent_data({key: val}) |
||||
|
if not (isinstance(item, yaml.ScalarNode) and not item.style): |
||||
|
best_style = False |
||||
|
value.append(item) |
||||
|
if flow_style is None: |
||||
|
if dump.default_flow_style is not None: |
||||
|
node.flow_style = dump.default_flow_style |
||||
|
else: |
||||
|
node.flow_style = best_style |
||||
|
return node |
||||
|
|
||||
|
def convert_file(fname): |
||||
|
reader = csv.reader(open(fname, 'r')) |
||||
|
|
||||
|
registers = OrderedDict() |
||||
|
for register_name, lsb, width, field_name, description, reset_value, access in reader: |
||||
|
if register_name not in registers: |
||||
|
registers[register_name] = { |
||||
|
'fields': OrderedDict(), |
||||
|
} |
||||
|
|
||||
|
register = registers[register_name] |
||||
|
fields = register['fields'] |
||||
|
if field_name in fields: |
||||
|
raise RuntimeError('Duplicate field name "%s" in register "%s"' % |
||||
|
field_name, register_name) |
||||
|
else: |
||||
|
fields[field_name] = { |
||||
|
'lsb': int(lsb), |
||||
|
'width': int(width), |
||||
|
'description': description, |
||||
|
'reset_value': reset_value, |
||||
|
'access': access, |
||||
|
} |
||||
|
|
||||
|
with open(fname.replace('.csv', '.yaml'), 'w') as out_file: |
||||
|
yaml.dump(registers, out_file, default_flow_style=False) |
||||
|
|
||||
|
for fname in sys.argv[1:]: |
||||
|
convert_file(fname) |
Loading…
Reference in new issue