"""DistributionType model."""

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


class DistributionType(db.Model, CRUDMixin):
    """DistributionType model.

    This model maps `distribution_type` table from art_relations.
    """

    __tablename__ = 'distribution_type'
    distribution_type_id = db.Column('id', db.Integer, primary_key=True)
    distribution_type_name = db.Column('name', db.String(20), nullable=True)

    vendor_contract_distribution_type = db.relationship(
        'VendorContractDistributionType',
        backref='distribution_type',
        cascade='all, delete-orphan',
    )

    vendor_dms_master_restriction = db.relationship(
        'VendorDmsMasterRestriction',
        backref='distribution_type',
        cascade='all, delete-orphan',
    )