You've already forked Promocode-API
mirror of
https://github.com/devitq/Promocode-API.git
synced 2026-05-22 20:57:11 +00:00
146 lines
3.7 KiB
Python
146 lines
3.7 KiB
Python
import datetime
|
|
import uuid
|
|
from typing import Any, ClassVar, Literal
|
|
|
|
from ninja import ModelSchema, Schema
|
|
from pydantic import Field, StrictInt, field_validator
|
|
from pydantic_extra_types.country import CountryAlpha2
|
|
|
|
from apps.business.models import Business
|
|
from apps.promo.models import Promocode, PromocodeTarget
|
|
|
|
|
|
class BusinessSignUpIn(ModelSchema):
|
|
class Meta:
|
|
model = Business
|
|
fields: ClassVar[list[str]] = [
|
|
Business.name.field.name,
|
|
Business.email.field.name,
|
|
Business.password.field.name,
|
|
]
|
|
|
|
|
|
class BusinessSignUpOut(Schema):
|
|
token: str
|
|
company_id: uuid.UUID
|
|
|
|
|
|
class BusinessSignInIn(ModelSchema):
|
|
class Meta:
|
|
model = Business
|
|
fields: ClassVar[list[str]] = [
|
|
Business.email.field.name,
|
|
Business.password.field.name,
|
|
]
|
|
|
|
|
|
class BusinessSignInOut(Schema):
|
|
token: str
|
|
|
|
|
|
class PromocodeTarget(ModelSchema):
|
|
categories: list[str] | None = None
|
|
age_from: StrictInt | None = None
|
|
age_until: StrictInt | None = None
|
|
|
|
class Meta:
|
|
model = PromocodeTarget
|
|
fields: ClassVar[list[str]] = [PromocodeTarget.country.field.name]
|
|
|
|
|
|
class CreatePromocodeIn(ModelSchema):
|
|
target: PromocodeTarget
|
|
promo_unique: list[str] | None = None
|
|
max_count: StrictInt
|
|
|
|
class Meta:
|
|
model = Promocode
|
|
fields: ClassVar[list[str]] = [
|
|
Promocode.description.field.name,
|
|
Promocode.image_url.field.name,
|
|
Promocode.active_from.field.name,
|
|
Promocode.active_until.field.name,
|
|
Promocode.mode.field.name,
|
|
Promocode.promo_common.field.name,
|
|
]
|
|
|
|
@field_validator("target", mode="before")
|
|
def validate_target(cls, value: Any) -> Any:
|
|
if not isinstance(value, dict) and not isinstance(
|
|
value,
|
|
PromocodeTarget,
|
|
):
|
|
err = "The 'target' field must be a valid object."
|
|
raise ValueError(err)
|
|
return value
|
|
|
|
|
|
class CreatePromocodeOut(Schema):
|
|
id: uuid.UUID
|
|
|
|
|
|
class PromocodeListFilters(Schema):
|
|
limit: int = Field(
|
|
10, ge=0, description="Limit must be greater than or equal 0"
|
|
)
|
|
offset: int = Field(
|
|
0, ge=0, description="Offset must be greater than or equal to 0"
|
|
)
|
|
sort_by: Literal["active_from", "active_until", None] = None
|
|
country__in: list[CountryAlpha2] = Field(None, alias="country")
|
|
|
|
|
|
class PromocodeTargetViewOut(Schema):
|
|
age_from: int | None
|
|
age_until: int | None
|
|
country: str | None
|
|
categories: list[str] | None
|
|
|
|
|
|
class PromocodeViewOut(Schema):
|
|
promo_id: uuid.UUID
|
|
company_id: uuid.UUID
|
|
description: str
|
|
image_url: str | None
|
|
target: PromocodeTargetViewOut
|
|
max_count: int
|
|
active_from: datetime.date | None
|
|
active_until: datetime.date | None
|
|
mode: str
|
|
promo_common: str | None
|
|
promo_unique: list[str] | None
|
|
company_name: str
|
|
like_count: int
|
|
used_count: int
|
|
active: bool
|
|
|
|
|
|
class PatchPromocodeIn(Schema):
|
|
description: str | None = None
|
|
image_url: str | None = None
|
|
target: PromocodeTarget | None = None
|
|
max_count: StrictInt | None = None
|
|
active_from: datetime.date | None = None
|
|
active_until: datetime.date | None = None
|
|
|
|
@staticmethod
|
|
@field_validator("target", mode="before")
|
|
def validate_target(value: Any) -> Any:
|
|
if not isinstance(value, dict) and not isinstance(
|
|
value,
|
|
PromocodeTarget,
|
|
):
|
|
err = "The 'target' field must be a valid object."
|
|
raise TypeError(err)
|
|
return value
|
|
|
|
|
|
class PromocodeStatsForCountry(Schema):
|
|
country: str
|
|
activations_count: int
|
|
|
|
|
|
class PromocodeStats(Schema):
|
|
activations_count: int
|
|
countries: list[PromocodeStatsForCountry] | None = None
|