sqlite: use lastrowid instead of RETURNING id

RETURNING was introduced quite recently and is not available in all sqlite versions.
iterative · Jan 31, 2023 · f22402d · f22402d
1 parent b5997ed
commit f22402d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -20,7 +20,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        os: [ubuntu-22.04, windows-latest, macos-latest]
+        os: [ubuntu-20.04, windows-latest, macos-latest]
         pyv: ['3.8', '3.9', '3.10', '3.11']
 
     steps:

diff --git a/src/sqltrie/sqlite/sqlite.py b/src/sqltrie/sqlite/sqlite.py
@@ -98,16 +98,16 @@ def _create_node(self, key):
         node_key = longest_prefix
         for name in key[len(longest_prefix) :]:
             node_key = (*node_key, name)
-            row = self._conn.execute(
+            cur = self._conn.execute(
                 """
                 INSERT OR IGNORE
                     INTO nodes (pid, name)
                     VALUES (?, ?)
                     RETURNING id
                 """,
                 (pid, name),
-            ).fetchone()
-            nid = row["id"]
+            )
+            nid = cur.lastrowid
             self._ids[node_key] = nid
             pid = nid