about summary refs log tree commit diff
path: root/gn3/auth/db.py
diff options
context:
space:
mode:
authorJohn Nduli2024-10-18 15:06:36 +0300
committerFrederick Muriuki Muriithi2024-10-18 09:28:40 -0500
commit86b81be8c67cf82e8f26a7f84cbdc5df7198e214 (patch)
tree7e27fe0d5876d6c473b61bcacf5e73a9a8d83a39 /gn3/auth/db.py
parent27531c0a1350097ca36a555b8546bc282948de04 (diff)
downloadgenenetwork3-86b81be8c67cf82e8f26a7f84cbdc5df7198e214.tar.gz
refactor: replace gn3.auth.db with gn3.sqlite_db_utils and drop all refs to gn3.auth
Diffstat (limited to 'gn3/auth/db.py')
-rw-r--r--gn3/auth/db.py72
1 files changed, 0 insertions, 72 deletions
diff --git a/gn3/auth/db.py b/gn3/auth/db.py
deleted file mode 100644
index 5cd230f..0000000
--- a/gn3/auth/db.py
+++ /dev/null
@@ -1,72 +0,0 @@
-"""Handle connection to auth database."""
-import sqlite3
-import logging
-import contextlib
-from typing import Any, Callable, Iterator, Protocol
-
-import traceback
-
-class DbConnection(Protocol):
-    """Type annotation for a generic database connection object."""
-    def cursor(self) -> Any:
-        """A cursor object"""
-
-    def commit(self) -> Any:
-        """Commit the transaction."""
-
-    def rollback(self) -> Any:
-        """Rollback the transaction."""
-
-
-class DbCursor(Protocol):
-    """Type annotation for a generic database cursor object."""
-    def execute(self, *args, **kwargs) -> Any:
-        """Execute a single query"""
-
-    def executemany(self, *args, **kwargs) -> Any:
-        """
-        Execute parameterized SQL statement sql against all parameter sequences
-        or mappings found in the sequence parameters.
-        """
-
-    def fetchone(self, *args, **kwargs):
-        """Fetch single result if present, or `None`."""
-
-    def fetchmany(self, *args, **kwargs):
-        """Fetch many results if present or `None`."""
-
-    def fetchall(self, *args, **kwargs):
-        """Fetch all results if present or `None`."""
-
-
-@contextlib.contextmanager
-def connection(db_path: str, row_factory: Callable = sqlite3.Row) -> Iterator[DbConnection]:
-    """Create the connection to the auth database."""
-    logging.debug("SQLite3 DB Path: '%s'.", db_path)
-    conn = sqlite3.connect(db_path)
-    conn.row_factory = row_factory
-    conn.set_trace_callback(logging.debug)
-    conn.execute("PRAGMA foreign_keys = ON")
-    try:
-        yield conn
-    except sqlite3.Error as exc:
-        conn.rollback()
-        logging.debug(traceback.format_exc())
-        raise exc
-    finally:
-        conn.commit()
-        conn.close()
-
-@contextlib.contextmanager
-def cursor(conn: DbConnection) -> Iterator[DbCursor]:
-    """Get a cursor from the given connection to the auth database."""
-    cur = conn.cursor()
-    try:
-        yield cur
-    except sqlite3.Error as exc:
-        conn.rollback()
-        logging.debug(traceback.format_exc())
-        raise exc
-    finally:
-        conn.commit()
-        cur.close()