partner-core/tg_models.py

48 lines
998 B
Python

from pydantic import BaseModel, Field
from typing import Optional, List
from uuid import UUID
# TG Models
class RefResponse(BaseModel):
ref: str
description: str
class RefAddRequest(BaseModel):
description: str
class TokenRequest(BaseModel):
tg_id: int
class RegisterRequest(BaseModel):
tg_id: int
chat_id: Optional[int] = None
phone: Optional[str] = None
name: Optional[str] = None
login: Optional[str] = None
company_key: str
# New Response Models for TG APIs
class RefAddResponse(BaseModel):
ref: str
class RefStatItem(BaseModel):
description: str
sales: int
income: float
class RefStatResponse(BaseModel):
refData: List[RefStatItem]
class StatResponse(BaseModel):
totalSales: int
totalIncome: float
availableWithdrawal: float
class PromoCodeAddRequest(BaseModel):
description: Optional[str] = None
class PromoCodeResponse(BaseModel):
promocode: str
perc: float
description: Optional[str] = None