Create crud.py

lnbits/extensions/tpos/crud.py

@@ -0,0 +1,44 @@
+from base64 import urlsafe_b64encode
+from uuid import uuid4
+from typing import List, Optional, Union
+
+from lnbits.db import open_ext_db
+
+from .models import TPoS
+
+
+def create_tpos(*, wallet_id: str, name: str, currency: str) -> TPoS:
+    with open_ext_db("tpos") as db:
+        tpos_id = urlsafe_b64encode(uuid4().bytes_le).decode('utf-8')
+        db.execute(
+            """
+            INSERT INTO tposs (id, wallet, name, currency)
+            VALUES (?, ?, ?, ?)
+            """,
+            (tpos_id, wallet_id, name, currency),
+        )
+
+    return get_tpos(tpos_id)
+
+
+def get_tpos(tpos_id: str) -> Optional[TPoS]:
+    with open_ext_db("tpos") as db:
+        row = db.fetchone("SELECT * FROM tposs WHERE id = ?", (tpos_id,))
+
+    return TPoS(**row) if row else None
+
+
+def get_tposs(wallet_ids: Union[str, List[str]]) -> List[TPoS]:
+    if isinstance(wallet_ids, str):
+        wallet_ids = [wallet_ids]
+
+    with open_ext_db("tpos") as db:
+        q = ",".join(["?"] * len(wallet_ids))
+        rows = db.fetchall(f"SELECT * FROM tposs WHERE wallet IN ({q})", (*wallet_ids,))
+
+    return [TPoS(**row) for row in rows]
+
+
+def delete_tpos(tpos_id: str) -> None:
+    with open_ext_db("tpos") as db:
+        db.execute("DELETE FROM tposs WHERE id = ?", (tpos_id,))