77 lines
1.9 KiB
Python
77 lines
1.9 KiB
Python
import ast
|
|
import glob
|
|
import json
|
|
|
|
from collections import OrderedDict
|
|
|
|
|
|
def extract_i18n_strings(node):
|
|
i18n_strings = []
|
|
|
|
if (
|
|
isinstance(node, ast.Call)
|
|
and isinstance(node.func, ast.Name)
|
|
and node.func.id == "i18n"
|
|
):
|
|
for arg in node.args:
|
|
if isinstance(arg, ast.Str):
|
|
i18n_strings.append(arg.s)
|
|
|
|
for child_node in ast.iter_child_nodes(node):
|
|
i18n_strings.extend(extract_i18n_strings(child_node))
|
|
|
|
return i18n_strings
|
|
|
|
|
|
# scan the directory for all .py files (recursively)
|
|
# for each file, parse the code into an AST
|
|
# for each AST, extract the i18n strings
|
|
|
|
strings = []
|
|
for filename in glob.iglob("**/*.py", recursive=True):
|
|
with open(filename, "r") as f:
|
|
code = f.read()
|
|
if "I18nAuto" in code:
|
|
tree = ast.parse(code)
|
|
i18n_strings = extract_i18n_strings(tree)
|
|
print(filename, len(i18n_strings))
|
|
strings.extend(i18n_strings)
|
|
code_keys = set(strings)
|
|
"""
|
|
n_i18n.py
|
|
gui_v1.py 26
|
|
app.py 16
|
|
infer-web.py 147
|
|
scan_i18n.py 0
|
|
i18n.py 0
|
|
lib/train/process_ckpt.py 1
|
|
"""
|
|
print()
|
|
print("Total unique:", len(code_keys))
|
|
|
|
|
|
standard_file = "zh_CN.json"
|
|
with open(f"lib/i18n/{standard_file}", "r", encoding="utf-8") as f:
|
|
standard_data = json.load(f, object_pairs_hook=OrderedDict)
|
|
standard_keys = set(standard_data.keys())
|
|
|
|
# Define the standard file name
|
|
unused_keys = standard_keys - code_keys
|
|
print("Unused keys:", len(unused_keys))
|
|
for unused_key in unused_keys:
|
|
print("\t", unused_key)
|
|
|
|
missing_keys = code_keys - standard_keys
|
|
print("Missing keys:", len(missing_keys))
|
|
for missing_key in missing_keys:
|
|
print("\t", missing_key)
|
|
|
|
code_keys_dict = OrderedDict()
|
|
for s in strings:
|
|
code_keys_dict[s] = s
|
|
|
|
# write back
|
|
with open(f"lib/i18n/{standard_file}", "w", encoding="utf-8") as f:
|
|
json.dump(code_keys_dict, f, ensure_ascii=False, indent=4, sort_keys=True)
|
|
f.write("\n")
|