Files
bellsystems-cp/backend/users/router.py

108 lines
3.1 KiB
Python

from fastapi import APIRouter, Depends, Query, UploadFile, File
from typing import Optional, List
from auth.models import TokenPayload
from auth.dependencies import require_permission
from users.models import (
UserCreate, UserUpdate, UserInDB, UserListResponse,
)
from users import service
router = APIRouter(prefix="/api/users", tags=["users"])
@router.get("", response_model=UserListResponse)
async def list_users(
search: Optional[str] = Query(None),
status: Optional[str] = Query(None),
_user: TokenPayload = Depends(require_permission("app_users", "view")),
):
users = service.list_users(search=search, status=status)
return UserListResponse(users=users, total=len(users))
@router.get("/{user_id}", response_model=UserInDB)
async def get_user(
user_id: str,
_user: TokenPayload = Depends(require_permission("app_users", "view")),
):
return service.get_user(user_id)
@router.post("", response_model=UserInDB, status_code=201)
async def create_user(
body: UserCreate,
_user: TokenPayload = Depends(require_permission("app_users", "add")),
):
return service.create_user(body)
@router.put("/{user_id}", response_model=UserInDB)
async def update_user(
user_id: str,
body: UserUpdate,
_user: TokenPayload = Depends(require_permission("app_users", "edit")),
):
return service.update_user(user_id, body)
@router.delete("/{user_id}", status_code=204)
async def delete_user(
user_id: str,
_user: TokenPayload = Depends(require_permission("app_users", "delete")),
):
service.delete_user(user_id)
@router.post("/{user_id}/block", response_model=UserInDB)
async def block_user(
user_id: str,
_user: TokenPayload = Depends(require_permission("app_users", "edit")),
):
return service.block_user(user_id)
@router.post("/{user_id}/unblock", response_model=UserInDB)
async def unblock_user(
user_id: str,
_user: TokenPayload = Depends(require_permission("app_users", "edit")),
):
return service.unblock_user(user_id)
@router.get("/{user_id}/devices", response_model=List[dict])
async def get_user_devices(
user_id: str,
_user: TokenPayload = Depends(require_permission("app_users", "view")),
):
return service.get_user_devices(user_id)
@router.post("/{user_id}/devices/{device_id}", response_model=UserInDB)
async def assign_device(
user_id: str,
device_id: str,
_user: TokenPayload = Depends(require_permission("app_users", "edit")),
):
return service.assign_device(user_id, device_id)
@router.delete("/{user_id}/devices/{device_id}", response_model=UserInDB)
async def unassign_device(
user_id: str,
device_id: str,
_user: TokenPayload = Depends(require_permission("app_users", "edit")),
):
return service.unassign_device(user_id, device_id)
@router.post("/{user_id}/photo")
async def upload_photo(
user_id: str,
file: UploadFile = File(...),
_user: TokenPayload = Depends(require_permission("app_users", "edit")),
):
contents = await file.read()
content_type = file.content_type or "image/jpeg"
url = service.upload_photo(user_id, contents, file.filename, content_type)
return {"photo_url": url}