mirror of
https://github.com/fastapi-users/fastapi-users.git
synced 2025-08-15 19:30:47 +08:00
58 lines
1.3 KiB
Python
58 lines
1.3 KiB
Python
from beanie import init_beanie
|
|
from fastapi import Depends, FastAPI
|
|
|
|
from app.db import User, db
|
|
from app.schemas import UserCreate, UserRead, UserUpdate
|
|
from app.users import (
|
|
auth_backend,
|
|
current_active_user,
|
|
fastapi_users,
|
|
google_oauth_client,
|
|
)
|
|
|
|
app = FastAPI()
|
|
|
|
app.include_router(
|
|
fastapi_users.get_auth_router(auth_backend), prefix="/auth/jwt", tags=["auth"]
|
|
)
|
|
app.include_router(
|
|
fastapi_users.get_register_router(UserRead, UserCreate),
|
|
prefix="/auth",
|
|
tags=["auth"],
|
|
)
|
|
app.include_router(
|
|
fastapi_users.get_reset_password_router(),
|
|
prefix="/auth",
|
|
tags=["auth"],
|
|
)
|
|
app.include_router(
|
|
fastapi_users.get_verify_router(UserRead),
|
|
prefix="/auth",
|
|
tags=["auth"],
|
|
)
|
|
app.include_router(
|
|
fastapi_users.get_users_router(UserRead, UserUpdate),
|
|
prefix="/users",
|
|
tags=["users"],
|
|
)
|
|
app.include_router(
|
|
fastapi_users.get_oauth_router(google_oauth_client, auth_backend, "SECRET"),
|
|
prefix="/auth/google",
|
|
tags=["auth"],
|
|
)
|
|
|
|
|
|
@app.get("/authenticated-route")
|
|
async def authenticated_route(user: User = Depends(current_active_user)):
|
|
return {"message": f"Hello {user.email}!"}
|
|
|
|
|
|
@app.on_event("startup")
|
|
async def on_startup():
|
|
await init_beanie(
|
|
database=db,
|
|
document_models=[
|
|
User,
|
|
],
|
|
)
|