Improve validating database before import

This commit is contained in:
JonnyWong16 2020-05-02 23:14:17 -07:00
parent 15a638b86e
commit a869859491
No known key found for this signature in database
GPG key ID: B1F1F9807184697A

View file

@ -39,10 +39,17 @@ db_lock = threading.Lock()
def validate_database(database=None):
try:
connection = sqlite3.connect(database, timeout=20)
except sqlite3.OperationalError as e:
except (sqlite3.OperationalError, sqlite3.DatabaseError, ValueError) as e:
logger.error("Tautulli Database :: Invalid database specified: %s", e)
return 'Invalid database specified'
except ValueError as e:
except Exception as e:
logger.error("Tautulli Database :: Uncaught exception: %s", e)
return 'Uncaught exception'
try:
connection.execute('SELECT started from session_history')
connection.close()
except (sqlite3.OperationalError, sqlite3.DatabaseError, ValueError) as e:
logger.error("Tautulli Database :: Invalid database specified: %s", e)
return 'Invalid database specified'
except Exception as e: