You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
3 years ago
|
import os
|
||
|
|
import json
|
||
|
|
import re
|
||
|
|
data_dir = os.path.join(os.path.dirname(__file__), '..', 'data')
|
||
|
|
map_info_path = os.path.join(data_dir, 'MapInfos.json')
|
||
|
|
|
||
|
|
event_ext_id_start = 100000
|
||
|
|
|
||
|
|
label_pattern = re.compile("<SF_EXTRA_EVENT:\s*(\d*)>", re.IGNORECASE)
|
||
|
|
|
||
|
|
|
||
|
|
def extra_map_event(map_id: int):
|
||
|
|
"""
|
||
|
|
Extra map event data from map_id
|
||
|
|
:param map_id: map id
|
||
|
|
|
||
|
|
"""
|
||
|
|
map_path = os.path.join(data_dir, f'Map{map_id:03}.json')
|
||
|
|
map_ext_path = os.path.join(data_dir, f'Map{map_id:03}Ext.json')
|
||
|
|
with open(map_path, 'r', encoding='utf-8') as f:
|
||
|
|
map_data = json.load(f)
|
||
|
|
|
||
|
|
map_events = map_data['events']
|
||
|
|
map_ext_event_list = []
|
||
|
|
map_event_list = []
|
||
|
|
for event in map_events:
|
||
|
|
if event is not None:
|
||
|
|
note: str = event['note']
|
||
|
|
m = label_pattern.search(note)
|
||
|
|
if m is not None:
|
||
|
|
event["ext_id"] = int(m.group(1)) + event_ext_id_start
|
||
|
|
map_ext_event_list.append(event)
|
||
|
|
map_event_list.append(None)
|
||
|
|
else:
|
||
|
|
map_event_list.append(event)
|
||
|
|
else:
|
||
|
|
map_event_list.append(None)
|
||
|
|
|
||
|
|
map_data['events'] = map_event_list
|
||
|
|
with open(map_path, 'w', encoding='utf-8') as f:
|
||
|
|
json.dump(map_data, f, ensure_ascii=False)
|
||
|
|
|
||
|
|
old_ext_data = None
|
||
|
|
map_ext_data = None
|
||
|
|
|
||
|
|
if os.path.exists(map_ext_path):
|
||
|
|
with open(map_ext_path, 'r', encoding='utf-8') as f:
|
||
|
|
map_ext_data = json.load(f)
|
||
|
|
old_ext_data = map_ext_data['events']
|
||
|
|
else:
|
||
|
|
old_ext_data = {}
|
||
|
|
map_ext_data = {'events': old_ext_data}
|
||
|
|
|
||
|
|
for event in map_ext_event_list:
|
||
|
|
event['id'] = event['ext_id']
|
||
|
|
old_ext_data[event['id']] = event
|
||
|
|
map_ext_data['events'] = old_ext_data
|
||
|
|
with open(map_ext_path, 'w', encoding='utf-8') as f:
|
||
|
|
json.dump(map_ext_data, f, ensure_ascii=False)
|
||
|
|
|
||
|
|
|
||
|
|
map_info_f = open(map_info_path, "r")
|
||
|
|
map_info_json = json.load(map_info_f)
|
||
|
|
map_info_f.close()
|
||
|
|
|
||
|
|
for info in map_info_json:
|
||
|
|
if info is not None:
|
||
|
|
extra_map_event(info['id'])
|