From f8ba6230f03ded7761c640a2465b3b566961fb58 Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Mon, 18 May 2026 15:50:31 -0500
Subject: Call a function `func` with a database connection.

---
 gn_libs/sqlite3.py | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'gn_libs/sqlite3.py')

diff --git a/gn_libs/sqlite3.py b/gn_libs/sqlite3.py
index 78e1c41..565ed82 100644
--- a/gn_libs/sqlite3.py
+++ b/gn_libs/sqlite3.py
@@ -43,3 +43,12 @@ def cursor(conn: DbConnection) -> Iterator[DbCursor]:
         raise exc
     finally:
         cur.close()
+
+
+def with_db_connection(db_uri: str, func: Callable[[DbConnection], Any]) -> Any:
+    """
+    Call `func`, a function of one argument with the SQLite3 connection created
+    from the connection string `db_uri`.
+    """
+    with connection(db_uri) as conn:
+        return func(conn)
-- 
cgit 1.4.1