50abbcee11
* autotranslate python script * 8/52 translated * autotranslate finished * updated autotranslate script * fixes * updated tool * once more updated tool * more translates * more translates * fixes
81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
#FDEV NDA licensed https://assets.station13.ru/LICENSE.txt
|
|
import yaml
|
|
import os
|
|
import re
|
|
|
|
locale_template = "ent-{} = {}\n .desc = {}\n"
|
|
|
|
enitites_path = "../Resources/Prototypes/Entities"
|
|
locales_path = "../Resources/Locale/ru-RU"
|
|
new_locales_path = locales_path+"/locales-new"
|
|
|
|
existing_prototypes = {}
|
|
existing_locales = []
|
|
|
|
#counter for poor ss14 to now break with > 20 locales in one file
|
|
locales = 0
|
|
new_filename = 0
|
|
num = 0
|
|
locale_text = ""
|
|
|
|
#treat custom constructors as text
|
|
def default_ctor(loader, tag_suffix, node):
|
|
return f"{tag_suffix} {node.value}"
|
|
|
|
yaml.add_multi_constructor('', default_ctor)
|
|
|
|
#cache all existing enitites
|
|
for root, dirs, files in os.walk(enitites_path):
|
|
for filename in files:
|
|
# print(f"{root}/{filename}")
|
|
with open(f"{root}/{filename}", "r", encoding="utf-8") as file:
|
|
# text = file.read().encode()
|
|
# text = text.decode("utf-8-sig")
|
|
data = yaml.full_load_all(file)
|
|
if not data: continue
|
|
for entrylist in data:
|
|
for entry in entrylist:
|
|
entry: dict
|
|
if entry["type"] != "entity": continue
|
|
existing_prototypes[entry["id"]] = {"name":entry.get("name", '"name"'), "desc":entry.get("description", '"desc"')}
|
|
|
|
#cache all existing locales
|
|
for root, dirs, files in os.walk(locales_path):
|
|
for filename in files:
|
|
# print(f"{root}/{filename}")
|
|
if "autotranslate-" in filename:
|
|
_num = filename.split("autotranslate-")[1]
|
|
_num = int(_num.split(".")[0])
|
|
if _num >= num:
|
|
num = _num
|
|
new_filename = _num + 1
|
|
with open(f"{root}/{filename}", "r", encoding="utf-8") as file:
|
|
data = re.findall(r'(?<=^ent-)(\w+)', file.read(), re.MULTILINE)
|
|
if not len(data): continue
|
|
for entity in data:
|
|
existing_locales.append(entity)
|
|
|
|
os.makedirs(new_locales_path, exist_ok=True)
|
|
|
|
#create locales for not existing entities
|
|
for entity, entity_data in existing_prototypes.items():
|
|
if entity in existing_locales: continue
|
|
name = entity_data["name"]
|
|
desc = entity_data["desc"]
|
|
locale_text += locale_template.format(entity, name, desc)
|
|
locales += 1
|
|
if locales == 20:
|
|
with open(f"{new_locales_path}/autotranslate-{new_filename}.ftl", "w", encoding="utf-8") as file:
|
|
file.write(locale_text)
|
|
locale_text = ""
|
|
new_filename += 1
|
|
locales = 0
|
|
|
|
#write remaining stuff if we have that
|
|
if locale_text:
|
|
with open(f"{new_locales_path}/autotranslate-{new_filename}.ftl", "w", encoding="utf-8") as file:
|
|
file.write(locale_text)
|
|
new_filename += 1
|
|
|
|
print(f"Wrote {new_filename-num} new files.")
|