DRY & clean up sqlite row_factory (#66)

sqlite_export_for_ynab/_main.py

@@ -100,9 +100,7 @@ async def sync(token: str, db: Path, full_refresh: bool) -> None:
         db.parent.mkdir(parents=True, exist_ok=True)
 
     with sqlite3.connect(db) as con:
-        con.row_factory = lambda c, row: dict(
-            zip([name for name, *_ in c.description], row, strict=True)
-        )
+        con.row_factory = _row_factory
         cur = con.cursor()
 
         if full_refresh:
@@ -177,6 +175,10 @@ async def sync(token: str, db: Path, full_refresh: bool) -> None:
             print("Done")
 
 
+def _row_factory(c: sqlite3.Cursor, row: tuple[Any, ...]) -> dict[str, Any]:
+    return {d[0]: r for d, r in zip(c.description, row, strict=True)}
+
+
 def contents(filename: str) -> str:
     return (resources.files(ddl) / filename).read_text()
 

testing/fixtures.py

@@ -8,6 +8,7 @@
 import pytest
 from aioresponses import aioresponses
 
+from sqlite_export_for_ynab._main import _row_factory
 from sqlite_export_for_ynab._main import contents
 
 BUDGET_ID_1 = str(uuid4())
@@ -186,11 +187,9 @@
 @pytest.fixture
 def cur():
     with sqlite3.connect(":memory:") as con:
+        con.row_factory = _row_factory
         cursor = con.cursor()
         cursor.executescript(contents("create-relations.sql"))
-        cursor.row_factory = lambda c, row: dict(
-            zip([name for name, *_ in c.description], row, strict=True)
-        )
         yield cursor