diff --git a/src/aki_prj23_transparenzregister/utils/sql/reset_sql.py b/src/aki_prj23_transparenzregister/utils/sql/reset_sql.py index 878ec38..045a572 100644 --- a/src/aki_prj23_transparenzregister/utils/sql/reset_sql.py +++ b/src/aki_prj23_transparenzregister/utils/sql/reset_sql.py @@ -26,13 +26,16 @@ def reset_tables(db: Session, all_tables: bool = False) -> None: db.commit() else: logger.info(f"Resetting the main SQL tables in {db.bind}.") - for table in Base.metadata.sorted_tables: - if str(table) == entities.MissingCompany.__tablename__: - continue - logger.debug(f"Dropping {table}") - table.drop(db.bind) - db.commit() + tables = [ + table + for table in Base.metadata.sorted_tables + if str(table) != entities.MissingCompany.__tablename__ + ] + logger.debug(f"Dropping tables: {', '.join([str(_) for _ in tables])}") + Base.metadata.drop_all(db.bind, tables=tables) + db.commit() init_db(db) + db.commit() def cli() -> None: