"""AccountContractVatDetail model."""

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


class AccountContractVatDetail(BaseModel):
    """AccountContractVatDetail model."""

    __tablename__ = 'account_contract_vat_detail'
    account_contract_vat_detail_id = db.Column(db.Integer, primary_key=True)
    account_id = db.Column(
        db.Integer,
        db.ForeignKey('account.account_id'),
        nullable=False
    )
    contract_id = db.Column(db.Integer, nullable=False)
    company_code = db.Column(db.String(15), nullable=False)
    vat_country = db.Column(db.String(15), nullable=False)
    is_vat_registered = db.Column(db.Boolean, default=False, nullable=False)
    is_vat_eligible = db.Column(db.Boolean, default=False, nullable=False)

    @classmethod
    def get_by_account_id_contract_id(cls, account_id, contract_id):
        """Get all account vat details by account_id and contract_id."""
        return cls.query.filter(
            (cls.account_id == account_id) & (cls.contract_id == contract_id)
        ).first()