Browse Source

增加用户日志

yf_yzl 2 years ago
parent
commit
3e07099ea5
2 changed files with 31 additions and 22 deletions
  1. 1 1
      smartfarming/urls.py
  2. 30 21
      smartfarming/views/user.py

+ 1 - 1
smartfarming/urls.py

@@ -106,7 +106,7 @@ urlpatterns = [
     path('role_update', RoleUpdateAPIView.as_view()),  # 更新角色 
     path('users', UserListAPIView.as_view()),  # 用户列表
     path('user_update', UserUpdateAPIView.as_view()),  # 删除角色 
-    path('user_add', UserAddAPIView.as_view()),  # 删除角色 
+    path('user_add', UserAddAPIView.as_view()),  # 添加用户 
     path('user_delete', UserDeleteAPIView.as_view()),  # 删除角色 
     path('theme', HomeThemeModelListAPIView.as_view()),  # 主题列表 
     path('theme_add', HomeThemeModelAPIView.as_view()),  # 修改主题 

+ 30 - 21
smartfarming/views/user.py

@@ -2,6 +2,7 @@ from django.contrib.auth import authenticate
 from rest_framework.views import APIView
 from smartfarming.models.user import DeviceUser, MongoUserLoginLog, Role, HomeThemeModel
 import time 
+import logging
 from kedong.utils import PrAes
 from smartfarming.utils import get_perm_list, get_all_pers, get_captcha
 from rest_framework.response import Response
@@ -11,6 +12,10 @@ from django.core.paginator import Paginator
 from kedong import settings 
 from django.db.models import Q
 from kedong.tools import RedisPool
+import logging
+
+
+logger = logging.getLogger("myapp")
 
 
 class RoleAPIView(APIView):
@@ -269,27 +274,31 @@ class UserAddAPIView(APIView):
 
     def post(self, request):
         # 新增用户
-        request_data = request.data 
-        username = request_data.get("username")
-        password = request_data.get("password")
-        real_name = request_data.get("real_name")
-        mobile = request_data.get("mobile")
-        role_id = request_data.get("role_id")
-        state = request_data.get("state")  # 1 正常
-        user = User.objects.filter(username=username, is_active=1)
-        if user:
-            return Response({"code": 2, "msg": "该用户已存在"})
-        user = User.objects.create_user(username=username, password=password)
-        device_user = DeviceUser.objects.create(
-                uid = user.id,
-                real_name = real_name,
-                mobile = mobile,
-                role_id = role_id,
-                state = state
-            )
-        if user and device_user:
-            return Response({"code": 0, "msg": "success"})
-        else:
+        try:
+            request_data = request.data 
+            username = request_data.get("username")
+            password = request_data.get("password")
+            real_name = request_data.get("real_name")
+            mobile = request_data.get("mobile")
+            role_id = request_data.get("role_id")
+            state = request_data.get("state")  # 1 正常
+            user = User.objects.filter(username=username, is_active=1)
+            if user:
+                return Response({"code": 2, "msg": "该用户已存在"})
+            user = User.objects.create_user(username=username, password=password)
+            device_user = DeviceUser.objects.create(
+                    uid = user.id,
+                    real_name = real_name,
+                    mobile = mobile,
+                    role_id = role_id,
+                    state = state
+                )
+            if user and device_user:
+                return Response({"code": 0, "msg": "success"})
+            else:
+                return Response({"code": 2, "msg": "创建用户失败"})
+        except Exception as e:
+            logger.error(f"创建用户失败: {e}")
             return Response({"code": 2, "msg": "创建用户失败"})