def __init__(self): # TODO(allench): these are temporary, they obviously need to be # persisted somewhere, and loaded in every time we start up the # todo service. self.snapshot_version = 0 self.id_counter = 0 self.task_store = ServerTaskStore()
class TodoService: def __init__(self): # TODO(allench): these are temporary, they obviously need to be # persisted somewhere, and loaded in every time we start up the # todo service. self.snapshot_version = 0 self.id_counter = 0 self.task_store = ServerTaskStore() def pre_cleanup(self): # do stuff like delay_priority_upgrade checks pass def get_unique_id(self): self.id_counter += 1 return self.id_counter def increment_snapshot_version(self): self.snapshot_version += 1 return self.snapshot_version def get_active_snapshot(self): return self.task_store.active_tasks def get_pending_snapshot(self): return self.task_store.pending_tasks def get_completed_snapshot(self): return self.task_store.completed_tasks def get_all_snapshots(self): return { "pending_tasks": self.task_store.pending_tasks, "active_tasks": self.task_store.active_tasks, "completed_tasks": self.task_store.completed_tasks, } def get_all_versions(self): return { "pending_tasks": self.task_store.pending_tasks.version, "active_tasks": self.task_store.active_tasks.version, "completed_tasks": self.task_store.completed_tasks.version, } def add_task(self, input_task): task = Task(snapshot=input_task) task.id = self.get_unique_id() self.task_store.add_task(task) print "Added task: " + str(task) return task def complete_task(self, input_task): task = Task(snapshot=input_task) self.task_store.complete_task(task) return task def activate_task(self, input_task): task = Task(snapshot=input_task) self.task_store.activate_task(task) return task