keyset working

cashu/core/crud.py

@@ -0,0 +1,74 @@
+from typing import Optional
+
+from cashu.core.base import Keyset, KeyBase
+from cashu.core.db import Connection, Database
+
+
+async def store_keyset(
+    keyset: Keyset,
+    mint_url: str,
+    db: Database,
+    conn: Optional[Connection] = None,
+):
+
+    await (conn or db).execute(
+        """
+        INSERT INTO keysets
+          (id, mint_url, valid_from, valid_to, first_seen, active)
+        VALUES (?, ?, ?, ?, ?, ?)
+        """,
+        (
+            keyset.id,
+            mint_url,
+            keyset.valid_from,
+            keyset.valid_to,
+            keyset.first_seen,
+            True,
+        ),
+    )
+
+
+async def get_keyset(
+    id: str = None,
+    mint_url: str = None,
+    db: Database = None,
+    conn: Optional[Connection] = None,
+):
+    clauses = []
+    values = []
+    clauses.append("active = ?")
+    values.append(True)
+    if id:
+        clauses.append("id = ?")
+        values.append(id)
+    if mint_url:
+        clauses.append("mint_url = ?")
+        values.append(mint_url)
+    where = ""
+    if clauses:
+        where = f"WHERE {' AND '.join(clauses)}"
+
+    row = await (conn or db).fetchone(
+        f"""
+        SELECT * from keysets
+        {where}
+        """,
+        tuple(values),
+    )
+    return Keyset.from_row(row)
+
+
+async def store_mint_pubkey(
+    key: KeyBase,
+    db: Database,
+    conn: Optional[Connection] = None,
+):
+
+    await (conn or db).execute(
+        """
+        INSERT INTO mint_pubkeys
+          (id, amount, pubkey)
+        VALUES (?, ?, ?)
+        """,
+        (key.id, key.amount, key.pubkey),
+    )