Stardream
143 lines
4.6 KiB
Python
143 lines
4.6 KiB
Python
import uuid
|
|
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
from pydantic import BaseModel, Field, field_validator
|
|
|
|
|
|
class PrizeCreate(BaseModel):
|
|
name: str = Field(..., min_length=1, max_length=255)
|
|
description: Optional[str] = None
|
|
quantity: int = Field(1, ge=1, le=1000)
|
|
|
|
|
|
class PrizeResponse(BaseModel):
|
|
id: uuid.UUID
|
|
name: str
|
|
description: Optional[str] = None
|
|
quantity: int
|
|
sort_order: int
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
def _to_naive_utc(v: datetime | None) -> datetime | None:
|
|
if v is None:
|
|
return None
|
|
if v.tzinfo is not None:
|
|
from datetime import timezone
|
|
return v.astimezone(timezone.utc).replace(tzinfo=None)
|
|
return v
|
|
|
|
|
|
class LotteryCreate(BaseModel):
|
|
title: str = Field(..., min_length=1, max_length=255)
|
|
description: Optional[str] = None
|
|
contact_info: Optional[str] = None
|
|
target_chat_id: Optional[int] = None
|
|
target_chat_title: Optional[str] = None
|
|
require_membership: bool = False
|
|
required_chat_ids: list[int] = Field(default_factory=list)
|
|
required_chat_titles: list[str] = Field(default_factory=list)
|
|
required_chat_types: list[str] = Field(default_factory=list)
|
|
required_chat_usernames: list[str] = Field(default_factory=list)
|
|
allow_multiple_wins: bool = False
|
|
invite_link_chat_ids: list[int] = Field(default_factory=list)
|
|
required_chat_passphrases: list[str] = Field(default_factory=list)
|
|
max_participants: Optional[int] = Field(None, ge=1)
|
|
draw_at: Optional[datetime] = None
|
|
auto_draw_count: Optional[int] = Field(None, ge=1)
|
|
creator_timezone: Optional[str] = Field(None, max_length=64)
|
|
|
|
@field_validator("draw_at", mode="after")
|
|
@classmethod
|
|
def normalize_draw_at(cls, v: datetime | None) -> datetime | None:
|
|
return _to_naive_utc(v)
|
|
|
|
|
|
class LotteryUpdate(BaseModel):
|
|
title: Optional[str] = Field(None, min_length=1, max_length=255)
|
|
description: Optional[str] = None
|
|
contact_info: Optional[str] = None
|
|
target_chat_id: Optional[int] = None
|
|
target_chat_title: Optional[str] = None
|
|
require_membership: Optional[bool] = None
|
|
required_chat_ids: Optional[list[int]] = None
|
|
required_chat_titles: Optional[list[str]] = None
|
|
required_chat_types: Optional[list[str]] = None
|
|
required_chat_usernames: Optional[list[str]] = None
|
|
allow_multiple_wins: Optional[bool] = None
|
|
invite_link_chat_ids: Optional[list[int]] = None
|
|
required_chat_passphrases: Optional[list[str]] = None
|
|
max_participants: Optional[int] = Field(None, ge=1)
|
|
draw_at: Optional[datetime] = None
|
|
auto_draw_count: Optional[int] = Field(None, ge=1)
|
|
creator_timezone: Optional[str] = Field(None, max_length=64)
|
|
|
|
@field_validator("draw_at", mode="after")
|
|
@classmethod
|
|
def normalize_draw_at(cls, v: datetime | None) -> datetime | None:
|
|
return _to_naive_utc(v)
|
|
|
|
|
|
class LotteryResponse(BaseModel):
|
|
id: uuid.UUID
|
|
creator_id: int
|
|
title: str
|
|
description: Optional[str] = None
|
|
contact_info: Optional[str] = None
|
|
status: str
|
|
target_chat_id: Optional[int] = None
|
|
target_chat_title: Optional[str] = None
|
|
require_membership: bool
|
|
required_chat_ids: list[int] = Field(default_factory=list)
|
|
required_chat_titles: list[str] = Field(default_factory=list)
|
|
required_chat_types: list[str] = Field(default_factory=list)
|
|
required_chat_usernames: list[str] = Field(default_factory=list)
|
|
announcement_targets: list[dict] = Field(default_factory=list)
|
|
allow_multiple_wins: bool = False
|
|
invite_link_chat_ids: list[int] = Field(default_factory=list)
|
|
required_chat_passphrases: list[str] = Field(default_factory=list)
|
|
max_participants: Optional[int] = None
|
|
draw_at: Optional[datetime] = None
|
|
auto_draw_count: Optional[int] = None
|
|
creator_timezone: Optional[str] = None
|
|
participant_count: int = 0
|
|
prize_count: int = 0
|
|
total_prizes: int = 0
|
|
created_at: datetime
|
|
drawn_at: Optional[datetime] = None
|
|
draw_trigger: Optional[str] = None
|
|
prizes: list[PrizeResponse] = Field(default_factory=list)
|
|
is_winner: Optional[bool] = None
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class ParticipantResponse(BaseModel):
|
|
user_id: int
|
|
username: Optional[str] = None
|
|
first_name: str
|
|
last_name: Optional[str] = None
|
|
joined_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class WinnerResponse(BaseModel):
|
|
user_id: int
|
|
username: Optional[str] = None
|
|
first_name: str
|
|
last_name: Optional[str] = None
|
|
prize_name: str
|
|
drawn_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class ChatInfo(BaseModel):
|
|
id: int
|
|
title: str
|
|
username: Optional[str] = None
|
|
chat_type: str
|