mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-03-10 09:19:42 +01:00
conv.py: warn if table is empty
This commit is contained in:
parent
00af60b1df
commit
64c330459b
1 changed files with 13 additions and 9 deletions
|
@ -1,3 +1,8 @@
|
||||||
|
# Python script to migrate an LNbits SQLite DB to Postgres
|
||||||
|
# All credits to @Fritz446 for the awesome work
|
||||||
|
|
||||||
|
# pip install psycopg2 OR psycopg2-binary
|
||||||
|
|
||||||
import argparse
|
import argparse
|
||||||
import os
|
import os
|
||||||
import sqlite3
|
import sqlite3
|
||||||
|
@ -8,13 +13,6 @@ import psycopg2
|
||||||
|
|
||||||
from lnbits.settings import settings
|
from lnbits.settings import settings
|
||||||
|
|
||||||
# Python script to migrate an LNbits SQLite DB to Postgres
|
|
||||||
# All credits to @Fritz446 for the awesome work
|
|
||||||
|
|
||||||
# pip install psycopg2 OR psycopg2-binary
|
|
||||||
|
|
||||||
# Change these values as needed
|
|
||||||
|
|
||||||
sqfolder = settings.lnbits_data_folder
|
sqfolder = settings.lnbits_data_folder
|
||||||
db_url = settings.lnbits_database_url
|
db_url = settings.lnbits_database_url
|
||||||
|
|
||||||
|
@ -31,7 +29,7 @@ else:
|
||||||
pgschema = ""
|
pgschema = ""
|
||||||
|
|
||||||
|
|
||||||
def get_sqlite_cursor(sqdb) -> sqlite3:
|
def get_sqlite_cursor(sqdb):
|
||||||
consq = sqlite3.connect(sqdb)
|
consq = sqlite3.connect(sqdb)
|
||||||
return consq.cursor()
|
return consq.cursor()
|
||||||
|
|
||||||
|
@ -112,12 +110,15 @@ def migrate_core(file: str, exclude_tables: List[str] = []):
|
||||||
def migrate_ext(file: str):
|
def migrate_ext(file: str):
|
||||||
filename = os.path.basename(file)
|
filename = os.path.basename(file)
|
||||||
schema = filename.replace("ext_", "").split(".")[0]
|
schema = filename.replace("ext_", "").split(".")[0]
|
||||||
print(f"Migrating ext: {file}.{schema}")
|
print(f"Migrating ext: {schema} from file {file}")
|
||||||
migrate_db(file, schema)
|
migrate_db(file, schema)
|
||||||
print(f"✅ Migrated ext: {schema}")
|
print(f"✅ Migrated ext: {schema}")
|
||||||
|
|
||||||
|
|
||||||
def migrate_db(file: str, schema: str, exclude_tables: List[str] = []):
|
def migrate_db(file: str, schema: str, exclude_tables: List[str] = []):
|
||||||
|
# first we check if this file exists:
|
||||||
|
assert os.path.isfile(file), f"{file} does not exist!"
|
||||||
|
|
||||||
sq = get_sqlite_cursor(file)
|
sq = get_sqlite_cursor(file)
|
||||||
tables = sq.execute(
|
tables = sq.execute(
|
||||||
"""
|
"""
|
||||||
|
@ -126,6 +127,9 @@ def migrate_db(file: str, schema: str, exclude_tables: List[str] = []):
|
||||||
"""
|
"""
|
||||||
).fetchall()
|
).fetchall()
|
||||||
|
|
||||||
|
if len(tables) == 0:
|
||||||
|
print(f"⚠️ You sneaky dev! Schema {schema} is empty!")
|
||||||
|
|
||||||
for table in tables:
|
for table in tables:
|
||||||
tableName = table[0]
|
tableName = table[0]
|
||||||
print(f"Migrating table {tableName}")
|
print(f"Migrating table {tableName}")
|
||||||
|
|
Loading…
Add table
Reference in a new issue