def task_add(request):
'''add task to project'''
pid = int(request.POST.get('pid',0))
task = Task()
task.task_name = request.POST.get('task_name')
task.task_project = get_object_or_404(Project,id__exact = pid)
priority = request.POST.get('priority')
if priority:
task.task_priority = priority
else:
task.task_priority = TASK_PRIORITY['medium']
task.task_completed = 0
task.save()
#add project task count
task.task_project.project_tasks += 1
task.task_project.save()
#return HttpResponseRedirect('/todo/')
return HttpResponse('success')
def put(self, list_id):
data = request.get_json()
user = User.find_by_id(get_jwt_identity())
todoList = TodoList.query.filter(and_(TodoList.id == list_id, TodoList.user_id == user.userID)).first()
if todoList is None:
task_lst = []
for task in data['tasks']:
obj = Task(
task_name=task['task_name'],
task_done=task['task_done']
)
task_lst.append(obj)
todoList = TodoList(
todoList_name=data['todoList_name'],
todoList_done=data['todoList_done'],
user_id=user.userID
)
todoList.tasks = task_lst
todoList.save()
return {'message': 'Successfully saved new data'}, 201
else:
todoList.todoList_name = data['todoList_name']
todoList.todoList_done = data['todoList_done']
for t in data['tasks']:
if t['id'] is "":
task = Task(
task_name=t['task_name'],
task_done=t['task_done']
)
todoList.tasks.append(task)
else:
task = Task.get_by_id(t['id'])
task.task_name = t['task_name']
task.task_done = t['task_done']
todoList.save()
return {'message': 'Updated data'}
