fix integer indexing of row fields

lollerfirst · Oct 3, 2024 · a152df2 · a152df2
1 parent 130abae
commit a152df2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/cashu/mint/crud.py b/cashu/mint/crud.py
@@ -677,7 +677,10 @@ async def get_balance(
             """
         )
         assert row, "Balance not found"
-        return int(row[0])
+
+        # Thank SQLAlchemy for this!
+        key = next(iter(row))
+        return int(row[key])
 
     async def get_keyset(
         self,

diff --git a/cashu/wallet/crud.py b/cashu/wallet/crud.py
@@ -377,7 +377,7 @@ async def bump_secret_derivation(
         )
         counter = 0
     else:
-        counter = int(rows[0])
+        counter = int(rows["counter"])
 
     if not skip:
         await (conn or db).execute(
@@ -437,8 +437,8 @@ async def get_seed_and_mnemonic(
     )
     return (
         (
-            row[0],
-            row[1],
+            row["seed"],
+            row["mnemonic"],
         )
         if row
         else None