dibang-xiaoshou/backend/app/schemas/master_data.py

from __future__ import annotations

from pydantic import BaseModel, Field


class CompanyBase(BaseModel):
    name: str = Field(min_length=1, max_length=255)
    remarks: str | None = Field(default=None)

    model_config = {"from_attributes": True}


class CompanyCreate(CompanyBase):
    pass


class CompanyUpdate(BaseModel):
    name: str | None = Field(default=None, min_length=1, max_length=255)
    remarks: str | None = Field(default=None)


class CompanyRead(CompanyBase):
    id: int


class CargoBase(BaseModel):
    name: str = Field(min_length=1, max_length=255)
    remarks: str | None = Field(default=None)

    model_config = {"from_attributes": True}


class CargoCreate(CargoBase):
    pass


class CargoUpdate(BaseModel):
    name: str | None = Field(default=None, min_length=1, max_length=255)
    remarks: str | None = Field(default=None)


class CargoRead(CargoBase):
    id: int


# ==================== 车牌号 ====================


class VehicleBase(BaseModel):
    vehicle_no: str = Field(min_length=1, max_length=255, alias="vehicleNo")
    remarks: str | None = Field(default=None)

    model_config = {"from_attributes": True, "populate_by_name": True}


class VehicleCreate(VehicleBase):
    pass


class VehicleUpdate(BaseModel):
    vehicle_no: str | None = Field(default=None, min_length=1, max_length=255, alias="vehicleNo")
    remarks: str | None = Field(default=None)

    model_config = {"populate_by_name": True}


class VehicleRead(VehicleBase):
    id: int