rc5 3

lnbits · Oct 18, 2024 · cab62b5 · cab62b5
1 parent 8394e56
commit cab62b5
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 29 deletions.
diff --git a/crud.py b/crud.py
@@ -78,14 +78,15 @@ async def get_withdraw_links(
         WithdrawLink,  # type: ignore
     )
 
-    total = await db.fetchone(
+    result = await db.execute(
         f"""
         SELECT COUNT(*) as total FROM withdraw.withdraw_link
         WHERE wallet IN ({q})
         """
     )
+    total = await result.mappings().first()
 
-    return links, total["total"]
+    return links, total.total
 
 
 async def remove_unique_withdraw_link(link: WithdrawLink, unique_hash: str) -> None:

diff --git a/poetry.lock b/poetry.lock
diff --git a/views_api.py b/views_api.py
@@ -3,7 +3,7 @@
 from typing import Optional
 
 from fastapi import APIRouter, Depends, HTTPException, Query, Request
-from lnbits.core.crud import get_user_by_id
+from lnbits.core.crud import get_user
 from lnbits.core.models import WalletTypeInfo
 from lnbits.decorators import require_admin_key, require_invoice_key
 from lnurl.exceptions import InvalidUrl as LnurlInvalidUrl
@@ -32,7 +32,7 @@ async def api_links(
     wallet_ids = [key_info.wallet.id]
 
     if all_wallets:
-        user = await get_user_by_id(key_info.wallet.user)
+        user = await get_user(key_info.wallet.user)
         wallet_ids = user.wallet_ids if user else []
 
     try: