from __future__ import annotations

from sqlmodel import Field, SQLModel


class Company(SQLModel, table=True):
    __tablename__ = "companies"

    id: int | None = Field(default=None, primary_key=True)
    name: str = Field(index=True, unique=True, max_length=255)
    remarks: str | None = Field(default=None)
    is_deleted: int = Field(default=0, index=True)
    created_at: str = Field(max_length=255)


class Cargo(SQLModel, table=True):
    __tablename__ = "cargos"

    id: int | None = Field(default=None, primary_key=True)
    name: str = Field(index=True, unique=True, max_length=255)
    remarks: str | None = Field(default=None)
    is_deleted: int = Field(default=0, index=True)
    created_at: str = Field(max_length=255)


class Vehicle(SQLModel, table=True):
    """车牌号基础资料表"""
    __tablename__ = "vehicles"

    id: int | None = Field(default=None, primary_key=True)
    vehicle_no: str = Field(index=True, unique=True, max_length=255)  # 车牌号
    remarks: str | None = Field(default=None)
    is_deleted: int = Field(default=0, index=True)
    created_at: str = Field(max_length=255)