Merge branch 'eason/main' of https://github.com/project-kxkg/project-t into eason/main

dict_util.py

@@ -7,11 +7,12 @@ import pickle
 # 1_2_3, 1 is action, 2 is supply object, 3 is accept object
 def update_dict_csv(term_dict, f):
     for rows in csv.reader(f):
-        if rows[0] in term_dict:
-            if rows[1] not in term_dict[rows[0]]:
-                term_dict[rows[0]] = term_dict[rows[0]]+[rows[1]]
+        word = rows[0].lower()
+        if word in term_dict:
+            if rows[1] not in term_dict[word]:
+                term_dict[word] = term_dict[word]+[rows[1]]
         else:
-            term_dict[rows[0]]=[rows[1]]
+            term_dict[word]=[rows[1]]
     pass
 
 def export_dict_csv(term_dict, f):
@@ -21,17 +22,14 @@ def export_dict_csv(term_dict, f):
 
 def save_dict_pickle(term_dict, f):
     pickle.dump(term_dict, f, pickle.HIGHEST_PROTOCOL)
+    pass
 
 def update_csv_pickle(pickle_f, csv_f):
     term_dict = pickle.load(pickle_f)
-    for rows in csv.reader(csv_f):
-        if rows[0] in term_dict:
-            if rows[1] not in term_dict[rows[0]]:
-                term_dict[rows[0]] = term_dict[rows[0]]+[rows[1]]
-        else:
-            term_dict[rows[0]]=[rows[1]]
+    update_dict_csv(term_dict, csv_f)
     #save to pickle file, highest protocal to get better performance
     pickle.dump(term_dict, pickle_f, pickle.HIGHEST_PROTOCOL)
+    pass
 
 
 #demo