From 9690fc0e4a319941f3103b1c43d8509c22c20a4f Mon Sep 17 00:00:00 2001 From: Anson Bridges Date: Tue, 2 Jul 2024 16:47:39 +0200 Subject: slight improvements --- dashboard_website/datastructs.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) (limited to 'dashboard_website/datastructs.py') 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): -- cgit v1.2.3