slight improvements

1 files changed, 15 insertions, 4 deletions

diff --git a/dashboard_website/db.py b/dashboard_website/db.py
index 7e81680..6991121 100644
--- a/dashboard_website/db.py
+++ b/dashboard_website/db.py
@@ -29,7 +29,7 @@ startup = False  #
 # a bike is added/removed
 # determines/assigns clusters, and assigns routes to bikes
 def updateRoutes_background():  # run in thread due to long runtime
-    global currently_updating, routes_last_changed, routes
+    global currently_updating, routes_last_changed, routes, clusters
     print("Calculating clusters...")
     routes = {"clusters": [], "cluster_times": {}, "individual_routes": []}  # reset
     clusters, paths, times = router.getClusters(bikes, clues, homeBase)
@@ -140,8 +140,8 @@ def getCluesJSON(timestamp):
     return False
 
 
-def addClue(clue_name, clue_info, longitude, latitude, visited="UNVISITED"):
-    newClue = Clue(latitude, longitude, clue_name, clue_info, visited)
+def addClue(clue_name, clue_info, longitude, latitude, status="UNVISITED"):
+    newClue = Clue(latitude, longitude, clue_name, clue_info, status)
     clues.append(newClue)
     clues_last_changed = time.time()
 
@@ -165,7 +165,18 @@ def visitClue(clue_name):
             clues_last_changed = time.time()
             updateRoutes()
             return 0  # OK
-    return 2  # no clue found
+    return 2  # no clue 
+
+
+def toggleEnableClue(clue_name):
+    global clues_last_changed
+    for clue in clues:
+        if clue.name == clue_name:
+            clue.toggle_enable()
+            clues_last_changed = time.time()
+            updateRoutes()
+            return 0  # OK
+    return 2  # no clue     
 
 
 def visitClueTeam(team_name, clue_name):