fix: bug if 'temporary.txt' does not exist

Mazars-Tech · Oct 3, 2023 · 2ed8326 · 2ed8326
1 parent 7b75a3d
commit 2ed8326
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ad_miner/__main__.py b/ad_miner/__main__.py
@@ -7,6 +7,7 @@
 from pathlib import Path
 import time
 import traceback
+import os
 
 # Local library imports
 from ad_miner.sources.modules import logger, main_page, render_order, utils
@@ -184,7 +185,8 @@ def main() -> None:
 
     neo4j.close()
 
-    (Path().cwd() / 'temporary.txt').unlink()
+    if os.path.exists(Path().cwd() / 'temporary.txt'):
+        (Path().cwd() / 'temporary.txt').unlink()
 
     logger.print_success(f"Program finished in {utils.timer_format(time.time() - start)}!")