aljaz/nutshell
1
0
Fork 0
mirror of https://github.com/aljazceru/nutshell.git synced 2025-12-20 10:34:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

remove column check during migration (#408)

Browse Source
This commit is contained in:
callebtc
2024-02-05 18:43:11 +01:00
committed by GitHub
parent f8f167fe23
commit 087ee957a5
1 changed files with 9 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
cashu/mint/migrations.py
View File

@@ -1,4 +1,4 @@
from ..core.db import SQLITE, Connection, Database, table_with_schema, timestamp_now
from ..core.db import Connection, Database, table_with_schema, timestamp_now
from ..core.settings import settings
@@ -222,34 +222,18 @@ async def m010_add_index_to_proofs_used(db: Database):
async def m011_add_quote_tables(db: Database):
async def get_columns(db: Database, conn: Connection, table: str):
if db.type == SQLITE:
query = f"PRAGMA table_info({table})"
else:
query = (
"SELECT column_name FROM information_schema.columns WHERE table_name ="
f" '{table}'"
)
res = await conn.execute(query)
if db.type == SQLITE:
return [r["name"] async for r in res]
else:
return [r["column_name"] async for r in res]
async with db.connect() as conn:
# add column "created" to tables invoices, promises, proofs_used, proofs_pending
tables = ["invoices", "promises", "proofs_used", "proofs_pending"]
for table in tables:
columns = await get_columns(db, conn, table)
if "created" not in columns:
await conn.execute(
f"ALTER TABLE {table_with_schema(db, table)} ADD COLUMN created"
" TIMESTAMP"
)
await conn.execute(
f"UPDATE {table_with_schema(db, table)} SET created ="
f" '{timestamp_now(db)}'"
)
await conn.execute(
f"ALTER TABLE {table_with_schema(db, table)} ADD COLUMN created"
" TIMESTAMP"
)
await conn.execute(
f"UPDATE {table_with_schema(db, table)} SET created ="
f" '{timestamp_now(db)}'"
)
# add column "witness" to table proofs_used
await conn.execute(