diff --git a/CTFd/plugins/migrations.py b/CTFd/plugins/migrations.py index a3661de2..7b9393fa 100644 --- a/CTFd/plugins/migrations.py +++ b/CTFd/plugins/migrations.py @@ -73,6 +73,9 @@ def upgrade(plugin_name=None, revision=None, lower="current"): for r in revs: with context.begin_transaction(): r.module.upgrade(op=op) + # Set revision that succeeded so we don't need + # to start from the beginning on failure + set_config(plugin_name + "_alembic_version", r.revision) finally: conn.close()