slight improvements

1 files changed, 12 insertions, 3 deletions

diff --git a/dashboard_website/datastructs.py b/dashboard_website/datastructs.py
index 0171d8f..997d3e7 100644
--- a/dashboard_website/datastructs.py
+++ b/dashboard_website/datastructs.py
@@ -43,22 +43,31 @@ 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}
+                     'clue_status' : self.status,
+                     'clue_required' : self.required}
         return json_dict
 
 class Bike(Point):