yamlutils.py

   1 from collections import OrderedDict
   2
   3 import yaml
   4
   5 import core
   6 import utils
   7
   8
   9 yaml.Dumper.ignore_aliases = lambda self, data: True
  10
  11
  12 def set_representer(dumper, data):
  13     return dumper.represent_list(sorted(list(data), key=utils.natural_sort_key))
  14
  15
  16 yaml.add_representer(set, set_representer)
  17
  18
  19 yaml.add_representer(tuple, lambda dumper, data: dumper.represent_list(data))
  20
  21
  22 # Workaround pyyaml idiocy of forcibly sort dicts by key
  23 class DictRenderWrapper:
  24
  25     def __init__(self, d):
  26         self.d = d
  27
  28     def __iter__(self):
  29         return iter(self.d.items())
  30
  31
  32 def dict_representer(dumper, data):
  33     ordered = OrderedDict()
  34     if "label" in data:
  35         ordered["label"] = data["label"]
  36     ordered.update(sorted(data.items()))
  37
  38     #return dumper.represent_dict(DictRenderWrapper(ordered))
  39     flow_style = None
  40     if len(str(data)) > 50:
  41         flow_style = False
  42     return dumper.represent_mapping('tag:yaml.org,2002:map', DictRenderWrapper(ordered), flow_style)
  43
  44
  45 yaml.add_representer(dict, dict_representer)
  46
  47 yaml.add_representer(core.REG, lambda dumper, data: dumper.represent_str(data.name))
  48 yaml.add_representer(core.ADDR, lambda dumper, data: dumper.represent_str(data.addr))
  49 yaml.add_representer(core.MEM, lambda dumper, data: dumper.represent_str(str(data)))