Fix: Check for common_connection

Signed-off-by: Aryan Rajoria <[email protected]>
AppThreat · Nov 7, 2024 · cdb753c · cdb753c
1 parent eaa46a5
commit cdb753c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/blint_db/cli.py b/blint_db/cli.py
@@ -84,9 +84,9 @@ def arguments_parser():
     return parser.parse_args()
 
 def reset_and_backup():
-    if os.path.exists(BLINTDB_LOCATION) and os.path.isfile(BLINTDB_LOCATION):
-        os.remove(BLINTDB_LOCATION)
     if COMMON_CONNECTION:
+        if os.path.exists(BLINTDB_LOCATION) and os.path.isfile(BLINTDB_LOCATION):
+            os.remove(BLINTDB_LOCATION)
         COMMON_CONNECTION.execute(f"vacuum main into '{BLINTDB_LOCATION}'")