Source code for q2_cores.OSI.models.email_model

from dataclasses import dataclass
from enum import Enum




class EmailType(Enum):
    Primary = "BEML"
    Secondary = "EML"
    Work = "WORK"
    Home = "HEML"
    Unknown = None





@dataclass
class EmailInfo:
    email_address: str
    raw_email_type: str
    email_description: str
    email_type: EmailType = None
    unknown_email_type: bool = False

    def __post_init__(self):
        if self.email_type is None:
            try:
                self.email_type = EmailType(self.raw_email_type)
            except ValueError:
                self.unknown_email_type = True
                self.email_type = EmailType.Unknown