86 lines
2.8 KiB
Python
86 lines
2.8 KiB
Python
from typing import Annotated
|
|
|
|
from asyncreq import request
|
|
from fastapi import APIRouter, Header, HTTPException, Request
|
|
|
|
from src.api.config import API_USERS
|
|
from src.api.routes.auth import checkAuth
|
|
from src.api.schemas.user import User, UserMin, UserRegister
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post("", response_model=UserMin)
|
|
async def create_users(user: UserRegister, req: Request):
|
|
request_id = req.state.request_id
|
|
header = {"x-api-request-id": request_id}
|
|
(response, status, _) = await request(
|
|
f"{API_USERS}", "POST", json=user.model_dump(), headers=header
|
|
)
|
|
if status < 200 or status > 204:
|
|
raise HTTPException(status_code=status, detail=response)
|
|
return response
|
|
|
|
|
|
@router.post("/airline", response_model=UserMin)
|
|
async def create_airline(
|
|
user: UserRegister,
|
|
req: Request,
|
|
authorization: Annotated[str | None, Header()] = None,
|
|
):
|
|
await checkAuth(req, authorization, roles=["admin"])
|
|
request_id = req.state.request_id
|
|
header = {"x-api-request-id": request_id}
|
|
data = user.model_dump()
|
|
data["role"] = "airline"
|
|
(response, status, _) = await request(
|
|
f"{API_USERS}", "POST", json=data, headers=header
|
|
)
|
|
if status < 200 or status > 204:
|
|
raise HTTPException(status_code=status, detail=response)
|
|
return response
|
|
|
|
|
|
@router.get("/{id}", response_model=User)
|
|
async def get_user(
|
|
id: str, req: Request, authorization: Annotated[str | None, Header()] = None
|
|
):
|
|
await checkAuth(req, authorization, userId=id)
|
|
request_id = req.state.request_id
|
|
header = {"x-api-request-id": request_id}
|
|
(response, status, _) = await request(f"{API_USERS}/{id}", "GET", headers=header)
|
|
if status < 200 or status > 204:
|
|
raise HTTPException(status_code=status, detail=response)
|
|
return response
|
|
|
|
|
|
@router.put("/{id}", response_model=User)
|
|
async def update_user(
|
|
id: str,
|
|
user: UserRegister,
|
|
req: Request,
|
|
authorization: Annotated[str | None, Header()] = None,
|
|
):
|
|
await checkAuth(req, authorization, userId=id)
|
|
request_id = req.state.request_id
|
|
header = {"x-api-request-id": request_id}
|
|
(response, status, _) = await request(
|
|
f"{API_USERS}/{id}", "PUT", json=user.model_dump(), headers=header
|
|
)
|
|
if status < 200 or status > 204:
|
|
raise HTTPException(status_code=status, detail=response)
|
|
return response
|
|
|
|
|
|
@router.delete("/{id}")
|
|
async def delete_user(
|
|
id: str, req: Request, authorization: Annotated[str | None, Header()] = None
|
|
):
|
|
await checkAuth(req, authorization, userId=id)
|
|
request_id = req.state.request_id
|
|
header = {"x-api-request-id": request_id}
|
|
(response, status, _) = await request(f"{API_USERS}/{id}", "DELETE", headers=header)
|
|
if status < 200 or status > 204:
|
|
raise HTTPException(status_code=status, detail=response)
|
|
return response
|