ngfchl
95 lines
2.4 KiB
Python
95 lines
2.4 KiB
Python
from typing import List
|
|
|
|
from django.db.models import Prefetch
|
|
from django.shortcuts import get_object_or_404
|
|
from ninja import Router
|
|
from ninja.pagination import paginate, PageNumberPagination
|
|
|
|
from users.models import User, Role
|
|
from users.schema import UserUpdate, UserOut, UserCreate
|
|
|
|
router = Router(tags=['users'])
|
|
|
|
|
|
@router.get("/", response=List[UserOut], summary="获取用户列表")
|
|
@paginate(PageNumberPagination, page_size=20)
|
|
def list_users(request):
|
|
# if not has_permission(request.auth.id, "user:list"):
|
|
# raise HttpError(403, "权限不足")
|
|
return User.objects.all()
|
|
|
|
|
|
@router.get("/", response=dict)
|
|
def list_users(request, page: int = 1, page_size: int = 10, username: str = None):
|
|
qs = User.objects.all().order_by("-id")
|
|
|
|
if username:
|
|
qs = qs.filter(username__icontains=username)
|
|
|
|
total = qs.count()
|
|
start = (page - 1) * page_size
|
|
end = start + page_size
|
|
|
|
items = list(qs[start:end].values(
|
|
"id", "username", "cname", "phone", "email",
|
|
"avatar", "dept",
|
|
"is_active", "is_staff", "is_superuser",
|
|
"last_login", "date_joined"
|
|
))
|
|
|
|
# roles 需要手动补
|
|
for i, obj in enumerate(qs[start:end]):
|
|
items[i]["roles"] = list(obj.roles.values_list("id", flat=True))
|
|
|
|
return {"items": items, "count": total}
|
|
|
|
|
|
@router.get("/{id}", response=UserOut)
|
|
def get_user(request, id: int):
|
|
obj = get_object_or_404(User, id=id)
|
|
data = obj.__dict__
|
|
data["roles"] = list(obj.roles.values_list("id", flat=True))
|
|
return data
|
|
|
|
|
|
@router.post("/", response=UserOut)
|
|
def create_user(request, data: UserCreate):
|
|
user = User.objects.create_user(
|
|
username=data.username,
|
|
password=data.password
|
|
)
|
|
|
|
for k, v in data.dict(exclude={"password"}).items():
|
|
setattr(user, k, v)
|
|
|
|
user.save()
|
|
|
|
if data.roles:
|
|
user.roles.set(data.roles)
|
|
|
|
return user
|
|
|
|
|
|
@router.put("/{id}")
|
|
def update_user(request, id: int, data: UserUpdate):
|
|
user = get_object_or_404(User, id=id)
|
|
|
|
if data.password:
|
|
user.set_password(data.password)
|
|
|
|
for k, v in data.dict(exclude_unset=True, exclude={"password", "roles"}).items():
|
|
setattr(user, k, v)
|
|
|
|
user.save()
|
|
|
|
if data.roles is not None:
|
|
user.roles.set(data.roles)
|
|
|
|
return {"success": True}
|
|
|
|
|
|
@router.delete("/{id}")
|
|
def delete_user(request, id: int):
|
|
User.objects.filter(id=id).delete()
|
|
return {"success": True}
|