baoxiang/backend/app/schemas/game.py

import json
from pydantic import BaseModel, Field
from typing import Optional
from datetime import datetime

class ChestBase(BaseModel):
    title: str = Field(..., min_length=1, max_length=200)
    option_a: str = Field(..., min_length=1, max_length=100)
    option_b: str = Field(..., min_length=1, max_length=100)
    countdown_seconds: int = Field(default=300, ge=10, le=3600)  # 10秒到1小时


class ChestCreate(ChestBase):
    pass


class ChestResponse(ChestBase):
    id: int
    streamer_id: int
    status: int
    pool_a: int
    pool_b: int
    total_bets: int
    locked_at: Optional[datetime] = None
    settled_at: Optional[datetime] = None
    result: Optional[str] = None
    created_at: datetime
    time_remaining: int = 0  # 改为非可选字段，默认值为0

    # 计算属性
    @property
    def total_pool(self) -> int:
        return self.pool_a + self.pool_b

    @property
    def odds_a(self) -> float:
        """计算A边赔率 (1 : X)"""
        if self.pool_a == 0:
            return 0.0
        # 修改赔率计算方式，只对获胜方抽水
        return round((self.pool_a + self.pool_b * 0.9) / self.pool_a, 2)

    @property
    def odds_b(self) -> float:
        """计算B边赔率 (1 : X)"""
        if self.pool_b == 0:
            return 0.0
        # 修改赔率计算方式，只对获胜方抽水
        return round((self.pool_b + self.pool_a * 0.9) / self.pool_b, 2)

    class Config:
        from_attributes = True


class BetCreate(BaseModel):
    chest_id: int
    option: str = Field(..., pattern="^[AB]$")
    amount: int = Field(..., gt=0)


class BetResponse(BaseModel):
    id: int
    user_id: int
    chest_id: int
    option: str
    amount: int
    payout: int
    status: str
    created_at: datetime

    class Config:
        from_attributes = True


class ChestSettle(BaseModel):
    """宝箱结算"""
    result: str = Field(..., pattern="^(A|B|REFUND)$")


class ChestLock(BaseModel):
    """宝箱封盘"""
    pass


# ==================== 宝箱模板相关 schemas ====================

class ChestTemplateBase(BaseModel):
    """宝箱模板基础模型"""
    name: str = Field(..., min_length=1, max_length=100, description="模板名称")
    title: str = Field(..., min_length=1, max_length=200, description="宝箱标题")
    option_a: str = Field(..., min_length=1, max_length=100, description="选项A")
    option_b: str = Field(..., min_length=1, max_length=100, description="选项B")
    countdown_seconds: int = Field(default=300, ge=10, le=3600, description="倒计时(秒)")


class ChestTemplateCreate(ChestTemplateBase):
    """创建宝箱模板"""
    pass


class ChestTemplateUpdate(BaseModel):
    """更新宝箱模板"""
    name: Optional[str] = Field(None, min_length=1, max_length=100, description="模板名称")
    title: Optional[str] = Field(None, min_length=1, max_length=200, description="宝箱标题")
    option_a: Optional[str] = Field(None, min_length=1, max_length=100, description="选项A")
    option_b: Optional[str] = Field(None, min_length=1, max_length=100, description="选项B")
    countdown_seconds: Optional[int] = Field(None, ge=10, le=3600, description="倒计时(秒)")


class ChestTemplateResponse(ChestTemplateBase):
    """宝箱模板响应模型"""
    id: int
    streamer_id: int
    created_at: datetime
    updated_at: datetime

    class Config:
        from_attributes = True