Add deleted flag wallet (#1826)

* add deleted flag on wallets set deleted on delete wallet 2 twelves format fail on create invoice make deleted check on SQL query nazi flake8 add_test boom... it works and passes!! * add app fixture vlad's recommendations add deleted * Add deleted flag to Wallet * restore crud * do not check for wallet in services.py * add deleted flag on wallets set deleted on delete wallet 2 twelves format fail on create invoice make deleted check on SQL query nazi flake8 add_test boom... it works and passes!! * add app fixture vlad's recommendations * add deleted * error checks --------- Co-authored-by: callebtc <93376500+callebtc@users.noreply.github.com>
2023-09-11 14:06:31 +01:00
parent 6773a0f533
commit 576e20d0cd
8 changed files with 88 additions and 10 deletions
@@ -327,3 +327,54 @@ async def m012_add_currency_to_wallet(db):
        ALTER TABLE wallets ADD COLUMN currency TEXT
        """
    )
+
+
+async def m013_add_deleted_to_wallets(db):
+    """
+    Adds deleted column to wallets.
+    """
+    try:
+        await db.execute(
+            "ALTER TABLE wallets ADD COLUMN deleted BOOLEAN NOT NULL DEFAULT false"
+        )
+    except OperationalError:
+        pass
+
+
+async def m014_set_deleted_wallets(db):
+    """
+    Sets deleted column to wallets.
+    """
+    try:
+        rows = await (
+            await db.execute(
+                """
+                SELECT *
+                FROM wallets
+                WHERE user LIKE 'del:%'
+                AND adminkey LIKE 'del:%'
+                AND inkey LIKE 'del:%'
+                """
+            )
+        ).fetchall()
+
+        for row in rows:
+            try:
+                user = row[2].split(":")[1]
+                adminkey = row[3].split(":")[1]
+                inkey = row[4].split(":")[1]
+                await db.execute(
+                    """
+                    UPDATE wallets SET user = ?, adminkey = ?, inkey = ?, deleted = true
+                    WHERE id = ?
+                    """,
+                    (user, adminkey, inkey, row[0]),
+                )
+            except Exception:
+                continue
+    except OperationalError:
+        # this is necessary now because it may be the case that this migration will
+        # run twice in some environments.
+        # catching errors like this won't be necessary in anymore now that we
+        # keep track of db versions so no migration ever runs twice.
+        pass