From a58aaf8399102aa32008bb643a0ab1a8bf92fea1 Mon Sep 17 00:00:00 2001
From: Son Nguyen Kim <nguyenkims@users.noreply.github.com>
Date: Thu, 22 Jul 2021 09:09:13 +0200
Subject: [PATCH] add disable OTP to admin page

---
 app/admin_model.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/app/admin_model.py b/app/admin_model.py
index 33295d9e..31437c84 100644
--- a/app/admin_model.py
+++ b/app/admin_model.py
@@ -109,6 +109,19 @@ class UserAdmin(SLModelView):
 
         db.session.commit()
 
+    @action(
+        "disable_otp",
+        "Disable OTP",
+        "Disable OTP?",
+    )
+    def disable_otp(self, ids):
+        for user in User.query.filter(User.id.in_(ids)):
+            if user.enable_otp:
+                user.enable_otp = False
+                flash(f"Disable OTP for {user}", "info")
+
+        db.session.commit()
+
 
 def manual_upgrade(way: str, ids: [int], is_giveaway: bool):
     query = User.query.filter(User.id.in_(ids))