mirror of
https://github.com/FutureOfMedTech-FITM-hack/backend.git
synced 2024-09-21 02:48:48 +03:00
81 lines
2.2 KiB
Python
81 lines
2.2 KiB
Python
from datetime import timedelta
|
|
from typing import Dict
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from starlette import status
|
|
|
|
from med_backend.auth.schemas import (
|
|
Token,
|
|
UpdateUserProfile,
|
|
User,
|
|
UserCreate,
|
|
UserLogin,
|
|
UserPublicInfo,
|
|
)
|
|
from med_backend.auth.services import (
|
|
ACCESS_TOKEN_EXPIRE_MINUTES,
|
|
authenticate_user,
|
|
create_access_token,
|
|
get_current_active_user,
|
|
)
|
|
from med_backend.db.dependencies import get_db_session
|
|
from med_backend.users.crud import create_user, delete_user, update_user
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post("/token", response_model=Token)
|
|
async def login_for_access_token(
|
|
data: UserLogin,
|
|
session: AsyncSession = Depends(get_db_session),
|
|
) -> Dict[str, str]:
|
|
user = await authenticate_user(session, data.email, data.password)
|
|
if not user:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail="Incorrect username or password",
|
|
headers={"WWW-Authenticate": "Bearer"},
|
|
)
|
|
access_token_expires = timedelta(minutes=ACCESS_TOKEN_EXPIRE_MINUTES)
|
|
access_token = create_access_token(
|
|
data={"sub": user.email},
|
|
expires_delta=access_token_expires,
|
|
)
|
|
return {"access_token": access_token, "token_type": "bearer"}
|
|
|
|
|
|
@router.post("/signup", response_model=UserPublicInfo)
|
|
async def create_user_view(
|
|
data: UserCreate,
|
|
db: AsyncSession = Depends(get_db_session),
|
|
) -> User:
|
|
user = await create_user(db, data)
|
|
return user
|
|
|
|
|
|
@router.get("/me", response_model=UserPublicInfo)
|
|
async def get_self(
|
|
current_user: User = Depends(get_current_active_user),
|
|
) -> User:
|
|
return current_user
|
|
|
|
|
|
@router.put("/me")
|
|
async def update_self(
|
|
data: UpdateUserProfile,
|
|
current_user: User = Depends(get_current_active_user),
|
|
session: AsyncSession = Depends(get_db_session),
|
|
):
|
|
await update_user(session, current_user.id, data)
|
|
return {"detail": "updated"}
|
|
|
|
|
|
@router.delete("/me")
|
|
async def update_self(
|
|
current_user: User = Depends(get_current_active_user),
|
|
session: AsyncSession = Depends(get_db_session),
|
|
):
|
|
await delete_user(session, current_user.id)
|
|
return {"detail": "updated"}
|