laantungir/routstr-core
1
0
Fork 0
mirror of https://github.com/Routstr/routstr-core.git synced 2026-08-09 02:54:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

batch checkstates + lru cache for wallet objects

Browse Source
This commit is contained in:
Shroominic
2025-08-20 10:18:44 -03:00
parent f784a2a36d
commit dbd9b72a23
1 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
routstr/wallet.py
+10 -6
View File
@@ -1,5 +1,6 @@
import asyncio
import os
from functools import lru_cache
from typing import TypedDict
from cashu.core.base import Proof, Token
@@ -162,6 +163,7 @@ async def credit_balance(
raise
@lru_cache(maxsize=10)
async def get_wallet(mint_url: str, unit: str = "sat") -> Wallet:
wallet = await Wallet.with_db(
mint_url, db=".wallet", load_all_keysets=True, unit=unit
@@ -188,14 +190,16 @@ async def get_proofs_per_mint_and_unit(
async def slow_filter_spend_proofs(proofs: list[Proof], wallet: Wallet) -> list[Proof]:
if not proofs:
return []
proof_states = await wallet.check_proof_state(proofs)
_proofs = []
_spent_proofs = []
for proof, state in zip(proofs, proof_states.states):
if str(state.state) != "spent":
_proofs.append(proof)
else:
_spent_proofs.append(proof)
for i in range(0, len(proofs), 1000):
pb = proofs[i : i + 1000]
proof_states = await wallet.check_proof_state(pb)
for proof, state in zip(proofs, proof_states.states):
if str(state.state) != "spent":
_proofs.append(proof)
else:
_spent_proofs.append(proof)
await wallet.set_reserved_for_send(_spent_proofs, reserved=True)
return _proofs