-
Notifications
You must be signed in to change notification settings - Fork 0
/
replace_field.py
45 lines (37 loc) · 1.41 KB
/
replace_field.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
from pathlib import Path
import argparse
import json
import re
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument('-data', help="JSON file from which data is loaded and saved",
type=Path, required=True)
parser.add_argument('-key', help="OrigKey=NewKey value pairs",
nargs='*', type=str, required=True)
parser.add_argument('-rec', action='store_true',
help="Specifies if all the document is affected")
return parser.parse_args()
def replace(data: dict, key: dict, loc: str, rec: bool):
def loop(dictionary):
for key_, value_ in dictionary.items():
if isinstance(value_, dict):
print(loc)
if loc is None or re.match(loc, key_):
for k in key:
if k in value_:
value_[key[k]] = value_.pop(k)
if rec or loc:
loop(value_)
loop(data)
return data
if __name__ == "__main__":
parser = parse_args()
with open(parser.data, 'r') as json_file:
json_data = json.load(json_file)
replace_dict = {}
for x in parser.key:
key, value = x.split('=')
replace_dict[key] = value
result = replace(data=json_data, key=replace_dict, rec=parser.rec)
with open(parser.data, 'w') as json_file:
json.dump(result, json_file)