diff --git a/backend/alembic/env.py b/backend/alembic/env.py index a3be7b2..8396e25 100644 --- a/backend/alembic/env.py +++ b/backend/alembic/env.py @@ -1,4 +1,5 @@ import asyncio +import sys from logging.config import fileConfig from sqlalchemy import engine_from_config, pool @@ -75,7 +76,12 @@ async def run_migrations_online() -> None: async with connectable.connect() as connection: await connection.run_sync(do_run_migrations) -if context.is_offline_mode(): - asyncio.run(run_migrations_offline()) -else: - asyncio.run(run_migrations_online()) +def main() -> None: + if context.is_offline_mode(): + asyncio.run(run_migrations_offline()) + else: + asyncio.run(run_migrations_online()) + + +if __name__ == '__main__': + main() \ No newline at end of file