MMXXIV PROGRESS

1 files changed, 21 insertions, 2 deletions

diff --git a/dashboard_website/db.py b/dashboard_website/db.py
index 6991121..7bb5a18 100644
--- a/dashboard_website/db.py
+++ b/dashboard_website/db.py
@@ -141,9 +141,14 @@ def getCluesJSON(timestamp):
 
 
 def addClue(clue_name, clue_info, longitude, latitude, status="UNVISITED"):
+    global clues_last_changed
+    for clue in clues:
+        if clue.name == clue_name:
+            return -1 # clue already exists
     newClue = Clue(latitude, longitude, clue_name, clue_info, status)
     clues.append(newClue)
     clues_last_changed = time.time()
+    return 0
 
 
 def deleteClue(clue_name):
@@ -155,11 +160,15 @@ def deleteClue(clue_name):
             break
 
 
-def visitClue(clue_name):
+def visitClue(clue_name, unvisit=False):
     global clues_last_changed
     for clue in clues:
         if clue.name == clue_name:
             if clue.status == "VISITED":
+                if unvisit:
+                    clue.unvisit()
+                    clues_last_changed = time.time()
+                    return 0
                 return 3  # already visited
             clue.visit()
             clues_last_changed = time.time()
@@ -178,6 +187,16 @@ def toggleEnableClue(clue_name):
             return 0  # OK
     return 2  # no clue     
 
+def toggleClueRequired(clue_name):
+    global clues_last_changed
+    for clue in clues:
+        if clue.name == clue_name:
+            clue.toggle_required()
+            clues_last_changed = time.time()
+            updateRoutes()
+            return 0  # OK
+    return 2  # no clue   
+
 
 def visitClueTeam(team_name, clue_name):
     global clues_last_changed
@@ -287,4 +306,4 @@ def save():
             csvwriter.writerow([clue.name, clue.latitude, clue.longitude, clue.info, clue.status])
 
 
-load()
\ No newline at end of file
+#load("all_clues.csv")
\ No newline at end of file