import enum

from sqlalchemy import Column, Integer, Enum, VARCHAR
from sqlalchemy.orm import declarative_base

Base = declarative_base()


class SendEmailEnum(enum.Enum):
    yes = "yes"
    no = "no"


class ContactTypeEnum(enum.Enum):
    marketing = 'marketing'
    operations = 'operations'
    delivery_notification = 'delivery_notification'
    correction = 'correction'
    deletion = 'deletion'
    correction_ringtone = 'correction_ringtone'
    deletion_ringtone = 'deletion_ringtone'
    delivery_verification = 'delivery_verification'


class CustomerMasterMasterContact(Base):
    __tablename__ = 'customer_master_master_contact'
    customer_master_master_contact_id = Column(
        Integer, primary_key=True, unique=True, autoincrement=True)
    customer_master_master_id = Column(Integer, unique=True)
    sendemail = Column(Enum(SendEmailEnum))
    contact_type = Column(Enum(ContactTypeEnum))
    contact_email = Column(VARCHAR)
    contact_first_name = Column(VARCHAR)
    contact_last_name = Column(VARCHAR)

    def __init__(self, dms_id, sendemail, contact_type):
        self.customer_master_master_id = dms_id
        self.sendemail = SendEmailEnum(sendemail)
        self.contact_type = ContactTypeEnum(contact_type)
        self.contact_email = 'automationqa@theorchard.com'
        self.contact_first_name = 'Vector'
        self.contact_last_name = 'Testing'