From b74b91774c5d9d11fbcb9983bb73dc60901faef6 Mon Sep 17 00:00:00 2001
From: Kevin Chung <kchung@ctfd.io>
Date: Tue, 16 Mar 2021 18:48:18 -0400
Subject: [PATCH] Set plugin migration version after a successful migration
 (#1827)

* Set plugin migration version after successful migrations
* Closes #1763
---
 CTFd/plugins/migrations.py | 3 +++
 1 file changed, 3 insertions(+)

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()