from db import db
from enum import Enum


class PRSMappingEntityType(Enum):
    BUDGET = 0
    CAMPAIGN = 1
    MOMENT = 2


class RawPRSPurchaseOrdersMap(db.Model):
    """RawPRSPurchaseOrdersMap model"""

    __tablename__ = "RawPRSPurchaseOrdersMap"

    label_id = db.Column(db.Integer, db.ForeignKey("Label.id", ondelete="CASCADE"), primary_key=True)
    budget_group_id = db.Column(db.Integer, db.ForeignKey("PRSBudgetGroup.id", ondelete="CASCADE"), primary_key=True)
    budget_category_id = db.Column(
        db.Integer, db.ForeignKey("PRSBudgetCategory.id", ondelete="CASCADE"), primary_key=True
    )

    mapping_entity_type = db.Column(db.Integer, nullable=False)

    campaign_taxonomy = db.Column(db.JSON, nullable=True)

    moment_group_id = db.Column(
        db.Integer, db.ForeignKey("ArtistMomentTypeGroup.id", ondelete="CASCADE"), nullable=True
    )
    moment_type_id = db.Column(db.Integer, db.ForeignKey("ArtistMomentType.id", ondelete="CASCADE"), nullable=True)