From c2a13e55a773c191a5fbfcb6cc2c965dd0e63c17 Mon Sep 17 00:00:00 2001 From: Kevin Chung Date: Mon, 24 Jul 2023 00:28:30 -0400 Subject: [PATCH] Fix eb68f277ab61_add_func_column_to_dynamic_challenges migration (#2374) * Fix issue in migration with some MySQL versions --- ...f277ab61_add_func_column_to_dynamic_challenges.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/CTFd/plugins/dynamic_challenges/migrations/eb68f277ab61_add_func_column_to_dynamic_challenges.py b/CTFd/plugins/dynamic_challenges/migrations/eb68f277ab61_add_func_column_to_dynamic_challenges.py index ce8ad747..140bda58 100644 --- a/CTFd/plugins/dynamic_challenges/migrations/eb68f277ab61_add_func_column_to_dynamic_challenges.py +++ b/CTFd/plugins/dynamic_challenges/migrations/eb68f277ab61_add_func_column_to_dynamic_challenges.py @@ -19,9 +19,15 @@ def upgrade(op=None): "dynamic_challenge", sa.Column("function", sa.String(length=32), nullable=True) ) conn = op.get_bind() - conn.execute( - "UPDATE dynamic_challenge SET function = 'logarithmic' WHERE function IS NULL" - ) + url = str(conn.engine.url) + if url.startswith("postgres"): + conn.execute( + "UPDATE dynamic_challenge SET function = 'logarithmic' WHERE function IS NULL" + ) + else: + conn.execute( + "UPDATE dynamic_challenge SET `function` = 'logarithmic' WHERE `function` IS NULL" + ) def downgrade(op=None):