refactor: decorators, models and more broken bits

2020-03-04 23:11:15 +01:00
parent ee70161854
commit f98a5040ac
69 changed files with 52715 additions and 1018 deletions
@@ -0,0 +1,180 @@
+from uuid import uuid4
+
+from lnbits.db import open_db
+from lnbits.settings import DEFAULT_USER_WALLET_NAME, FEE_RESERVE
+from typing import List, Optional
+
+from .models import User, Transaction, Wallet
+
+
+# accounts
+# --------
+
+
+def create_account() -> User:
+    with open_db() as db:
+        user_id = uuid4().hex
+        db.execute("INSERT INTO accounts (id) VALUES (?)", (user_id,))
+
+    return get_account(user_id=user_id)
+
+
+def get_account(user_id: str) -> Optional[User]:
+    with open_db() as db:
+        row = db.fetchone("SELECT id, email, pass as password FROM accounts WHERE id = ?", (user_id,))
+
+    return User(**row) if row else None
+
+
+def get_user(user_id: str) -> Optional[User]:
+    with open_db() as db:
+        user = db.fetchone("SELECT id, email FROM accounts WHERE id = ?", (user_id,))
+
+        if user:
+            extensions = db.fetchall("SELECT extension FROM extensions WHERE user = ? AND active = 1", (user_id,))
+            wallets = db.fetchall(
+                """
+                SELECT *, COALESCE((SELECT balance/1000 FROM balances WHERE wallet = wallets.id), 0) * ? AS balance
+                FROM wallets
+                WHERE user = ?
+                """,
+                (1 - FEE_RESERVE, user_id),
+            )
+
+    return (
+        User(**{**user, **{"extensions": [e[0] for e in extensions], "wallets": [Wallet(**w) for w in wallets]}})
+        if user
+        else None
+    )
+
+
+def update_user_extension(*, user_id: str, extension: str, active: int) -> None:
+    with open_db() as db:
+        db.execute(
+            """
+            INSERT OR REPLACE INTO extensions (user, extension, active)
+            VALUES (?, ?, ?)
+            """,
+            (user_id, extension, active),
+        )
+
+
+# wallets
+# -------
+
+
+def create_wallet(*, user_id: str, wallet_name: Optional[str]) -> Wallet:
+    with open_db() as db:
+        wallet_id = uuid4().hex
+        db.execute(
+            """
+            INSERT INTO wallets (id, name, user, adminkey, inkey)
+            VALUES (?, ?, ?, ?, ?)
+            """,
+            (wallet_id, wallet_name or DEFAULT_USER_WALLET_NAME, user_id, uuid4().hex, uuid4().hex),
+        )
+
+    return get_wallet(wallet_id=wallet_id)
+
+
+def delete_wallet(*, user_id: str, wallet_id: str) -> None:
+    with open_db() as db:
+        db.execute(
+            """
+            UPDATE wallets AS w
+            SET
+                user = 'del:' || w.user,
+                adminkey = 'del:' || w.adminkey,
+                inkey = 'del:' || w.inkey
+            WHERE id = ? AND user = ?
+            """,
+            (wallet_id, user_id),
+        )
+
+
+def get_wallet(wallet_id: str) -> Optional[Wallet]:
+    with open_db() as db:
+        row = db.fetchone(
+            """
+            SELECT *, COALESCE((SELECT balance/1000 FROM balances WHERE wallet = wallets.id), 0) * ? AS balance
+            FROM wallets
+            WHERE id = ?
+            """,
+            (1 - FEE_RESERVE, wallet_id),
+        )
+
+    return Wallet(**row) if row else None
+
+
+def get_wallet_for_key(key: str, key_type: str = "invoice") -> Optional[Wallet]:
+    with open_db() as db:
+        check_field = "adminkey" if key_type == "admin" else "inkey"
+        row = db.fetchone(
+            f"""
+            SELECT *, COALESCE((SELECT balance/1000 FROM balances WHERE wallet = wallets.id), 0) * ? AS balance
+            FROM wallets
+            WHERE {check_field} = ?
+            """,
+            (1 - FEE_RESERVE, key),
+        )
+
+    return Wallet(**row) if row else None
+
+
+# wallet transactions
+# -------------------
+
+
+def get_wallet_transaction(wallet_id: str, payhash: str) -> Optional[Transaction]:
+    with open_db() as db:
+        row = db.fetchone(
+            """
+            SELECT payhash, amount, fee, pending, memo, time
+            FROM apipayments
+            WHERE wallet = ? AND payhash = ?
+            """,
+            (wallet_id, payhash),
+        )
+
+    return Transaction(**row) if row else None
+
+
+def get_wallet_transactions(wallet_id: str, *, pending: bool = False) -> List[Transaction]:
+    with open_db() as db:
+        rows = db.fetchall(
+            """
+            SELECT payhash, amount, fee, pending, memo, time
+            FROM apipayments
+            WHERE wallet = ? AND pending = ?
+            ORDER BY time DESC
+            """,
+            (wallet_id, int(pending)),
+        )
+
+    return [Transaction(**row) for row in rows]
+
+
+# transactions
+# ------------
+
+
+def create_transaction(*, wallet_id: str, payhash: str, amount: str, memo: str) -> Transaction:
+    with open_db() as db:
+        db.execute(
+            """
+            INSERT INTO apipayments (wallet, payhash, amount, pending, memo)
+            VALUES (?, ?, ?, ?, ?)
+            """,
+            (wallet_id, payhash, amount, 1, memo),
+        )
+
+    return get_wallet_transaction(wallet_id, payhash)
+
+
+def update_transaction_status(payhash: str, pending: bool) -> None:
+    with open_db() as db:
+        db.execute("UPDATE apipayments SET pending = ? WHERE payhash = ?", (int(pending), payhash,))
+
+
+def check_pending_transactions(wallet_id: str) -> None:
+    pass