94 lines
2.9 KiB
Python
94 lines
2.9 KiB
Python
from datetime import datetime
|
||
from typing import Self
|
||
from pydantic import BaseModel, Field, model_validator
|
||
|
||
from app.models.game import GameType
|
||
from app.models.challenge import ProofType
|
||
from app.schemas.user import UserPublic
|
||
|
||
|
||
class GameBase(BaseModel):
|
||
title: str = Field(..., min_length=1, max_length=100)
|
||
download_url: str = Field(..., min_length=1)
|
||
genre: str | None = Field(None, max_length=50)
|
||
|
||
|
||
class GameCreate(GameBase):
|
||
cover_url: str | None = None
|
||
|
||
# Тип игры
|
||
game_type: GameType = GameType.CHALLENGES
|
||
|
||
# Поля для типа "Прохождение"
|
||
playthrough_points: int | None = Field(None, ge=1, le=1000)
|
||
playthrough_description: str | None = None
|
||
playthrough_proof_type: ProofType | None = None
|
||
playthrough_proof_hint: str | None = None
|
||
|
||
@model_validator(mode='after')
|
||
def validate_playthrough_fields(self) -> Self:
|
||
if self.game_type == GameType.PLAYTHROUGH:
|
||
if self.playthrough_points is None:
|
||
raise ValueError('playthrough_points обязателен для типа "Прохождение"')
|
||
if self.playthrough_description is None:
|
||
raise ValueError('playthrough_description обязателен для типа "Прохождение"')
|
||
if self.playthrough_proof_type is None:
|
||
raise ValueError('playthrough_proof_type обязателен для типа "Прохождение"')
|
||
return self
|
||
|
||
|
||
class GameUpdate(BaseModel):
|
||
title: str | None = Field(None, min_length=1, max_length=100)
|
||
download_url: str | None = None
|
||
genre: str | None = None
|
||
|
||
# Тип игры
|
||
game_type: GameType | None = None
|
||
|
||
# Поля для типа "Прохождение"
|
||
playthrough_points: int | None = Field(None, ge=1, le=1000)
|
||
playthrough_description: str | None = None
|
||
playthrough_proof_type: ProofType | None = None
|
||
playthrough_proof_hint: str | None = None
|
||
|
||
|
||
class GameShort(BaseModel):
|
||
id: int
|
||
title: str
|
||
cover_url: str | None = None
|
||
download_url: str
|
||
game_type: str = "challenges"
|
||
|
||
class Config:
|
||
from_attributes = True
|
||
|
||
|
||
class GameResponse(GameBase):
|
||
id: int
|
||
cover_url: str | None = None
|
||
status: str = "pending"
|
||
proposed_by: UserPublic | None = None
|
||
approved_by: UserPublic | None = None
|
||
challenges_count: int = 0
|
||
created_at: datetime
|
||
|
||
# Тип игры
|
||
game_type: str = "challenges"
|
||
|
||
# Поля для типа "Прохождение"
|
||
playthrough_points: int | None = None
|
||
playthrough_description: str | None = None
|
||
playthrough_proof_type: str | None = None
|
||
playthrough_proof_hint: str | None = None
|
||
|
||
class Config:
|
||
from_attributes = True
|
||
|
||
|
||
class PlaythroughInfo(BaseModel):
|
||
"""Информация о прохождении для игр типа playthrough"""
|
||
description: str
|
||
points: int
|
||
proof_type: str
|
||
proof_hint: str | None = None
|