def init_db(engine: Engine, *, debug: bool, message: str = "hmm"): """init tables""" if debug: log.instance_logger(engine, echoflag=True) metadata.create_all(bind=engine) engine.dispose() print(f"** init_db {debug=} {message=} **")
def execute_query(engine: Engine, query: str) -> List[Tuple[Any]]: """ Execute DB queries safely. """ try: logging.warning(f"Running query on Snowflake: \n{query}") connection = engine.connect() results = connection.execute(query).fetchall() finally: connection.close() engine.dispose() return results