"""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={}
    )