from dataclasses import dataclass
from enum import Enum
from dataclasses_json import dataclass_json


class PublishType(Enum):
    SUCCESS = "SUCCESS"
    ERROR = "ERROR"


@dataclass_json
@dataclass
class ReserveResultPojo:
    type: PublishType
    phone: str
    message: str
    url: str
    id = None

    def to_firestore_dict(self):
        dest = {
            u'type': self.type.value,
            u'id': self.id,
            u'message': self.message,
            u'phone': self.phone,
            u'url': self.url
        }

        return dest