def add_task(self, task):
bucket_tag = common_utils.determine_bucket(task)
if bucket_tag == "ACTIVE":
self.active_tasks.add_task(task)
elif bucket_tag == "COMPLETED":
self.completed_tasks.add_task(task)
else:
self.pending_tasks.add_task(task)
def edit_task(self, task):
bucket_tag = common_utils.determine_bucket(task)
if bucket_tag == "ACTIVE":
bucket = self.active_tasks
elif bucket_tag == "COMPLETED":
bucket = self.completed_tasks
else:
bucket = self.pending_tasks
replaced_task = bucket.get_by_id(task.id)
bucket.remove_task(replaced_task)
bucket.add_task(task)
return task