from decimal import Decimal

from sqlalchemy import Integer, Numeric, String
from sqlalchemy.orm import Mapped, mapped_column

from app.core.database import Base
from app.modules.business.constants import (
    RiskLevel,
    StatusValue,
)
from app.modules.business.models.common import TimestampMixin


class Supplier(Base, TimestampMixin):
    __tablename__ = "suppliers"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
    name: Mapped[str] = mapped_column(String(255), index=True)
    category: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
    contact: Mapped[str | None] = mapped_column(String(128), nullable=True)
    quality_score: Mapped[Decimal] = mapped_column(Numeric(5, 2), default=0)
    delivery_score: Mapped[Decimal] = mapped_column(Numeric(5, 2), default=0)
    price_score: Mapped[Decimal] = mapped_column(Numeric(5, 2), default=0)
    risk_level: Mapped[str] = mapped_column(String(32), default=RiskLevel.LOW, index=True)
    blacklist_status: Mapped[str] = mapped_column(
        String(32),
        default=StatusValue.NORMAL,
        index=True,
    )