29 lines
763 B
Python
29 lines
763 B
Python
|
import json
|
||
|
|
import re
|
||
|
|
|
||
|
|
# Define regular expression patterns
|
||
|
|
pattern = r'i18n\([^)]*\)'
|
||
|
|
|
||
|
|
# Initialize the dictionary to store key-value pairs
|
||
|
|
data = {}
|
||
|
|
|
||
|
|
# Extract labels from infer-webui.py
|
||
|
|
with open('infer-web.py', 'r', encoding='utf-8') as f:
|
||
|
|
contents = f.read()
|
||
|
|
matches = re.findall(pattern, contents)
|
||
|
|
for match in matches:
|
||
|
|
key = match.strip('()"')
|
||
|
|
data[key] = key
|
||
|
|
|
||
|
|
# Extract labels from gui.py
|
||
|
|
with open('gui.py', 'r', encoding='utf-8') as f:
|
||
|
|
contents = f.read()
|
||
|
|
matches = re.findall(pattern, contents)
|
||
|
|
for match in matches:
|
||
|
|
key = match.strip('()"')
|
||
|
|
data[key] = key
|
||
|
|
|
||
|
|
# Save as a JSON file
|
||
|
|
with open('./locale/zh_CN.json', 'w', encoding='utf-8') as f:
|
||
|
|
json.dump(data, f, ensure_ascii=False, indent=4)
|