(db_engine: Engine)
| 20 | after=after_log(logger, logging.WARN), |
| 21 | ) |
| 22 | def init(db_engine: Engine) -> None: |
| 23 | try: |
| 24 | with Session(db_engine) as session: |
| 25 | # Try to create session to check if DB is awake |
| 26 | session.exec(select(1)) |
| 27 | except Exception as e: |
| 28 | logger.error(e) |
| 29 | raise e |
| 30 | |
| 31 | |
| 32 | def main() -> None: |