def get_rigt_task_for_task_screen(taskId,buildId): if buildId == 0 and taskId != 0: task = Task.getTask(taskId) print('editing new parent task') if buildId == 1: task = Task() print('') if buildId == 2 and taskId != 0: task = Task() task.task_id = taskId return task