aljaz/CTFd
1
0
Fork 0
mirror of https://github.com/aljazceru/CTFd.git synced 2025-12-17 05:54:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix eb68f277ab61_add_func_column_to_dynamic_challenges migration (#2374)

Browse Source 
* Fix issue in migration with some MySQL versions
This commit is contained in:
Kevin Chung
2023-07-24 00:28:30 -04:00
committed by GitHub
parent 100eb58ac5
commit c2a13e55a7
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
CTFd/plugins/dynamic_challenges/migrations/eb68f277ab61_add_func_column_to_dynamic_challenges.py
View File

@@ -19,9 +19,15 @@ def upgrade(op=None):
"dynamic_challenge", sa.Column("function", sa.String(length=32), nullable=True) "dynamic_challenge", sa.Column("function", sa.String(length=32), nullable=True)
) )
conn = op.get_bind() conn = op.get_bind()
url = str(conn.engine.url)
if url.startswith("postgres"):
conn.execute( conn.execute(
"UPDATE dynamic_challenge SET function = 'logarithmic' WHERE function IS NULL" "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): def downgrade(op=None):