init commit

2023-02-17 11:57:27 +01:00 · 2023-02-17 11:57:27 +01:00 · fcb973b540
commit fcb973b540
21 changed files with 2267 additions and 0 deletions
--- a/migrations.py
+++ b/migrations.py
@ -0,0 +1,134 @@
+async def m001_initial(db):
+    """
+    Creates an improved withdraw table and migrates the existing data.
+    """
+    await db.execute(
+        f"""
+        CREATE TABLE withdraw.withdraw_links (
+            id TEXT PRIMARY KEY,
+            wallet TEXT,
+            title TEXT,
+            min_withdrawable {db.big_int} DEFAULT 1,
+            max_withdrawable {db.big_int} DEFAULT 1,
+            uses INTEGER DEFAULT 1,
+            wait_time INTEGER,
+            is_unique INTEGER DEFAULT 0,
+            unique_hash TEXT UNIQUE,
+            k1 TEXT,
+            open_time INTEGER,
+            used INTEGER DEFAULT 0,
+            usescsv TEXT
+        );
+    """
+    )
+
+
+async def m002_change_withdraw_table(db):
+    """
+    Creates an improved withdraw table and migrates the existing data.
+    """
+    await db.execute(
+        f"""
+        CREATE TABLE withdraw.withdraw_link (
+            id TEXT PRIMARY KEY,
+            wallet TEXT,
+            title TEXT,
+            min_withdrawable {db.big_int} DEFAULT 1,
+            max_withdrawable {db.big_int} DEFAULT 1,
+            uses INTEGER DEFAULT 1,
+            wait_time INTEGER,
+            is_unique INTEGER DEFAULT 0,
+            unique_hash TEXT UNIQUE,
+            k1 TEXT,
+            open_time INTEGER,
+            used INTEGER DEFAULT 0,
+            usescsv TEXT
+        );
+        """
+    )
+
+    for row in [
+        list(row) for row in await db.fetchall("SELECT * FROM withdraw.withdraw_links")
+    ]:
+        usescsv = ""
+
+        for i in range(row[5]):
+            if row[7]:
+                usescsv += "," + str(i + 1)
+            else:
+                usescsv += "," + str(1)
+        usescsv = usescsv[1:]
+        await db.execute(
+            """
+            INSERT INTO withdraw.withdraw_link (
+                id,
+                wallet,
+                title,
+                min_withdrawable,
+                max_withdrawable,
+                uses,
+                wait_time,
+                is_unique,
+                unique_hash,
+                k1,
+                open_time,
+                used,
+                usescsv
+            )
+            VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+            """,
+            (
+                row[0],
+                row[1],
+                row[2],
+                row[3],
+                row[4],
+                row[5],
+                row[6],
+                row[7],
+                row[8],
+                row[9],
+                row[10],
+                row[11],
+                usescsv,
+            ),
+        )
+    await db.execute("DROP TABLE withdraw.withdraw_links")
+
+
+async def m003_make_hash_check(db):
+    """
+    Creates a hash check table.
+    """
+    await db.execute(
+        """
+        CREATE TABLE withdraw.hash_check (
+            id TEXT PRIMARY KEY,
+            lnurl_id TEXT
+        );
+    """
+    )
+
+
+async def m004_webhook_url(db):
+    """
+    Adds webhook_url
+    """
+    await db.execute("ALTER TABLE withdraw.withdraw_link ADD COLUMN webhook_url TEXT;")
+
+
+async def m005_add_custom_print_design(db):
+    """
+    Adds custom print design
+    """
+    await db.execute("ALTER TABLE withdraw.withdraw_link ADD COLUMN custom_url TEXT;")
+
+
+async def m006_webhook_headers_and_body(db):
+    """
+    Add headers and body to webhooks
+    """
+    await db.execute(
+        "ALTER TABLE withdraw.withdraw_link ADD COLUMN webhook_headers TEXT;"
+    )
+    await db.execute("ALTER TABLE withdraw.withdraw_link ADD COLUMN webhook_body TEXT;")