1 files changed, 9 insertions, 3 deletions

diff --git a/dashboard_website/datastructs.py b/dashboard_website/datastructs.py
index f150825..1b4b92b 100644
--- a/dashboard_website/datastructs.py
+++ b/dashboard_website/datastructs.py
@@ -48,31 +48,37 @@ class Point:
 
 
 class Clue(Point):
-    def __init__(self, lat, long, name, info, status, pool_index = 0, required=False):
+    def __init__(self, lat, long, name, info, status, assigned_team = 0, required=False):
         self.longitude = long
         self.latitude = lat
         self.name = name
         self.info = info
         self.status = status # UNVISITED | VISITED | DISABLED | ASSIGNED
-        self.pool_inbex = pool_index
+        self.assigned_team = assigned_team # 0 = to be assigned algorithmically
         self.required = required
-    
 
     def visit(self):
         self.status = "VISITED"
     
+    def unvisit(self):
+        self.status = "UNVISITED"
+    
     def toggle_enable(self):
         if self.status == "UNVISITED" or self.status == "ASSIGNED":
             self.status = "DISABLED"
         elif self.status == "DISABLED":
             self.status = "UNVISITED"
 
+    def toggle_required(self):
+        self.required = False if self.required else True
+
     def toJSON(self):
         json_dict = {'longitude' : self.longitude,
                      'latitude' : self.latitude,
                      'clue_name' : self.name.replace('"', "'"),
                      'clue_info' : self.info.replace('"', "'"),
                      'clue_status' : self.status,
+                     'assigned_team' : self.assigned_team,
                      'clue_required' : self.required}
 
         return json_dict