correct placement of warning
This commit is contained in:
+4
-3
@@ -127,9 +127,6 @@ 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}")
|
||||||
@@ -143,6 +140,10 @@ def migrate_db(file: str, schema: str, exclude_tables: List[str] = []):
|
|||||||
q = build_insert_query(schema, tableName, columns)
|
q = build_insert_query(schema, tableName, columns)
|
||||||
|
|
||||||
data = sq.execute(f"SELECT * FROM {tableName};").fetchall()
|
data = sq.execute(f"SELECT * FROM {tableName};").fetchall()
|
||||||
|
|
||||||
|
if len(data) == 0:
|
||||||
|
print(f"⚠️ You sneaky dev! Table {tableName} is empty!")
|
||||||
|
|
||||||
insert_to_pg(q, data)
|
insert_to_pg(q, data)
|
||||||
sq.close()
|
sq.close()
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user