Update

devplacepy/database/core.py

@@ -61,10 +61,11 @@ def _ensure_cache_state() -> None:
     global _cache_state_ready
     if _cache_state_ready:
         return
-    db.query(
-        "CREATE TABLE IF NOT EXISTS cache_state "
-        "(name TEXT PRIMARY KEY, version INTEGER NOT NULL DEFAULT 0)"
-    )
+    with db:
+        db.query(
+            "CREATE TABLE IF NOT EXISTS cache_state "
+            "(name TEXT PRIMARY KEY, version INTEGER NOT NULL DEFAULT 0)"
+        )
     _cache_state_ready = True
 
 
@@ -74,14 +75,15 @@ def get_cache_version(name: str) -> int:
         return cached
     try:
         _ensure_cache_state()
-        row = next(
-            iter(
-                db.query(
-                    "SELECT version FROM cache_state WHERE name = :name", name=name
-                )
-            ),
-            None,
-        )
+        with db:
+            row = next(
+                iter(
+                    db.query(
+                        "SELECT version FROM cache_state WHERE name = :name", name=name
+                    )
+                ),
+                None,
+            )
         version = int(row["version"]) if row else 0
     except Exception as e:
         logger.warning(f"Could not read cache version {name}: {e}")
@@ -93,16 +95,15 @@ def get_cache_version(name: str) -> int:
 def bump_cache_version(name: str) -> None:
     try:
         _ensure_cache_state()
-        db.query(
-            "INSERT OR IGNORE INTO cache_state (name, version) VALUES (:name, 0)",
-            name=name,
-        )
-        db.query(
-            "UPDATE cache_state SET version = version + 1 WHERE name = :name", name=name
-        )
-        connection = db.executable
-        if connection.in_transaction():
-            connection.commit()
+        with db:
+            db.query(
+                "INSERT OR IGNORE INTO cache_state (name, version) VALUES (:name, 0)",
+                name=name,
+            )
+            db.query(
+                "UPDATE cache_state SET version = version + 1 WHERE name = :name",
+                name=name,
+            )
         _cache_version_cache.pop(name)
     except Exception as e:
         logger.warning(f"Could not bump cache version {name}: {e}")