From aba908c0aa2b22d6d62dcd79763ec474df63f164 Mon Sep 17 00:00:00 2001
From: yuetsh <517252939@qq.com>
Date: Sun, 21 Sep 2025 20:14:17 +0800
Subject: [PATCH] add class_name when generate user

---
 account/views/admin.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/account/views/admin.py b/account/views/admin.py
index a7ab532..f11243a 100644
--- a/account/views/admin.py
+++ b/account/views/admin.py
@@ -22,6 +22,19 @@ from ..serializers import (
 from ..serializers import ImportUserSerializer
 
 
+# ks251XXX 或者 ks2510XX 返回 251 或者 2510
+# 其他返回 ""
+def get_class_name(username):
+    if username.startswith("ks"):
+        result = re.search(r"ks\d+", username)
+        if result:
+            return result.group(0)[2:]
+        else:
+            return ""
+    else:
+        return ""
+
+
 class UserAdminAPI(APIView):
     @validate_serializer(ImportUserSerializer)
     @super_admin_required
@@ -41,6 +54,7 @@ class UserAdminAPI(APIView):
                     password=make_password(user_data[1]),
                     email=user_data[2],
                     raw_password=user_data[1],
+                    class_name=get_class_name(user_data[0]),
                 )
             )
 
@@ -86,6 +100,7 @@ class UserAdminAPI(APIView):
 
         pre_username = user.username
         user.username = data["username"].lower()
+        user.class_name = get_class_name(data["username"])
         user.email = data["email"].lower()
         user.admin_type = data["admin_type"]
         user.is_disabled = data["is_disabled"]