File size: 961 Bytes
bf41ce7
 
 
 
 
 
 
 
 
1fac9b0
4cda30b
bf41ce7
 
 
 
 
 
 
 
4cda30b
 
 
bf41ce7
4cda30b
bf41ce7
4cda30b
bf41ce7
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
"""contact model:
{
    "contactId": "String",
    "displayName": "String",
    "phoneNumbers": ["String"],
    "status": "created | updated | deleted",
}"""
from typing import Any

from Brain.src.model.requests.request_model import TrainContacts


class ContactModel:
    def __init__(self):
        self.contact_id = ""
        self.display_name = ""
        self.phone_numbers = []
        self.status = ContactStatus.UPDATED

    def get_contact_model(self, data: TrainContacts.ContactReq) -> None:
        self.contact_id = data.contactId
        self.display_name = data.displayName
        self.phone_numbers = []
        for phone in data.phoneNumbers:
            self.phone_numbers.append(phone)
        self.status = data.status

    def get_str_phones(self):
        return "".join(self.phone_numbers)


"""contact status: created | updated | deleted"""


class ContactStatus:
    CREATED = "created"
    UPDATED = "updated"
    DELETED = "deleted"