medipal/enter.py at master · jlimcode/medipal · GitHub

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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
from Medication import Entry, Medication
from send_text import send_text
import os
from pymongo import MongoClient
from termcolor import cprint

mongo_key = os.getenv('MONGO_LOGIN')


def create_message(med: Medication) -> str:
    message = "It’s your Medipal here to remind you to take "
    if med.anonymous:
        message += "your medication"
    else:
        message += med.name
    if med.withFood:
        message += " with food!\n"
    else:
        message += " without food!\n"
    if med.restrictions != "" and med.restrictions != None:
        message += "Keep in mind these restrictions: " + med.restrictions + "!\n"
    message += med.message
    return message


def enter(entry: Entry) -> None:
    cluster = MongoClient(mongo_key, tlsAllowInvalidCertificates=True)
    db = cluster["Medipal"]
    Users = db["Users"]
    Meds = db["Meds"]

    num = entry.number

    u = Users.find_one({"number": num})

    if(u == None):  # user is not already in the system, adding them
        cprint(f'Creating user corresponding to f{num}', 'yellow')
        medHashes = []
        newUser = {"number": num, "meds": []}
        Users.insert_one(newUser)
        u = Users.find_one({"number": num})
        uid = u["_id"]

        for m in entry.meds:
            m.addMessage(create_message(m))
            m.addUID(uid)
            Meds.insert_one(m.getDBFormat())
            cprint(f'Medication entered to database:', 'green')
            print(m)

        userMeds = Meds.find({"user": uid})
        for m in userMeds:
            medHashes.append(m["_id"])
        Users.update_one({"_id": uid}, {"$set": {"meds": medHashes}})

    else:  # updating existing user
        uid = u["_id"]
        cprint(f'Updating user with id: f{uid}', 'yellow')
        for m in entry.meds:
            med = Meds.find_one({"user": uid, "name": m.name})
            if med == None:
                m.addMessage(create_message(m))
                m.addUID(uid)
                med = m.getDBFormat()
                Meds.insert_one(med)
                cprint(f'Medication added to database:', 'green')
                print(m)
            else:
                Meds.update_one({"_id": med["_id"]}, {
                                "$set": {"times": m.times}})
                cprint(f'Medication updated on database:', 'yellow')
                print(m)

        userMeds = Meds.find({"user": uid})
        medHashes = []
        for m in userMeds:
            medHashes.append(m["_id"])
        Users.update_one({"_id": uid}, {"$set": {"meds": medHashes}})


if __name__ == "__main__":

    medsDict1 = {"name": "testMed", "times": ["04:13", "12:34"],
                 "chronic": False, "withFood": False, "doses": 14, "restrictions": "no calcium",
                 "message": "hi", "anonymous": True}
    medsDict2 = {"name": "testMed7", "times": ["04:14", "2:38"], "chronic": True,
                 "withFood": True, "doses": -1, "restrictions": "calcium",
                 "message": "hello", "anonymous": False}
    meds = [medsDict1] + [medsDict2]
    testDict = {"number": "+1234567890", "meds": meds}
    enter(Entry(**testDict))
    print("done")