-
Notifications
You must be signed in to change notification settings - Fork 13
/
Copy pathschema.py
executable file
·65 lines (45 loc) · 1.14 KB
/
schema.py
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
#!/usr/bin/env python3
import json
import sys
def convert_field(jd, gfield):
name = jd['f9'][gfield['F4_f1']]
f2 = gfield['F4_f2']
f3 = gfield['F4_f3']
if f3 in [0, 1]:
gtype = jd['f5'][f2]['F5_f1']
elif f3 in [4, 5, 6]:
gtype = jd['f9'][f2]
if f3 in [1, 5]:
gtype = '[{}]'.format(gtype)
return {
'name': name,
'type': gtype,
}
def convert_type(jd, gtype):
name = gtype['F5_f1']
if 'F5_f2' in gtype:
primary_key = jd['f4'][gtype['F5_f2']]
primary_key = convert_field(jd, primary_key)
else:
primary_key = None
fields = []
for field in gtype['F5_f3']:
field = jd['f4'][field]
field = convert_field(jd, field)
fields.append(field)
return {
'name': name,
'primary_key': primary_key,
'fields': fields,
}
def main(fin):
fin = open(fin, 'rt')
jd = json.load(fin)
schema = []
for gtype in jd['f5']:
gtype = convert_type(jd, gtype)
schema.append(gtype)
json.dump(schema, sys.stdout)
fin.close()
if __name__ == '__main__':
main(sys.argv[1])