"""contract_exclusion Model."""

from abacus_common_logic.connectors.database import db
from abacus_common_logic.models.base import CRUDMixin


class ContractExclusion(db.Model, CRUDMixin):
    """contract_exclusion Model."""

    __tablename__ = 'contract_exclusion'

    contract_exclusion_id = db.Column(db.Integer, primary_key=True)
    contract_id = db.Column(
        db.Integer, db.ForeignKey('contract.contract_id'), nullable=False
    )
    exclusions = db.Column(db.JSON, nullable=False, default={})

    @classmethod
    def get_by_contract_ids(cls, contract_ids):
        """Exclusions for a batch of contract ids (one per contract at most)."""
        return cls.query.filter(cls.contract_id.in_(contract_ids)).all()