feat(main): initial commit
This commit is contained in:
52
api/routers/user.py
Normal file
52
api/routers/user.py
Normal file
@@ -0,0 +1,52 @@
|
||||
from ninja import Router
|
||||
from core.models import AppUser
|
||||
from django.shortcuts import get_object_or_404
|
||||
from ninja import Schema
|
||||
from typing import List, Optional
|
||||
from datetime import datetime
|
||||
|
||||
router = Router()
|
||||
|
||||
class UserSchema(Schema):
|
||||
id: int
|
||||
username: str
|
||||
email: str
|
||||
is_staff: bool
|
||||
is_superuser: bool
|
||||
is_active: bool
|
||||
date_joined: datetime
|
||||
|
||||
class UserCreateSchema(Schema):
|
||||
username: str
|
||||
email: str
|
||||
password: str
|
||||
is_superuser: bool = False
|
||||
|
||||
class UserUpdateSchema(Schema):
|
||||
username: Optional[str] = None
|
||||
email: Optional[str] = None
|
||||
is_active: Optional[bool] = None
|
||||
|
||||
@router.get("/", response=List[UserSchema])
|
||||
def list_users(request):
|
||||
return AppUser.objects.all()
|
||||
|
||||
@router.get("/{user_id}", response=UserSchema)
|
||||
def get_user(request, user_id: int):
|
||||
return get_object_or_404(AppUser, id=user_id)
|
||||
|
||||
@router.post("/", response={201: UserSchema})
|
||||
def create_user(request, payload: UserCreateSchema):
|
||||
if payload.is_superuser:
|
||||
user = AppUser.objects.create_superuser(payload.username, payload.email, payload.password)
|
||||
else:
|
||||
user = AppUser.objects.create_user(payload.username, payload.email, payload.password)
|
||||
return 201, user
|
||||
|
||||
@router.patch("/{user_id}", response=UserSchema)
|
||||
def update_user(request, user_id: int, payload: UserUpdateSchema):
|
||||
user = get_object_or_404(AppUser, id=user_id)
|
||||
for attr, value in payload.dict(exclude_unset=True).items():
|
||||
setattr(user, attr, value)
|
||||
user.save()
|
||||
return user
|
||||
Reference in New Issue
Block a user