better db file checking

ktmeaton · May 7, 2020 · 9ceeef2 · 9ceeef2
1 parent 22e5fb5
commit 9ceeef2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/scripts/sqlite_EAGER_tsv.py b/scripts/sqlite_EAGER_tsv.py
@@ -75,13 +75,16 @@
 #------------------------------------------------------------------------------#
 
 # Check if DATABASE file exists
+if not os.path.exists(db_path):
+    print('An error occurred while trying to open ', db_path)
+    sys.exit(1)
+
 try:
     conn = sqlite3.connect(db_path)
     cur = conn.cursor()
 except IOError:
     print('An error occurred while trying to open ', db_path)
     sys.exit(1)
-
 # Open the output file
 out_file = open(out_path, 'w')
 

diff --git a/scripts/sqlite_NextStrain_tsv.py b/scripts/sqlite_NextStrain_tsv.py
@@ -59,6 +59,10 @@
 #------------------------------------------------------------------------------#
 
 # Check if DATABASE file exists
+if not os.path.exists(db_path):
+    print('An error occurred while trying to open ', db_path)
+    sys.exit(1)
+
 try:
     conn = sqlite3.connect(db_path)
     cur = conn.cursor()