修复开启2fa登录后无法更新ding_user_id的问题 (#2104)

hhyo · Mar 31, 2023 · f275b56 · f275b56
1 parent 17bca47
commit f275b56
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/sql_api/api_user.py b/sql_api/api_user.py
@@ -22,6 +22,8 @@
 from django.http import Http404
 from sql.models import Users, ResourceGroup, TwoFactorAuthConfig
 from common.twofa import TwoFactorAuthBase, get_authenticator
+from common.config import SysConfig
+from common.utils.ding_api import get_ding_user_id
 import random
 import json
 import time
@@ -421,4 +423,8 @@ def post(self, request):
             login(request, user, backend="django.contrib.auth.backends.ModelBackend")
             request.session.set_expiry(settings.SESSION_COOKIE_AGE)
 
+            # 更新用户ding_user_id
+            if SysConfig().get("ding_to_person") is True and "admin" not in engineer:
+                get_ding_user_id(engineer)
+
         return Response(result)