from sqlalchemy import Boolean, ForeignKey, String, UniqueConstraint
from sqlalchemy.orm import Mapped, mapped_column, relationship

from app.db.base import Base, TimestampMixin


class PolicyScanKeyword(Base, TimestampMixin):
    __tablename__ = "policy_scan_keywords"
    __table_args__ = (UniqueConstraint("insurance_company_id", "keyword", name="uq_scan_keyword_company"),)

    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
    insurance_company_id: Mapped[int] = mapped_column(
        ForeignKey("insurance_companies.id", ondelete="CASCADE"),
        nullable=False,
        index=True,
    )
    keyword: Mapped[str] = mapped_column(String(255), nullable=False)
    is_primary: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False)

    insurance_company: Mapped["InsuranceCompany"] = relationship(back_populates="scan_keywords")


from app.models.insurance_company import InsuranceCompany  # noqa: E402