Add some error handling in rules update feature

wagga40 · Aug 17, 2022 · 387bc71 · 387bc71
1 parent 0f0eb78
commit 387bc71
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 4 deletions.
diff --git a/docs/Zircolite_manual.pdf b/docs/Zircolite_manual.pdf
diff --git a/zircolite.py b/zircolite.py
@@ -1264,11 +1264,16 @@ def checkIfNewerAndMove(self):
         rulesets = Path(self.tmpDir).rglob("*.json")
         for ruleset in rulesets:
             hash_new = hashlib.md5(open(ruleset, "rb").read()).hexdigest()
-            hash_old = hashlib.md5(
-                open(f"rules/{ruleset.name}", "rb").read()
-            ).hexdigest()
+            if Path(f"rules/{ruleset.name}").is_file():
+                hash_old = hashlib.md5(
+                    open(f"rules/{ruleset.name}", "rb").read()
+                ).hexdigest()
+            else:
+                hash_old = ""
             if hash_new != hash_old:
                 count += 1
+                if not Path(f"rules/").exists():
+                    Path(f"rules/").mkdir()
                 shutil.move(ruleset, f"rules/{ruleset.name}")
                 self.updatedRulesets.append(f"rules/{ruleset.name}")
                 self.logger.info(

diff --git a/zircolite_dev.py b/zircolite_dev.py
@@ -889,9 +889,13 @@ def checkIfNewerAndMove(self):
         rulesets = Path(self.tmpDir).rglob("*.json")
         for ruleset in rulesets:
             hash_new = hashlib.md5(open(ruleset,'rb').read()).hexdigest()
-            hash_old = hashlib.md5(open(f'rules/{ruleset.name}','rb').read()).hexdigest()
+            if Path(f'rules/{ruleset.name}').is_file():
+                hash_old = hashlib.md5(open(f'rules/{ruleset.name}','rb').read()).hexdigest()
+            else: hash_old = ""
             if hash_new != hash_old:
                 count += 1
+                if not Path(f'rules/').exists():
+                    Path(f'rules/').mkdir()
                 shutil.move(ruleset, f'rules/{ruleset.name}')
                 self.updatedRulesets.append(f'rules/{ruleset.name}')
                 self.logger.info(f"{Fore.CYAN}   [+] Updated : rules/{ruleset.name}{Fore.RESET}")