remove MetaDataTable and MetaDataColumn to cleanup schema
This commit is contained in:
+11
-20
@@ -12,7 +12,6 @@ from pathlib import Path
|
||||
from db.models import registry
|
||||
from db.models.base import Base
|
||||
from config import get_logger
|
||||
from db.repository.metadata import get_tables, get_columns_for_table
|
||||
|
||||
parser = ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter)
|
||||
parser.add_argument('--verbose', '-v', action='count', default=0)
|
||||
@@ -40,31 +39,23 @@ if __name__ == '__main__':
|
||||
__session__ = sessionmaker(bind=engine)
|
||||
with __session__() as db:
|
||||
data = {}
|
||||
tables = get_tables(db)
|
||||
tables = registry.keys()
|
||||
for table in tables:
|
||||
# logger.info(f"Table {table.name} with {table.id}")
|
||||
columns = get_columns_for_table(db, table)
|
||||
model = registry[table.name]
|
||||
model = registry[table]
|
||||
rows = db.query(model).all()
|
||||
entries = []
|
||||
for row in rows:
|
||||
entry = {}
|
||||
for column in columns:
|
||||
# logger.info(f" Column {column.order} {column.name} with {column.id}")
|
||||
try:
|
||||
value = getattr(row, column.name)
|
||||
if isinstance(value, datetime):
|
||||
entry[column.name] = str(value)
|
||||
else:
|
||||
entry[column.name] = value
|
||||
except AttributeError as error:
|
||||
logger.info(f"{error}")
|
||||
entry = row.export_dict()
|
||||
entries.append(entry)
|
||||
data[table.name] = entries
|
||||
logger.info(f"{table.name}: {len(entries)} exported")
|
||||
json_dump = json.dumps(data, indent=4)
|
||||
with open(args.file, "w") as dump_file:
|
||||
dump_file.write(json_dump)
|
||||
data[table] = entries
|
||||
logger.info(f"{table}: {len(entries)} exported")
|
||||
try:
|
||||
json_dump = json.dumps(data, indent=4)
|
||||
with open(args.file, "w") as dump_file:
|
||||
dump_file.write(json_dump)
|
||||
except TypeError as error:
|
||||
logger.info(f"{error}")
|
||||
logger.info(f"{len(data)} tables exported")
|
||||
#kontor_db = KontorDB(engine, logger)
|
||||
#kontor_db.export_db(ExportType.JSON, args.file)
|
||||
|
||||
Reference in New Issue
Block a user