添加谱面用户打分(评分)相关接口 (#24)
* feat(database): 添加 beatmap_ratings 表用于用户评分
- 新增 BeatmapRating 模型类,用于表示 beatmap_ratings 表
- 该表包含 id、beatmapset_id、user_id 字段
- 建立与 Beatmapset 和 User 表的关联关系
* feat(beatmapset_ratings): 添加判断用户是否可以对谱面集进行评分的接口
- 新增 /beatmapsets/{beatmapset_id}/can_rate 接口
- 判断用户是否能对谱面集进行过评分,返回True/False
* feat(ratings): 添加为谱面评分的接口
- 新增 POST /beatmapsets/{beatmapset_id} 路由,用于用户给谱面集评分
- 实现谱面集评分的添加和更新逻辑
- 在 BeatmapRating 模型中添加 rating 字段 (漏了最重要的,我真tm丢脸)
* chore(database): 添加alembic数据库迁移脚本
* fix(ratings): 更改上传谱面打分的api路径,防止冲突
* fix(ratings): add changes from pr review
* refactor(ratings): remove swears from code
* feat(ratings): 从beatmapset中移除ratings字段,并改由从beatmap_ratings表中直接计算评分
* chore(deps): 添加 git 包并更新依赖项
- 在 builder 阶段添加了 git 包的安装
* chore(database): 更新数据库连接地址并删除意外的迁移脚本
- 将 Alembic 配置文件中的数据库连接地址从本地地址改为 Docker Compose 中的 mysql 服务地址
- 删除了 migrations/versions 目录下的 dba1f8d9992e_add_beatmap_ratings_table.py 文件
* chore(database): generate alembic script for beatmap ratings
* fix(ratings): apply changes suggested in review
- revert changes to alembic.ini
- add name to apis
- modify migration scripts
* chore: format server.py using ruff
- who forgot to do this?
* fix(migrate): fix remove achievement index
* perf(rating): optimize SQL query
* fix(rating): ensure user can rate beatmapset
* fix(rating): add boundary check
* chore(project): remove submodule
---------
Co-authored-by: MingxuanGame <MingxuanGame@outlook.com>
This commit is contained in:
@@ -2,7 +2,7 @@ FROM ghcr.io/astral-sh/uv:python3.13-bookworm-slim AS builder
|
|||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
|
|
||||||
RUN apt-get update \
|
RUN apt-get update \
|
||||||
&& apt-get install -y gcc pkg-config default-libmysqlclient-dev \
|
&& apt-get install -y git gcc pkg-config default-libmysqlclient-dev \
|
||||||
&& rm -rf /var/lib/apt/lists/* \
|
&& rm -rf /var/lib/apt/lists/* \
|
||||||
&& curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
|
&& curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
|
||||||
|
|
||||||
|
|||||||
@@ -9,6 +9,7 @@ from .beatmapset import (
|
|||||||
Beatmapset,
|
Beatmapset,
|
||||||
BeatmapsetResp,
|
BeatmapsetResp,
|
||||||
)
|
)
|
||||||
|
from .beatmapset_ratings import BeatmapRating
|
||||||
from .best_score import BestScore
|
from .best_score import BestScore
|
||||||
from .chat import (
|
from .chat import (
|
||||||
ChannelType,
|
ChannelType,
|
||||||
@@ -71,6 +72,7 @@ __all__ = [
|
|||||||
"Beatmap",
|
"Beatmap",
|
||||||
"BeatmapPlaycounts",
|
"BeatmapPlaycounts",
|
||||||
"BeatmapPlaycountsResp",
|
"BeatmapPlaycountsResp",
|
||||||
|
"BeatmapRating",
|
||||||
"BeatmapResp",
|
"BeatmapResp",
|
||||||
"Beatmapset",
|
"Beatmapset",
|
||||||
"BeatmapsetResp",
|
"BeatmapsetResp",
|
||||||
|
|||||||
@@ -93,7 +93,6 @@ class BeatmapsetBase(SQLModel):
|
|||||||
# TODO: events: Optional[list[BeatmapsetEvent]] = None
|
# TODO: events: Optional[list[BeatmapsetEvent]] = None
|
||||||
|
|
||||||
pack_tags: list[str] = Field(default=[], sa_column=Column(JSON))
|
pack_tags: list[str] = Field(default=[], sa_column=Column(JSON))
|
||||||
ratings: list[int] | None = Field(default=None, sa_column=Column(JSON))
|
|
||||||
# TODO: related_users: Optional[list[User]] = None
|
# TODO: related_users: Optional[list[User]] = None
|
||||||
# TODO: user: Optional[User] = Field(default=None)
|
# TODO: user: Optional[User] = Field(default=None)
|
||||||
track_id: int | None = Field(default=None, index=True) # feature artist?
|
track_id: int | None = Field(default=None, index=True) # feature artist?
|
||||||
@@ -259,9 +258,21 @@ class BeatmapsetResp(BeatmapsetBase):
|
|||||||
**beatmapset.model_dump(),
|
**beatmapset.model_dump(),
|
||||||
}
|
}
|
||||||
|
|
||||||
# 确保 ratings 字段不为 null,避免客户端崩溃
|
if session is not None:
|
||||||
if update.get("ratings") is None:
|
# 从数据库读取对应谱面集的评分
|
||||||
update["ratings"] = []
|
from .beatmapset_ratings import BeatmapRating
|
||||||
|
|
||||||
|
beatmapset_all_ratings = (
|
||||||
|
await session.exec(select(BeatmapRating).where(BeatmapRating.beatmapset_id == beatmapset.id))
|
||||||
|
).all()
|
||||||
|
ratings_list = [0] * 11
|
||||||
|
for rating in beatmapset_all_ratings:
|
||||||
|
ratings_list[rating.rating] += 1
|
||||||
|
update["ratings"] = ratings_list
|
||||||
|
else:
|
||||||
|
# 返回非空值避免客户端崩溃
|
||||||
|
if update.get("ratings") is None:
|
||||||
|
update["ratings"] = []
|
||||||
|
|
||||||
beatmap_status = beatmapset.beatmap_status
|
beatmap_status = beatmapset.beatmap_status
|
||||||
if settings.enable_all_beatmap_leaderboard and not beatmap_status.has_leaderboard():
|
if settings.enable_all_beatmap_leaderboard and not beatmap_status.has_leaderboard():
|
||||||
|
|||||||
20
app/database/beatmapset_ratings.py
Normal file
20
app/database/beatmapset_ratings.py
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from app.database.beatmapset import Beatmapset
|
||||||
|
from app.database.lazer_user import User
|
||||||
|
|
||||||
|
from sqlmodel import BigInteger, Column, Field, ForeignKey, Relationship, SQLModel
|
||||||
|
|
||||||
|
|
||||||
|
class BeatmapRating(SQLModel, table=True):
|
||||||
|
__tablename__: str = "beatmap_ratings"
|
||||||
|
id: int | None = Field(
|
||||||
|
default=None,
|
||||||
|
sa_column=Column(BigInteger, primary_key=True, autoincrement=True),
|
||||||
|
)
|
||||||
|
beatmapset_id: int = Field(foreign_key="beatmapsets.id", index=True)
|
||||||
|
user_id: int = Field(sa_column=Column(BigInteger, ForeignKey("lazer_users.id"), index=True))
|
||||||
|
rating: int
|
||||||
|
|
||||||
|
beatmapset: Beatmapset = Relationship()
|
||||||
|
user: User = Relationship()
|
||||||
@@ -1,6 +1,6 @@
|
|||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
from . import avatar, cover, oauth, relationship, team, username # noqa: F401
|
from . import avatar, beatmapset_ratings, cover, oauth, relationship, team, username # noqa: F401
|
||||||
from .router import router as private_router
|
from .router import router as private_router
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
|
|||||||
77
app/router/private/beatmapset_ratings.py
Normal file
77
app/router/private/beatmapset_ratings.py
Normal file
@@ -0,0 +1,77 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from app.database.beatmap import Beatmap
|
||||||
|
from app.database.beatmapset import Beatmapset
|
||||||
|
from app.database.beatmapset_ratings import BeatmapRating
|
||||||
|
from app.database.lazer_user import User
|
||||||
|
from app.database.score import Score
|
||||||
|
from app.dependencies.database import Database
|
||||||
|
from app.dependencies.user import get_client_user
|
||||||
|
|
||||||
|
from .router import router
|
||||||
|
|
||||||
|
from fastapi import Body, HTTPException, Security
|
||||||
|
from sqlmodel import col, exists, select
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/beatmapsets/{beatmapset_id}/can_rate", name="判断用户能否为谱面集打分", response_model=bool)
|
||||||
|
async def can_rate_beatmapset(
|
||||||
|
beatmapset_id: int,
|
||||||
|
session: Database,
|
||||||
|
current_user: User = Security(get_client_user),
|
||||||
|
):
|
||||||
|
"""检查用户是否可以评价谱面集
|
||||||
|
|
||||||
|
检查当前用户是否可以对指定的谱面集进行评价
|
||||||
|
参数:
|
||||||
|
- 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)
|
||||||
|
async def rate_beatmaps(
|
||||||
|
beatmapset_id: int,
|
||||||
|
session: Database,
|
||||||
|
rating: int = Body(..., ge=0, le=10),
|
||||||
|
current_user: User = Security(get_client_user),
|
||||||
|
):
|
||||||
|
"""为谱面集评分
|
||||||
|
|
||||||
|
为指定的谱面集添加用户评分,并更新谱面集的评分统计信息
|
||||||
|
|
||||||
|
参数:
|
||||||
|
- 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()
|
||||||
51
migrations/versions/24a32515292d_add_beatmap_ratings.py
Normal file
51
migrations/versions/24a32515292d_add_beatmap_ratings.py
Normal file
@@ -0,0 +1,51 @@
|
|||||||
|
"""add beatmap ratings
|
||||||
|
|
||||||
|
Revision ID: 24a32515292d
|
||||||
|
Revises: af88493881eb
|
||||||
|
Create Date: 2025-08-28 11:36:17.874090
|
||||||
|
|
||||||
|
"""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from collections.abc import Sequence
|
||||||
|
|
||||||
|
from alembic import op
|
||||||
|
import sqlalchemy as sa
|
||||||
|
|
||||||
|
# revision identifiers, used by Alembic.
|
||||||
|
revision: str = "24a32515292d"
|
||||||
|
down_revision: str | Sequence[str] | None = "af88493881eb"
|
||||||
|
branch_labels: str | Sequence[str] | None = None
|
||||||
|
depends_on: str | Sequence[str] | None = None
|
||||||
|
|
||||||
|
|
||||||
|
def upgrade() -> None:
|
||||||
|
"""Upgrade schema."""
|
||||||
|
# ### commands auto generated by Alembic - please adjust! ###
|
||||||
|
op.create_table(
|
||||||
|
"beatmap_ratings",
|
||||||
|
sa.Column("id", sa.BigInteger(), autoincrement=True, nullable=False),
|
||||||
|
sa.Column("beatmapset_id", sa.Integer(), nullable=False),
|
||||||
|
sa.Column("user_id", sa.BigInteger(), nullable=True),
|
||||||
|
sa.Column("rating", sa.Integer(), nullable=False),
|
||||||
|
sa.ForeignKeyConstraint(
|
||||||
|
["beatmapset_id"],
|
||||||
|
["beatmapsets.id"],
|
||||||
|
),
|
||||||
|
sa.ForeignKeyConstraint(
|
||||||
|
["user_id"],
|
||||||
|
["lazer_users.id"],
|
||||||
|
),
|
||||||
|
sa.PrimaryKeyConstraint("id"),
|
||||||
|
)
|
||||||
|
op.create_index(op.f("ix_beatmap_ratings_beatmapset_id"), "beatmap_ratings", ["beatmapset_id"], unique=False)
|
||||||
|
op.create_index(op.f("ix_beatmap_ratings_user_id"), "beatmap_ratings", ["user_id"], unique=False)
|
||||||
|
# ### end Alembic commands ###
|
||||||
|
|
||||||
|
|
||||||
|
def downgrade() -> None:
|
||||||
|
"""Downgrade schema."""
|
||||||
|
# ### commands auto generated by Alembic - please adjust! ###
|
||||||
|
op.drop_table("beatmap_ratings")
|
||||||
|
# ### end Alembic commands ###
|
||||||
Reference in New Issue
Block a user