added exception base class to SQLiteDB

NOAA-EMC · Mar 8, 2024 · e90c5d9 · e90c5d9
1 parent 7e3c335
commit e90c5d9
Showing 1 changed file with 7 additions and 2 deletions.
diff --git 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