122 lines
3.8 KiB
Python
122 lines
3.8 KiB
Python
from typing import Annotated
|
||
|
||
from app.database.beatmap import Beatmap
|
||
from app.database.beatmapset import Beatmapset
|
||
from app.database.beatmapset_ratings import BeatmapRating
|
||
from app.database.score import Score
|
||
from app.dependencies.database import Database
|
||
from app.dependencies.user import ClientUser
|
||
from app.service.beatmapset_update_service import get_beatmapset_update_service
|
||
|
||
from .router import router
|
||
|
||
from fastapi import Body, Depends, HTTPException, Path, Query
|
||
from fastapi_limiter.depends import RateLimiter
|
||
from sqlmodel import col, exists, select
|
||
|
||
|
||
@router.get(
|
||
"/beatmapsets/{beatmapset_id}/can_rate",
|
||
name="判断用户能否为谱面集打分",
|
||
response_model=bool,
|
||
tags=["谱面集", "g0v0 API"],
|
||
)
|
||
async def can_rate_beatmapset(
|
||
beatmapset_id: int,
|
||
session: Database,
|
||
current_user: ClientUser,
|
||
):
|
||
"""检查用户是否可以评价谱面集
|
||
|
||
检查当前用户是否可以对指定的谱面集进行评价
|
||
参数:
|
||
- beatmapset_id: 谱面集ID
|
||
|
||
错误情况:
|
||
- 404: 找不到指定谱面集
|
||
|
||
返回:
|
||
- bool: 用户是否可以评价谱面集
|
||
"""
|
||
user_id = current_user.id
|
||
prev_ratings = (await session.exec(select(BeatmapRating).where(BeatmapRating.user_id == user_id))).first()
|
||
if prev_ratings is not None:
|
||
return False
|
||
query = select(exists()).where(
|
||
Score.user_id == user_id,
|
||
col(Score.beatmap).has(col(Beatmap.beatmapset_id) == beatmapset_id),
|
||
col(Score.passed).is_(True),
|
||
)
|
||
return (await session.exec(query)).first() or False
|
||
|
||
|
||
@router.post(
|
||
"/beatmapsets/{beatmapset_id}/ratings", name="上传对谱面集的打分", status_code=201, tags=["谱面集", "g0v0 API"]
|
||
)
|
||
async def rate_beatmaps(
|
||
beatmapset_id: int,
|
||
session: Database,
|
||
rating: Annotated[int, Body(..., ge=0, le=10)],
|
||
current_user: ClientUser,
|
||
):
|
||
"""为谱面集评分
|
||
|
||
为指定的谱面集添加用户评分，并更新谱面集的评分统计信息
|
||
|
||
参数:
|
||
- beatmapset_id: 谱面集ID
|
||
- rating: 评分
|
||
|
||
错误情况:
|
||
- 404: 找不到指定谱面集
|
||
|
||
返回:
|
||
- 成功: None
|
||
"""
|
||
user_id = current_user.id
|
||
current_beatmapset = (await session.exec(select(exists()).where(Beatmapset.id == beatmapset_id))).first()
|
||
if not current_beatmapset:
|
||
raise HTTPException(404, "Beatmapset Not Found")
|
||
can_rating = await can_rate_beatmapset(beatmapset_id, session, current_user)
|
||
if not can_rating:
|
||
raise HTTPException(403, "User Cannot Rate This Beatmapset")
|
||
new_rating: BeatmapRating = BeatmapRating(beatmapset_id=beatmapset_id, user_id=user_id, rating=rating)
|
||
session.add(new_rating)
|
||
await session.commit()
|
||
|
||
|
||
@router.post(
|
||
"/beatmapsets/{beatmapset_id}/sync",
|
||
name="请求同步谱面集",
|
||
status_code=202,
|
||
tags=["谱面集", "g0v0 API"],
|
||
dependencies=[Depends(RateLimiter(times=50, hours=1))],
|
||
)
|
||
async def sync_beatmapset(
|
||
beatmapset_id: Annotated[int, Path(..., description="谱面集ID")],
|
||
session: Database,
|
||
current_user: ClientUser,
|
||
immediate: Annotated[bool, Query(description="是否立即同步")] = False,
|
||
):
|
||
"""请求同步谱面集
|
||
|
||
请求将指定的谱面集从 Bancho 同步到服务器
|
||
|
||
默认情况下请求会加入同步队列，等待自动同步。
|
||
若设置 `immediate=true`，会尝试立刻同步该谱面集。
|
||
|
||
速率限制:
|
||
- 每个用户每小时最多50次请求
|
||
|
||
参数:
|
||
- beatmapset_id: 谱面集ID
|
||
- immediate: 是否立即同步（默认 false）
|
||
|
||
错误情况:
|
||
- 404: 找不到指定谱面集
|
||
"""
|
||
current_beatmapset = (await session.exec(select(exists()).where(Beatmapset.id == beatmapset_id))).first()
|
||
if not current_beatmapset:
|
||
raise HTTPException(404, "Beatmapset Not Found")
|
||
await get_beatmapset_update_service().add_missing_beatmapset(beatmapset_id, immediate)
|