File size: 1,568 Bytes
9e798a1 7754df5 9e798a1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
from pydantic import BaseModel, Field
from typing import Optional, List
from decimal import Decimal
class BaseResponse(BaseModel):
code: int
message: str
payload: Optional[dict] = None
class CreatePlanRequest(BaseModel):
name: str = Field(..., description="Name of the subscription plan")
amount: Decimal = Field(..., description="Cost of the plan")
duration: int = Field(..., description="Duration of the subscription in hours")
download_speed: float = Field(..., description="Download speed in Mbps")
upload_speed: float = Field(..., description="Upload speed in Mbps")
is_promo: Optional[bool] = Field(False, description="is it a promo")
promo_days: Optional[int] = Field(None, description="promotion days")
class UpdatePlanRequest(BaseModel):
name: Optional[str] = Field(None, description="Name of the subscription plan")
amount: Optional[Decimal] = Field(None, description="Cost of the plan")
duration: Optional[int] = Field(
None, description="Duration of the subscription in hours"
)
download_speed: Optional[float] = Field(None, description="Download speed in Mbps")
upload_speed: Optional[float] = Field(None, description="Upload speed in Mbps")
class PlanResponse(BaseModel):
id: str
name: str
amount: Decimal
duration: int
download_speed: float
upload_speed: float
class Config:
from_attributes = True
class PlanListResponse(BaseModel):
plans: List[PlanResponse]
total_count: int
class Config:
from_attributes = True
|