mirror of
https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot.git
synced 2026-02-16 00:50:31 +00:00
98 lines
3.8 KiB
Python
98 lines
3.8 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from typing import Annotated, Literal, Optional
|
|
|
|
from pydantic import BaseModel, Field, validator
|
|
|
|
CampaignBonusType = Annotated[Literal["balance", "subscription"], Field(description="Тип бонуса кампании")]
|
|
|
|
|
|
class CampaignBase(BaseModel):
|
|
name: str = Field(..., max_length=255)
|
|
start_parameter: str = Field(..., max_length=64, description="Start parameter для deep-link (уникальный)")
|
|
bonus_type: CampaignBonusType
|
|
balance_bonus_kopeks: int = Field(0, ge=0)
|
|
subscription_duration_days: Optional[int] = Field(None, ge=0)
|
|
subscription_traffic_gb: Optional[int] = Field(None, ge=0)
|
|
subscription_device_limit: Optional[int] = Field(None, ge=0)
|
|
subscription_squads: list[str] = Field(default_factory=list)
|
|
|
|
@validator("name", "start_parameter")
|
|
def strip_strings(cls, value: str) -> str: # noqa: D401,B902
|
|
return value.strip()
|
|
|
|
|
|
class CampaignCreateRequest(CampaignBase):
|
|
is_active: bool = True
|
|
|
|
@validator("balance_bonus_kopeks")
|
|
def validate_balance_bonus(cls, value: int, values: dict) -> int: # noqa: D401,B902
|
|
if values.get("bonus_type") == "balance" and value <= 0:
|
|
raise ValueError("balance_bonus_kopeks must be positive for balance bonus")
|
|
return value
|
|
|
|
@validator("subscription_duration_days")
|
|
def validate_subscription_bonus(cls, value: Optional[int], values: dict): # noqa: D401,B902
|
|
if values.get("bonus_type") == "subscription":
|
|
if value is None or value <= 0:
|
|
raise ValueError("subscription_duration_days must be positive for subscription bonus")
|
|
return value
|
|
|
|
|
|
class CampaignResponse(BaseModel):
|
|
id: int
|
|
name: str
|
|
start_parameter: str
|
|
bonus_type: CampaignBonusType
|
|
balance_bonus_kopeks: int
|
|
balance_bonus_rubles: float
|
|
subscription_duration_days: Optional[int] = None
|
|
subscription_traffic_gb: Optional[int] = None
|
|
subscription_device_limit: Optional[int] = None
|
|
subscription_squads: list[str] = Field(default_factory=list)
|
|
is_active: bool
|
|
created_by: Optional[int] = None
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
registrations_count: int = 0
|
|
|
|
|
|
class CampaignListResponse(BaseModel):
|
|
items: list[CampaignResponse]
|
|
total: int
|
|
limit: int
|
|
offset: int
|
|
|
|
|
|
class CampaignUpdateRequest(BaseModel):
|
|
name: Optional[str] = Field(None, max_length=255)
|
|
start_parameter: Optional[str] = Field(None, max_length=64)
|
|
bonus_type: Optional[CampaignBonusType] = None
|
|
balance_bonus_kopeks: Optional[int] = Field(None, ge=0)
|
|
subscription_duration_days: Optional[int] = Field(None, ge=0)
|
|
subscription_traffic_gb: Optional[int] = Field(None, ge=0)
|
|
subscription_device_limit: Optional[int] = Field(None, ge=0)
|
|
subscription_squads: Optional[list[str]] = None
|
|
is_active: Optional[bool] = None
|
|
|
|
@validator("name", "start_parameter", pre=True)
|
|
def strip_optional_strings(cls, value: Optional[str]): # noqa: D401,B902
|
|
if isinstance(value, str):
|
|
return value.strip()
|
|
return value
|
|
|
|
@validator("balance_bonus_kopeks")
|
|
def validate_balance_bonus(cls, value: Optional[int], values: dict): # noqa: D401,B902
|
|
bonus_type = values.get("bonus_type")
|
|
if bonus_type == "balance" and value is not None and value <= 0:
|
|
raise ValueError("balance_bonus_kopeks must be positive for balance bonus")
|
|
return value
|
|
|
|
@validator("subscription_duration_days")
|
|
def validate_subscription_bonus(cls, value: Optional[int], values: dict): # noqa: D401,B902
|
|
bonus_type = values.get("bonus_type")
|
|
if bonus_type == "subscription" and value is not None and value <= 0:
|
|
raise ValueError("subscription_duration_days must be positive for subscription bonus")
|
|
return value
|