first commit

code/system/apis/user.py

@@ -0,0 +1,125 @@
+# from application.ninja_cof import api
+# Author 臧成龙
+# coding=utf-8
+# @Time    : 2022/5/29 01:36
+# @File    : user.py
+# @Software: PyCharm
+# @qq: 939589097
+
+from typing import List
+
+from django.contrib.auth.hashers import make_password
+from django.shortcuts import get_object_or_404
+from ninja import Field, ModelSchema, Query, Router, Schema
+from ninja.pagination import paginate
+from system.models import Users
+from utils.fu_crud import create, delete, retrieve
+from utils.fu_ninja import FuFilters, MyPagination
+from utils.fu_response import FuResponse
+from utils.usual import get_user_info_from_token
+
+router = Router()
+
+
+class Filters(FuFilters):
+    name: str = Field(None, alias="name")
+    mobile: str = Field(None, alias="mobile")
+    status: bool = Field(None, alias="status")
+    dept_id__in: list = Field(None, alias="dept_ids[]")
+    id: int = Field(None, alias="id")
+
+
+class SchemaIn(ModelSchema):
+    dept_id: int = Field(None, alias="dept")
+    post: list = []
+    role: list = []
+
+    class Config:
+        model = Users
+        model_exclude = ['id', 'groups', 'user_permissions', 'is_superuser', 'dept', 'post', 'role', 'password',
+                         'create_datetime', 'update_datetime']
+
+
+class SchemaOut(ModelSchema):
+    class Config:
+        model = Users
+        model_exclude = ['password']
+
+
+@router.post("/user", response=SchemaOut)
+def create_user(request, data: SchemaIn):
+    data_dic = data.dict()
+    data_dic['password'] = make_password('123456')
+    post = data_dic.pop('post')
+    role = data_dic.pop('role')
+    user = create(request, data_dic, Users)
+    # 为多对多关系添加数据
+    user.post.set(post)
+    user.role.set(role)
+    return user
+
+
+@router.delete("/user/{user_id}")
+def delete_user(request, user_id: int):
+    delete(user_id, Users)
+    return {"success": True}
+
+
+@router.put("/user/{user_id}", response=SchemaOut)
+def update_user(request, user_id: int, payload: SchemaIn):
+    user = get_object_or_404(Users, id=user_id)
+    for attr, value in payload.dict().items():
+        if attr == 'role':
+            user.role.set(value)
+        elif attr == 'post':
+            user.post.set(value)
+        else:
+            setattr(user, attr, value)
+    user.save()
+    return user
+
+
+@router.get("/user", response=List[SchemaOut])
+@paginate(MyPagination)
+def list_user(request, filters: Filters = Query(...)):
+    qs = retrieve(request, Users, filters)
+    return qs
+
+
+@router.get("/user/all/list", response=List[SchemaOut])
+def all_list_user(request):
+    qs = retrieve(request, Users)
+    return qs
+
+
+@router.get("/user/{user_id}", response=SchemaOut)
+def get_user(request, user_id: int):
+    user = get_object_or_404(Users, id=user_id)
+    return user
+
+
+class SchemaIn(Schema):
+    id: int
+    password: str
+
+
+@router.post("/user/set/repassword", response=SchemaOut)
+def repassword(request, data: SchemaIn):
+    request_user = get_user_info_from_token(request)
+    request_user_id = request_user['id']
+    update_id = data.id
+    if request_user_id == update_id:
+        user = get_object_or_404(Users, id=update_id)
+        user.set_password(data.password)
+        user.save()
+        return FuResponse(msg='密码修改成功')
+    else:
+        return FuResponse(code=403, msg='对不起, 只能修改自己的密码')
+
+
+@router.put("/user/reset/password/{id}", response=SchemaOut)
+def reset_password(request, id:int):
+    user = get_object_or_404(Users, id=id)
+    user.set_password('123456')
+    user.save()
+    return FuResponse(msg='密码重置成功')