From 78f4e5a89a995f8db5305be8888d8e5982d21cbd Mon Sep 17 00:00:00 2001
From: ibuler <ibuler@qq.com>
Date: Tue, 10 Apr 2018 21:04:56 +0800
Subject: [PATCH] =?UTF-8?q?[Update]=20=E4=BF=AE=E6=94=B9=E7=94=A8=E6=88=B7?=
 =?UTF-8?q?Opt?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/users/models/user.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/apps/users/models/user.py b/apps/users/models/user.py
index 8ebfaf590..347848b7b 100644
--- a/apps/users/models/user.py
+++ b/apps/users/models/user.py
@@ -211,6 +211,12 @@ class User(AbstractUser):
     def otp_enabled(self):
         return self.otp_level > 0
 
+    def enabled_otp(self):
+        self.otp_level = 1
+
+    def force_enable_otp(self):
+        self.otp_level = 2
+
     @property
     def otp_force_enabled(self):
         return self.otp_level == 2