1 files changed, 18 insertions, 9 deletions
diff --git a/dashboard_website/datastructs.py b/dashboard_website/datastructs.py
index 4648fee..f150825 100644
--- a/dashboard_website/datastructs.py
+++ b/dashboard_website/datastructs.py
@@ -48,24 +48,33 @@ class Point:
 
 
 class Clue(Point):
-    def __init__(self, lat, long, name, info, status):
+    def __init__(self, lat, long, name, info, status, pool_index = 0, required=False):
         self.longitude = long
         self.latitude = lat
         self.name = name
         self.info = info
-        self.status = status  # UNVISITED | ASSIGNED | VISITED
+        self.status = status # UNVISITED | VISITED | DISABLED | ASSIGNED
+        self.pool_inbex = pool_index
+        self.required = required
+    
 
     def visit(self):
         self.status = "VISITED"
+    
+    def toggle_enable(self):
+        if self.status == "UNVISITED" or self.status == "ASSIGNED":
+            self.status = "DISABLED"
+        elif self.status == "DISABLED":
+            self.status = "UNVISITED"
 
     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,
-        }
+        json_dict = {'longitude' : self.longitude,
+                     'latitude' : self.latitude,
+                     'clue_name' : self.name.replace('"', "'"),
+                     'clue_info' : self.info.replace('"', "'"),
+                     'clue_status' : self.status,
+                     'clue_required' : self.required}
+
         return json_dict