From e90c5d9b52ef1115473947315147937b66034433 Mon Sep 17 00:00:00 2001 From: tmcguinness Date: Fri, 8 Mar 2024 17:40:17 +0000 Subject: [PATCH] added exception base class to SQLiteDB --- src/wxflow/sqlitedb.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) 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