diff options
| -rw-r--r-- | gn3/auth/db.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/gn3/auth/db.py b/gn3/auth/db.py index e0e009c..256940d 100644 --- a/gn3/auth/db.py +++ b/gn3/auth/db.py @@ -1,7 +1,7 @@ """Handle connection to auth database.""" import sqlite3 import contextlib -from typing import Any, Iterator, Protocol +from typing import Any, Callable, Iterator, Protocol import traceback @@ -47,9 +47,10 @@ class DbCursor(Protocol): ... @contextlib.contextmanager -def connection(db_path: str) -> Iterator[DbConnection]: +def connection(db_path: str, row_factory: Callable = sqlite3.Row) -> Iterator[DbConnection]: """Create the connection to the auth database.""" conn = sqlite3.connect(db_path) + conn.row_factory = row_factory try: yield conn except sqlite3.Error as exc: |