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
|
#!/usr/bin/env python3
# Make sure you have python3 installed.
# Ensure that the json_formatter is kept in Tools with this script.
# For Windows:
# Using command prompt type "python name.py"
# For Max OS X or Linux:
# Swap any "\\" with "/", then run the script as in windows.
import json
import os
def gen_new(path):
change = False
with open(path, "r", encoding="utf-8") as json_file:
try:
json_data = json.load(json_file)
except UnicodeDecodeError:
print(f"UnicodeDecodeError in {path}")
return None
except json.decoder.JSONDecodeError:
print(f"JSONDecodeError in {path}")
return None
for jo in json_data:
# We only want JsonObjects
if type(jo) is str:
continue
if not jo.get('name'):
continue
if type(jo['name']) == dict:
continue
if jo.get('type') not in ['AMMO', 'ARMOR', 'BATTERY', 'bionic',
'BIONIC_ITEM', 'BIONIC_ITEM', 'BOOK',
'COMESTIBLE', 'ENGINE', 'fault',
'GENERIC', 'GUN', 'GUNMOD',
'item_action', 'ITEM_CATEGORY',
'MAGAZINE', 'map_extra', 'martial_art',
'mission_definition', 'MONSTER',
'mutation', 'npc_class', 'PET_ARMOR',
'proficiency', 'skill', 'TOOL',
'TOOL_ARMOR', 'tool_quality', 'TOOLMOD',
'vehicle_part', 'vehicle_part_category',
'vitamin', 'WHEEL', 'SPELL']:
continue
if jo.get('name_plural'):
if jo['name'] == jo['name_plural']:
name_obj = {'str_sp': jo['name']}
else:
name_obj = {'str': jo['name'], 'str_pl': jo['name_plural']}
jo['name'] = name_obj
del jo['name_plural']
change = True
else:
name_obj = {'str': jo['name']}
jo['name'] = name_obj
change = True
return json_data if change else None
def change_file(json_dir):
for root, directories, filenames in os.walk(f"..\\{json_dir}"):
for filename in filenames:
path = os.path.join(root, filename)
if path.endswith(".json"):
new = gen_new(path)
if new is not None:
with open(path, "w", encoding="utf-8") as jf:
json.dump(new, jf, ensure_ascii=False)
os.system(f".\\json_formatter.exe {path}")
if __name__ == "__main__":
json_dir = input("What directory are the json files in? ")
change_file(json_dir)
|
