From 1074a0df19ec550e93504fffde45c2af7ec01e12 Mon Sep 17 00:00:00 2001
From: feng <1304903146@qq.com>
Date: Tue, 22 Jul 2025 16:06:33 +0800
Subject: [PATCH] perf: MFA coce reuse

---
 apps/authentication/mfa/base.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/apps/authentication/mfa/base.py b/apps/authentication/mfa/base.py
index b7f7ae4ee..9a2e0198f 100644
--- a/apps/authentication/mfa/base.py
+++ b/apps/authentication/mfa/base.py
@@ -1,5 +1,6 @@
 import abc
 
+from django.conf import settings
 from django.core.cache import cache
 from django.utils.translation import gettext_lazy as _
 
@@ -23,17 +24,21 @@ class BaseMFA(abc.ABC):
 
         cache_key = f'{self.name}_{self.user.username}'
         cache_code = cache.get(cache_key)
-        if cache_code == code:
+
+        is_match = cache_code == code
+
+        if settings.SAFE_MODE and is_match:
             return False, _(
                 "The two-factor code you entered has either already been used or has expired. "
                 "Please request a new one."
             )
 
         ok, msg = self._check_code(code)
+
         if not ok:
             return False, msg
 
-        cache.set(cache_key, code, 60 * 5)
+        cache.set(cache_key, code, 60)
         return True, msg
 
     def is_authenticated(self):