drink math + functions to manage mongodb

server.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 
 from datetime import datetime
+from time import sleep
 
 import pymongo
 
@@ -10,24 +11,48 @@ def checkpoint(message):
     """
     print("[Checkpoint] {}".format(message))
 
-def total_drinks(id):
-    """Looks up the number of drinks a given user has consumed
+def get_info(id):
+    """Looks up all data for an id
     """
-    # Lookup
-    return 10
+    checkpoint("Looking up data for id \'{}\'".format(id))
 
-def remove_drinks(id=None, finished_drinks):
-    """Remove finished drinks. Removes all drinks from all ids
-    if id==None
+    # Connect to mongodb & return user
+    collection = pymongo.MongoClient().group23.bac_monitoring
+    return collection.find({'id': 123456})[0]
+
+def add_drink(id):
+    """Adds 1 drink to the user
     """
+    checkpoint("Adding 1 drink for id \'{}\'".format(id))
+
+    ## Connect to mongodb & add drink
+    collection = pymongo.MongoClient().group23.bac_monitoring
+    
+    curr_time = int(datetime.today().timestamp())
+    collection.update({'id': id}, {'$push': {'drinks': curr_time}})
+
+def remove_drinks(id=None, finished_drinks=None):
+    """Remove finished drinks. Removes all drinks from all ids
+    if id==None, all drinks from specific if if finished_drinks==None, otherwise
+    removes drinks in finished_drinks
+    """
+    # Connecto to mongodb
+    collection = pymongo.MongoClient().group23.bac_monitoring
 
     # Remove all drinks from all users
     if id == None:
-        # Iterate through all users
-        # Remove set drink list to empty
+        checkpoint('Removing all drinks for all users')
+        collection.update({}, {'$set': {'drinks': []}}, multi=True)
+    # Remove all drinks for specific user
+    elif finished_drinks == None:
+        checkpoint("Removing all drinks for user \'{}\'".format(id))
+        collection.update({'id': id}, {'$set': {'drinks': []}})
+    # Remove only specified drinks
     else:
-        # Lookup specific user
-        # Remove drinks < start_time
+        checkpoint("Removing drinks: \'{}\' for user \'{}\'"
+            .format(finished_drinks, id))
+        for drink in finished_drinks:
+            collection.update({'id': id}, {'$pull': {'drinks': drink}})     
 
 def allowed_drinks(id):
     """Gets the amount of drinks someone is allowed to have, taking into account
@@ -40,7 +65,9 @@ def allowed_drinks(id):
                         '180': .035, '200': .03, '220': .033, '240': .025},
                   'F': {'100': .07, '120': .06, '140': .05,  '160': .04,
                         '180': .04,  '200': .035, '220': .03, '240': .03}}
-    one_drink = bac_lookup[user_info['gender']][user_info['weight']]
+    # Lookup gender & rounded weight
+    weight = int(20 * round(float(user_info['weight']) / 20))
+    one_drink = bac_lookup[user_info['gender']][str(weight)]
 
     # Add up BAC for each drink
     bac = 0
@@ -48,43 +75,44 @@ def allowed_drinks(id):
     # According the the BAC chart, for ever 40 minutes .01% is subtracted from BAC
     #  and there are 2400 seconds on 40 minutes
     curr_time = int(datetime.today().timestamp()) / 2400
-    for drink in drinks
+    for drink in user_info['drinks']:
         # Calculate drinking time and bac
         drink_time = drink / 2400
         drinking_time = curr_time - drink_time
         # Calculate BAC from specific drink, or remove it 
-        drink_bac = one_drink - (drinking_time * 01)
+        drink_bac = one_drink - (drinking_time * .01)
         if drink_bac == 0:
             finished_drinks.append[drink]
         else:
             bac += drink_bac
 
+    # Debugging
+    checkpoint("BAC for \'{}\': {}".format(id, bac))
+
     # Remove all drinks not contributing to current BAC
     remove_drinks(id, finished_drinks)
 
     # Intoxicated, no drinks consumed, or some drinks consumed
-    if bac >= .06:
-        return 0
-    elif bac == 0:
-        return int(.06 / one_drink)
-    else:
-        return int(.06 / bac)
+    drinks_left = 0
+    if bac == 0:
+        drinks_left = int(.06 / one_drink)
+    elif bac < .06:
+        drinks_left = int(.06 / bac)
+    checkpoint("Allowed drinks for id \'{}\': {}".format(id, drinks_left))
+    return drinks_left
 
 def main():
 
     # Arguments
+    id = 123456
+    #remove_drinks(id, [1523206500])
+    remove_drinks(id)
+    add_drink(id)
+    sleep(2)
+    add_drink(id)
+    sleep(3)
+    add_drink(id)
 
-    # Connect to to collection
-    # Assumes 'bac_monitoring' collection already exists
-    collection = pymongo.MongoClient().group23.bac_monitoring
-
-    ## Add new drink
-    # Current time in hours
-    curr_time = int(datetime.today().timestamp())
-    collection.update({'id': 123456}, {'$push': {'drinks': curr_time}})
-    
-    # Insert something
-    drink_data = collection.find({'id': 123456})[0]
-    print(drink_data)
+    allowed_drinks(id)
 
 main()