diff --git a/src/wxflow/sqlitedb.py b/src/wxflow/sqlitedb.py index 7771964..ea071e2 100644 --- a/src/wxflow/sqlitedb.py +++ b/src/wxflow/sqlitedb.py @@ -3,6 +3,12 @@ __all__ = ["SQLiteDB"] +class SQLiteDBError(Exception): + """ + Base class for SQLiteDB exceptions. + """ + OperationalError = sqlite3.OperationalError + IntegrityError = sqlite3.IntegrityError class SQLiteDB: """ @@ -17,8 +23,7 @@ class SQLiteDB: """ - OperationalError = sqlite3.OperationalError - IntegrityError = sqlite3.IntegrityError + Error = SQLiteDBError def __init__(self, db_name: str) -> None: self.db_name = db_name