List and Read on frontend, fullecrud on back

This commit is contained in:
2023-01-10 18:48:47 +01:00
parent 399b52a272
commit 0b8a93b256
32 changed files with 623 additions and 51 deletions

View File

@@ -1,10 +1,14 @@
import uuid
from typing import Any, Dict, Generic, Optional
from bson import ObjectId
from fastapi import Depends
from fastapi_users import BaseUserManager, UUIDIDMixin, models, exceptions, schemas
from fastapi_users import BaseUserManager, FastAPIUsers, UUIDIDMixin, models, exceptions, schemas
from fastapi_users.authentication import BearerTransport, AuthenticationBackend
from fastapi_users.authentication.strategy.db import AccessTokenDatabase, DatabaseStrategy
from .models import User, get_user_db, AccessToken, get_access_token_db
from .models import User, get_user_db
SECRET = "SECRET"
@@ -66,6 +70,44 @@ class UserManager(UUIDIDMixin, BaseUserManager[User, uuid.UUID]):
return created_user
def parse_id(self, value: Any) -> uuid.UUID:
if isinstance(value, ObjectId):
return value
if isinstance(value, uuid.UUID):
return value
try:
return uuid.UUID(value)
except ValueError as e:
raise exceptions.InvalidID() from e
async def get_user_manager(user_db=Depends(get_user_db)):
yield UserManager(user_db)
yield UserManager(user_db)
def get_database_strategy(
access_token_db: AccessTokenDatabase[AccessToken] = Depends(get_access_token_db),
) -> DatabaseStrategy:
return DatabaseStrategy(access_token_db, lifetime_seconds=3600)
bearer_transport = BearerTransport(tokenUrl="auth/jwt/login")
auth_backend = AuthenticationBackend(
name="db",
transport=bearer_transport,
get_strategy=get_database_strategy,
)
fastapi_users = FastAPIUsers[User, uuid.UUID](
get_user_manager,
[auth_backend],
)
get_current_user = fastapi_users.current_user(active=True)
def get_auth_router():
return fastapi_users.get_auth_router(auth_backend)

View File

@@ -1,45 +1,13 @@
import uuid
from typing import Optional
from fastapi import Depends, Request
from fastapi_users import BaseUserManager, FastAPIUsers, UUIDIDMixin, models, exceptions
from fastapi_users.authentication import BearerTransport, AuthenticationBackend
from fastapi_users.authentication.strategy.db import AccessTokenDatabase, DatabaseStrategy
from fastapi import Depends
from beanie import PydanticObjectId
from fastapi import APIRouter, HTTPException
from typing import List
from .models import User, AccessToken, get_user_db, get_access_token_db
from .models import User
from .schemas import UserRead, UserUpdate, UserCreate
from .manager import get_user_manager
def get_database_strategy(
access_token_db: AccessTokenDatabase[AccessToken] = Depends(get_access_token_db),
) -> DatabaseStrategy:
return DatabaseStrategy(access_token_db, lifetime_seconds=3600)
bearer_transport = BearerTransport(tokenUrl="auth/jwt/login")
auth_backend = AuthenticationBackend(
name="db",
transport=bearer_transport,
get_strategy=get_database_strategy,
)
fastapi_users = FastAPIUsers[User, uuid.UUID](
get_user_manager,
[auth_backend],
)
def get_auth_router():
return fastapi_users.get_auth_router(auth_backend)
from .manager import get_user_manager, get_current_user, get_auth_router
router = APIRouter()
@@ -51,10 +19,16 @@ async def create(user: UserCreate, user_manager=Depends(get_user_manager)) -> di
return {"message": "User added successfully"}
@router.get("/me", response_description="User record retrieved")
async def read_me(user=Depends(get_current_user)) -> UserRead:
user = await User.get(user.id)
return UserRead(**user.dict())
@router.get("/{id}", response_description="User record retrieved")
async def read_id(id: PydanticObjectId) -> UserRead:
user = await User.get(id)
return user
return UserRead(**user.dict())
@router.get("/", response_model=List[UserRead], response_description="User records retrieved")