Cura/scripts/lionbridge_import.py
2019-07-26 13:48:10 +02:00

68 lines
3.3 KiB
Python

# Copyright (c) 2019 Ultimaker B.V.
# Cura is released under the terms of the LGPLv3 or higher.
import argparse #To get the source directory from command line arguments.
import os #To find files from the source.
import os.path #To find files from the source and the destination path.
cura_files = {"cura", "fdmprinter.def.json", "fdmextruder.def.json"}
uranium_files = {"uranium"}
## Imports translation files from Lionbridge.
#
# Lionbridge has a bit of a weird export feature. It exports it to the same
# file type as what we imported, so that's a .pot file. However this .pot file
# only contains the translations, so the header is completely empty. We need
# to merge those translations into our existing files so that the header is
# preserved.
def lionbridge_import(source: str) -> None:
print("Importing from:", source)
print("Importing to Cura:", destination_cura())
print("Importing to Uranium:", destination_uranium())
for language in (directory for directory in os.listdir(source) if os.path.isdir(os.path.join(source, directory))):
print("================ Processing language:", language, "================")
directory = os.path.join(source, language)
for file_pot in (file for file in os.listdir(directory) if file.endswith(".pot")):
source_file = file_pot[:-4] #Strip extension.
if source_file in cura_files:
destination_file = os.path.join(destination_cura(), language.replace("-", "_"), source_file + ".po")
print("Merging", source_file, "(Cura) into", destination_file)
elif source_file in uranium_files:
destination_file = os.path.join(destination_uranium(), language.replace("-", "_"), source_file + ".po")
print("Merging", source_file, "(Uranium) into", destination_file)
else:
raise Exception("Unknown file: " + source_file + "... Is this Cura or Uranium?")
with open(os.path.join(directory, file_pot)) as f:
source_str = f.read()
with open(destination_file) as f:
destination_str = f.read()
result = merge(source_str, destination_str)
print(result) #DEBUG! Instead we should write this to a file.
## Gets the destination path to copy the translations for Cura to.
def destination_cura() -> str:
return os.path.abspath(os.path.join(__file__, "..", "..", "resources", "i18n"))
## Gets the destination path to copy the translations for Uranium to.
def destination_uranium() -> str:
try:
import UM
except ImportError:
relative_path = os.path.join(__file__, "..", "..", "..", "Uranium", "resources", "i18n", "uranium.pot")
print(os.path.abspath(relative_path))
if os.path.exists(relative_path):
return os.path.abspath(relative_path)
else:
raise Exception("Can't find Uranium. Please put UM on the PYTHONPATH or put the Uranium folder next to the Cura folder.")
return os.path.abspath(os.path.join(UM.__file__, "..", "..", "resources", "i18n"))
def merge(source: str, destination: str) -> str:
return "TODO"
if __name__ == "__main__":
argparser = argparse.ArgumentParser(description = "Import translation files from Lionbridge.")
argparser.add_argument("source")
args = argparser.parse_args()
lionbridge_import(args.source)