Initial Commit

localizer.py

@@ -0,0 +1,78 @@
+#!/usr/bin/env python3
+
+import json
+import argparse
+
+
+class Colors:
+    HEADER = '\033[95m'
+    OKBLUE = '\033[94m'
+    OKCYAN = '\033[96m'
+    OKGREEN = '\033[92m'
+    WARNING = '\033[93m'
+    FAIL = '\033[91m'
+    ENDC = '\033[0m'
+    BOLD = '\033[1m'
+    UNDERLINE = '\033[4m'
+
+    def color_string(self, message):
+        return f"{self.OKCYAN}{message}{self.ENDC}"
+
+
+class Localizer:
+    def __init__(self, reference_path, working_path):
+        self.reference_dict = dict()
+        self.working_dict = dict()
+        self.out_dict = dict()
+        self.reference_path = reference_path
+        self.working_path = working_path
+        self.init_dicts()
+
+    def init_dicts(self):
+        with open(self.reference_path, encoding="utf8") as reference_file:
+            self.reference_dict = json.load(reference_file)
+        with open(self.working_path, encoding="utf8") as working_file:
+            self.working_dict = json.load(working_file)
+
+    def get_new_dict(self):
+        self.out_dict = self.update_values(self.reference_dict, self.working_dict)[0]
+
+    def update_values(self, reference_dict, working_dict):
+        colors = Colors()
+        done = False
+        for key, value in reference_dict.items():
+            if done:
+                return (working_dict, True)
+            if isinstance(value, dict):
+                if key not in working_dict:
+                    working_dict[key] = dict()
+                working_dict[key], done = self.update_values(
+                    value, working_dict[key])
+            elif key not in working_dict:
+                working_val = input(
+                    f"Translation for key: [{key}] was not found.\nType in the translation in the field below.\nType \"_exit\" to save & quit and \"_skip\" to skip to the next key.\n<<< {colors.color_string(repr(value)[1:-1])}\n>>> ")
+                if working_val == "_exit":
+                    return (working_dict, True)
+                elif working_val == "_skip":
+                    continue
+                working_dict[key] = working_val
+        return (working_dict, False)
+
+    def update_file(self):
+        self.get_new_dict()
+        with open(self.working_path, 'w') as working_file:
+            json.dump(self.out_dict, working_file, indent=2, ensure_ascii=False)
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser(
+        description="Compare and update JSON translation files")
+    required = parser.add_argument_group('Required arguments')
+    required.add_argument('-r', '--reference',
+                          help='reference file', required=True)
+    required.add_argument(
+        '-w', '--working', help='working file, will be edited', required=True)
+    args = parser.parse_args()
+
+    loc = Localizer(args.reference, args.working)
+    loc.update_file()