71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
from typing import Annotated
|
|
|
|
from app.database import FavouriteBeatmapset, MeResp, User
|
|
from app.dependencies.database import Database
|
|
from app.dependencies.user import UserAndToken, get_current_user, get_current_user_and_token
|
|
from app.models.score import GameMode
|
|
|
|
from .router import router
|
|
|
|
from fastapi import Path, Security
|
|
from fastapi.responses import RedirectResponse
|
|
from sqlmodel import select
|
|
|
|
|
|
@router.get(
|
|
"/me/beatmapset-favourites",
|
|
response_model=list[int],
|
|
name="获取当前用户收藏的谱面集 ID 列表",
|
|
description="获取当前登录用户收藏的谱面集 ID 列表。",
|
|
tags=["用户", "谱面集"],
|
|
)
|
|
async def get_user_beatmapset_favourites(
|
|
session: Database,
|
|
current_user: Annotated[User, Security(get_current_user, scopes=["identify"])],
|
|
):
|
|
beatmapset_ids = await session.exec(
|
|
select(FavouriteBeatmapset.beatmapset_id).where(FavouriteBeatmapset.user_id == current_user.id)
|
|
)
|
|
return beatmapset_ids.all()
|
|
|
|
|
|
@router.get(
|
|
"/me/{ruleset}",
|
|
response_model=MeResp,
|
|
name="获取当前用户信息 (指定 ruleset)",
|
|
description="获取当前登录用户信息 （含指定 ruleset 统计）。",
|
|
tags=["用户"],
|
|
)
|
|
async def get_user_info_with_ruleset(
|
|
session: Database,
|
|
ruleset: Annotated[GameMode, Path(description="指定 ruleset")],
|
|
user_and_token: Annotated[UserAndToken, Security(get_current_user_and_token, scopes=["identify"])],
|
|
):
|
|
user_resp = await MeResp.from_db(user_and_token[0], session, ruleset, token_id=user_and_token[1].id)
|
|
return user_resp
|
|
|
|
|
|
@router.get(
|
|
"/me/",
|
|
response_model=MeResp,
|
|
name="获取当前用户信息",
|
|
description="获取当前登录用户信息。",
|
|
tags=["用户"],
|
|
)
|
|
async def get_user_info_default(
|
|
session: Database,
|
|
user_and_token: Annotated[UserAndToken, Security(get_current_user_and_token, scopes=["identify"])],
|
|
):
|
|
user_resp = await MeResp.from_db(user_and_token[0], session, None, token_id=user_and_token[1].id)
|
|
return user_resp
|
|
|
|
|
|
@router.put("/users/{user_id}/page", include_in_schema=False)
|
|
async def update_userpage():
|
|
return RedirectResponse(url="/api/private/user/page", status_code=307)
|
|
|
|
|
|
@router.post("/me/validate-bbcode", include_in_schema=False)
|
|
async def validate_bbcode():
|
|
return RedirectResponse(url="/api/private/user/validate-bbcode", status_code=307)
|