EXDTools/apply.py at main · xivdev/EXDTools · GitHub

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
85
86
87
88
from pathlib import Path
from ruamel.yaml import YAML
from ruamel.yaml.comments import CommentedMap
from ruamel.yaml.comments import CommentedSeq
from ruamel.yaml.compat import StringIO

class MyYAML(YAML):
    def dump(self, data, stream=None, **kw):
        inefficient = False
        if stream is None:
            inefficient = True
            stream = StringIO()
        YAML.dump(self, data, stream, **kw)
        if inefficient:
            return stream.getvalue()

def order_keys(data, order):
    if not isinstance(data, CommentedMap):
        raise Exception("Not a CommentedMap")

    reordered = CommentedMap()
    for key in order:
        if key in data:
            comment = data.ca.items.get(key, None)
            reordered[key] = data.pop(key)
            if comment:
                reordered.ca.items[key] = comment

    for key in list(data):
        comment = data.ca.items.get(key, None)
        reordered[key] = data.pop(key)
        if comment:
            reordered.ca.items[key] = comment

    data.clear()
    data.update(reordered)
    return data

def recurse(data, *args):
    for a in args:
        data = a(data)

    ret = []
    for fld in data.get('fields', []):
        ret.append(recurse(fld, *args))

    if len(ret) != 0:
        comment = data.ca.items.get('fields', None)
        data['fields'] = ret
        if comment:
            data.ca.items['fields'] = comment
    return data

def apply_flow_style(data):
    if 'targets' in data:
        comment = data.ca.items.get('targets', None)
        seq = CommentedSeq(data['targets'])
        seq.fa.set_flow_style()
        data['targets'] = seq
        if comment:
            data.ca.items['targets'] = comment
    return data

yaml = MyYAML()
yaml.preserve_quotes = True
yaml.width = 4096
yaml.indent(sequence=4, offset=2)

for path in sorted(list(Path('./schemas').glob('*/*.yml')), key = lambda p: p.stem):
    inp = path.read_text().strip()
    data = yaml.load(path)

    data = order_keys(data, ['name', 'displayField', 'fields' 'relations'])

    ret = []
    for fld in data['fields']:
        ret.append(recurse(fld, lambda f: order_keys(f, ['name', 'comment', 'type', 'count', 'fields']), apply_flow_style))
    if len(ret) != 0:
        comment = data.ca.items.get('fields', None)
        data['fields'] = ret
        if comment:
            data.ca.items['fields'] = comment

    output = yaml.dump(data).strip()
    if inp != output:
        print(path, 'changed')
        #print(output)
        path.write_text(output)