forked from FlandreCirno/AzurLaneWikiUtilities
-
Notifications
You must be signed in to change notification settings - Fork 0
/
util.py
35 lines (32 loc) · 1.29 KB
/
util.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
# -*- coding: utf-8 -*-
import re, os, json
from slpp import slpp
DataDirectory = os.path.join('AzurLaneData', 'zh-CN')
JsonDirectory = 'json'
WikiDirectory = 'Wiki'
def parseDataFile(fileName, filePath = r'sharecfg', mode = 0):
if mode == 0:
filePath = os.path.join(DataDirectory, filePath, fileName + '.lua')
with open(filePath, 'r', encoding='utf-8') as f:
content = f.read()
content = re.match(r".*" + fileName + r" = (\{.*\})", content, flags = re.DOTALL)[1]
output = slpp.decode(content)
if isinstance(output, dict) and 'all' in output.keys():
del output['all']
return output
elif mode == 1:
filePath = os.path.join(DataDirectory, filePath)
output = {}
templateFileNames = os.listdir(filePath)
for fNames in templateFileNames:
with open(os.path.join(filePath, fNames), 'r', encoding='utf-8') as f:
content = f.read()
content = re.match(r".*" + fileName + r"_\d+ = (\{.*\})", content, flags = re.DOTALL)[1]
o = slpp.decode(content)
for k, v in o.items():
output[k] = v
if isinstance(output, dict) and 'all' in output.keys():
del output['all']
return output
if __name__ == "__main__":
pass