catch missing database file when running 'test' instead of crashing

2014-08-10 18:03:28 +02:00 · 2014-08-10 18:03:28 +02:00 · a6e1bb9b4c
commit a6e1bb9b4c
parent e5f737b09d
1 changed files with 5 additions and 1 deletions
--- a/src/bitrot.py
+++ b/src/bitrot.py
@ -94,7 +94,11 @@ def run(verbosity=1, test=False, follow_links=False, commit_interval=300,
        chunk_size=DEFAULT_CHUNK_SIZE):
    current_dir = b'.'   # sic, relative path
    bitrot_db = os.path.join(current_dir, b'.bitrot.db')
-    conn = get_sqlite3_cursor(bitrot_db, copy=test)
+    try:
+        conn = get_sqlite3_cursor(bitrot_db, copy=test)
+    except ValueError:
+        print('No database exists so cannot test. Run the tool once first.')
+        sys.exit(2)
    cur = conn.cursor()
    new_paths = []
    updated_paths = []