Vladless
106 lines
1.9 KiB
Python
106 lines
1.9 KiB
Python
from pydantic import BaseModel
|
|
from typing import Optional, Union, Literal
|
|
from datetime import datetime
|
|
|
|
|
|
class PaymentBase(BaseModel):
|
|
tg_id: int
|
|
amount: float
|
|
payment_system: str
|
|
status: Literal["success", "pending", "failed"]
|
|
|
|
|
|
class PaymentResponse(PaymentBase):
|
|
id: int
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class ReferralResponse(BaseModel):
|
|
referred_tg_id: int
|
|
referrer_tg_id: int
|
|
reward_issued: bool = False
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class NotificationResponse(BaseModel):
|
|
tg_id: int
|
|
notification_type: str
|
|
last_notification_time: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class GiftBase(BaseModel):
|
|
sender_tg_id: int
|
|
recipient_tg_id: Optional[int] = None
|
|
selected_months: int
|
|
expiry_time: datetime
|
|
gift_link: str
|
|
is_used: bool = False
|
|
is_unlimited: bool = False
|
|
max_usages: Optional[int] = None
|
|
tariff_id: Optional[int] = None
|
|
|
|
|
|
class GiftResponse(GiftBase):
|
|
gift_id: str
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class GiftUsageResponse(BaseModel):
|
|
gift_id: str
|
|
tg_id: int
|
|
used_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class ManualBanResponse(BaseModel):
|
|
tg_id: int
|
|
banned_at: datetime
|
|
reason: str
|
|
banned_by: int
|
|
until: Optional[datetime] = None
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class TemporaryDataResponse(BaseModel):
|
|
tg_id: int
|
|
state: str
|
|
data: dict
|
|
updated_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class BlockedUserResponse(BaseModel):
|
|
tg_id: int
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class TrackingSourceResponse(BaseModel):
|
|
id: int
|
|
name: str
|
|
code: str
|
|
type: str
|
|
created_by: int
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|