From 1fb7df12d76fefb67f95e9c7d361ad5dd8c3ba18 Mon Sep 17 00:00:00 2001 From: "saga.rey" Date: Sun, 21 Apr 2024 09:44:35 +0800 Subject: [PATCH] fix: in alembic's offline mode (db migrate with --sql option), skip data operations (#3533) --- .../versions/de95f5c77138_migration_serpapi_api_key.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/api/migrations/versions/de95f5c77138_migration_serpapi_api_key.py b/api/migrations/versions/de95f5c77138_migration_serpapi_api_key.py index f1236df31..236576683 100644 --- a/api/migrations/versions/de95f5c77138_migration_serpapi_api_key.py +++ b/api/migrations/versions/de95f5c77138_migration_serpapi_api_key.py @@ -8,7 +8,7 @@ Create Date: 2024-01-21 12:09:04.651394 from json import dumps, loads import sqlalchemy as sa -from alembic import op +from alembic import context, op # revision identifiers, used by Alembic. revision = 'de95f5c77138' @@ -40,8 +40,13 @@ def upgrade(): {"serpapi_api_key": "$KEY"} - created_at <- tool_providers.created_at - updated_at <- tool_providers.updated_at - """ + + # in alembic's offline mode (with --sql option), skip data operations and output comments describing the migration to raw sql + if context.is_offline_mode(): + print(f" /*{upgrade.__doc__}*/\n") + return + # select all tool_providers tool_providers = op.get_bind().execute( sa.text(