prune and increase robustness

Former-commit-id: 3c421a28145741c17e1a7d09339b205da541996e

dict_util.py

@@ -55,12 +55,8 @@ with open("../test.csv", "w", encoding='utf-8') as w:
 # pickle.load(f)
 
 
-def form_dict(src_path,tgt_path) -> dict:
+def form_dict(src_dict:list, tgt_dict:list) -> dict:
     final_dict = {}
-    with open(src_path, 'r', encoding="utf-8") as file:
-        src_dict = list(csv.reader(file, delimiter=","))
-    with open(tgt_path, 'r', encoding="utf-8") as file:
-        tgt_dict = list(csv.reader(file, delimiter="," ))
     for idx, value in enumerate(src_dict):
         for item in value:
             final_dict.update({item:tgt_dict[idx]})
@@ -69,8 +65,10 @@ def form_dict(src_path,tgt_path) -> dict:
 
 class term_dict(dict):
     def __init__(self, path, src_lang, tgt_lang) -> None:
-        src_dict = f"{path}/{src_lang}.csv"
-        tgt_dict = f"{path}/{tgt_lang}.csv"
+        with open(f"{path}/{src_lang}.csv", 'r', encoding="utf-8") as file:
+            src_dict = list(csv.reader(file, delimiter=",")) 
+        with open(f"{path}/{tgt_lang}.csv", 'r', encoding="utf-8") as file:
+            tgt_dict = list(csv.reader(file, delimiter="," ))
         super().__init__(form_dict(src_dict, tgt_dict))
 
 

src/srt_util/srt.py

@@ -160,13 +160,13 @@ class SrtScript(object):
         self.segments = [SrtSegment(self.src_lang, self.tgt_lang, seg) for seg in segments]
 
         if self.domain != "General":
-            if os.path.exists(f"{dict_path}/{self.domain}"):
+            if os.path.exists(f"{dict_path}/{self.domain}") and\
+                              os.path.exists(f"{dict_path}/{self.domain}/{src_lang}.csv") and os.path.exists(f"{dict_path}/{self.domain}/{tgt_lang}.csv" ):
                 # TODO: load dictionary
                 self.dict = dict_util.term_dict(f"{dict_path}/{self.domain}", src_lang, tgt_lang)
-                print(self.dict["robo"])
                 ...
             else:
-                logging.error(f"domain {self.domain} doesn't exist, fallback to general domain, this will disable correct_with_force_term and spell_check_term")
+                logging.error(f"domain {self.domain} or related dictionary({src_lang} or {tgt_lang}) doesn't exist, fallback to general domain, this will disable correct_with_force_term and spell_check_term")
                 self.domain = "General"