"""Campaign Provider model."""
from enum import Enum
from db import db


class Providers(Enum):
    SonyMusic = 1
    GuptaMedia = 2
    SevenStars = 3
    Wavo = 4


class CampaignProvider(db.Model):

    __tablename__ = "CampaignProvider"

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String, nullable=False)
    is_prs_vendor = db.Column(db.Boolean, nullable=False, server_default="FALSE")