106 lines
2.6 KiB
Python
106 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from enum import Enum
|
|
|
|
from app.database import User
|
|
from app.database.beatmap import Beatmap
|
|
from app.models.mods import APIMod
|
|
|
|
from .model import UTCBaseModel
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class RoomCategory(str, Enum):
|
|
NORMAL = "normal"
|
|
SPOTLIGHT = "spotlight"
|
|
FEATURED_ARTIST = "featured_artist"
|
|
DAILY_CHALLENGE = "daily_challenge"
|
|
|
|
|
|
class MatchType(str, Enum):
|
|
PLAYLISTS = "playlists"
|
|
HEAD_TO_HEAD = "head_to_head"
|
|
TEAM_VERSUS = "team_versus"
|
|
|
|
|
|
class QueueMode(str, Enum):
|
|
HOST_ONLY = "host_only"
|
|
ALL_PLAYERS = "all_players"
|
|
ALL_PLAYERS_ROUND_ROBIN = "all_players_round_robin"
|
|
|
|
|
|
class RoomAvailability(str, Enum):
|
|
PUBLIC = "public"
|
|
FRIENDS_ONLY = "friends_only"
|
|
INVITE_ONLY = "invite_only"
|
|
|
|
|
|
class RoomStatus(str, Enum):
|
|
IDLE = "idle"
|
|
PLAYING = "playing"
|
|
|
|
|
|
class PlaylistItem(UTCBaseModel):
|
|
id: int | None
|
|
owner_id: int
|
|
ruleset_id: int
|
|
expired: bool
|
|
playlist_order: int | None
|
|
played_at: datetime | None
|
|
allowed_mods: list[APIMod] = Field(default_factory=list)
|
|
required_mods: list[APIMod] = Field(default_factory=list)
|
|
beatmap_id: int
|
|
beatmap: Beatmap | None
|
|
freestyle: bool
|
|
|
|
|
|
class RoomPlaylistItemStats(BaseModel):
|
|
count_active: int
|
|
count_total: int
|
|
ruleset_ids: list[int] = []
|
|
|
|
|
|
class RoomDifficultyRange(BaseModel):
|
|
min: float
|
|
max: float
|
|
|
|
|
|
class ItemAttemptsCount(BaseModel):
|
|
id: int
|
|
attempts: int
|
|
passed: bool
|
|
|
|
|
|
class PlaylistAggregateScore(BaseModel):
|
|
playlist_item_attempts: list[ItemAttemptsCount]
|
|
|
|
|
|
class Room(UTCBaseModel):
|
|
id: int | None
|
|
name: str = ""
|
|
password: str | None
|
|
has_password: bool = False
|
|
host: User | None
|
|
category: RoomCategory = RoomCategory.NORMAL
|
|
duration: int | None
|
|
starts_at: datetime | None
|
|
ends_at: datetime | None
|
|
participant_count: int = 0
|
|
recent_participants: list[User] = Field(default_factory=list)
|
|
max_attempts: int | None
|
|
playlist: list[PlaylistItem] = Field(default_factory=list)
|
|
playlist_item_stats: RoomPlaylistItemStats | None
|
|
difficulty_range: RoomDifficultyRange | None
|
|
type: MatchType = MatchType.PLAYLISTS
|
|
queue_mode: QueueMode = QueueMode.HOST_ONLY
|
|
auto_skip: bool = False
|
|
auto_start_duration: int = 0
|
|
current_user_score: PlaylistAggregateScore | None
|
|
current_playlist_item: PlaylistItem | None
|
|
channel_id: int = 0
|
|
status: RoomStatus = RoomStatus.IDLE
|
|
# availability 字段在当前序列化中未包含，但可能在某些场景下需要
|
|
availability: RoomAvailability | None
|