71 lines
2.0 KiB
Python
71 lines
2.0 KiB
Python
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
|
|
firstName: None
|
|
lastName: None
|
|
email: None
|
|
id = None
|
|
accepted = None
|
|
passport: str = ""
|
|
slot_position = None
|
|
sim_position = None
|
|
|
|
@staticmethod
|
|
def from_firestore_dict(source):
|
|
publish_type = source['type']
|
|
phone = source['phone']
|
|
message = source['message']
|
|
url = source['url']
|
|
id = source['id']
|
|
email = source['email']
|
|
lastName = source['lastName']
|
|
firstName = source['firstName']
|
|
result = ReserveResultPojo(type=publish_type, phone=phone,
|
|
message=message, url=url, email=email,
|
|
firstName=firstName, lastName=lastName)
|
|
if 'accepted' in source:
|
|
accepted = source['accepted']
|
|
result.accepted = accepted
|
|
if 'sim_position' in source:
|
|
sim_position = source['sim_position']
|
|
result.sim_position = sim_position
|
|
if 'slot_position' in source:
|
|
slot_position = source['slot_position']
|
|
result.slot_position = slot_position
|
|
if 'passport' in source:
|
|
passport = source['passport']
|
|
result.passport = passport
|
|
result.id = id
|
|
return result
|
|
|
|
def to_firestore_dict(self):
|
|
dest = {
|
|
u'type': self.type.value,
|
|
u'id': self.id,
|
|
u'message': self.message,
|
|
u'phone': self.phone,
|
|
u'firstName': self.firstName,
|
|
u'lastName': self.lastName,
|
|
u'email': self.email,
|
|
u'passport': self.passport,
|
|
u'url': self.url,
|
|
u'sim_position': self.sim_position,
|
|
u'slot_position': self.slot_position
|
|
}
|
|
|
|
return dest
|