feat(config): ✨ add config file check

AnzhiZhang · Jul 2, 2022 · 23bb19e · 23bb19e
1 parent 10452ef
commit 23bb19e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/utils/config.py b/utils/config.py
@@ -13,6 +13,14 @@ def __init__(self):
             super().__init__(CONFIG.DEFAULT)
             self.save()
 
+        save_flag = False
+        for key, value in CONFIG.DEFAULT.items():
+            if key not in self.keys():
+                self[key] = value
+                save_flag = True
+        if save_flag:
+            self.save()
+
     def save(self):
         """Save data"""
         with open(CONFIG.FILE_PATH, 'w', encoding='utf-8') as f: