119 lines
3.5 KiB
Python
119 lines
3.5 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
|
|
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: int,
|
|
session: Database,
|
|
current_user: ClientUser,
|
|
):
|
|
"""请求同步谱面集
|
|
|
|
请求将指定的谱面集从 Bancho 同步到服务器
|
|
|
|
请求发送后会加入同步队列，等待自动同步
|
|
|
|
速率限制:
|
|
- 每个用户每小时最多50次请求
|
|
|
|
参数:
|
|
- beatmapset_id: 谱面集ID
|
|
|
|
错误情况:
|
|
- 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)
|