def reassign_assignment(worker_id, assignment_id):
"""
Return a given assignment after reassigning it to the specified
worker.
Args:
worker_id (int):
The ID of the worker to be assigned.
assignment_id (int):
The ID of the assignment to be assigned.
Returns:
assignment (orchestra.models.TaskAssignment):
The newly assigned assignment.
Raises:
orchestra.core.errors.TaskAssignmentError:
The specified worker is already assigned to the given task.
orchestra.core.errors.WorkerCertificationError:
The specified worker is not certified for the given assignment.
"""
worker = Worker.objects.get(id=worker_id)
assignment = TaskAssignment.objects.get(id=assignment_id)
if assignment.assignment_counter > 0:
role = WorkerCertification.Role.REVIEWER
else:
role = WorkerCertification.Role.ENTRY_LEVEL
if not _worker_certified_for_task(worker, assignment.task, role):
raise WorkerCertificationError(
'Worker not certified for this assignment.')
if is_worker_assigned_to_task(worker, assignment.task):
raise TaskAssignmentError('Worker already assigned to this task.')
assignment.worker = worker
assignment.save()
add_worker_to_project_team(worker, assignment.task.project)
return assignment
def reassign_assignment(worker_id, assignment_id):
"""
Return a given assignment after reassigning it to the specified
worker.
Args:
worker_id (int):
The ID of the worker to be assigned.
assignment_id (int):
The ID of the assignment to be assigned.
Returns:
assignment (orchestra.models.TaskAssignment):
The newly assigned assignment.
Raises:
orchestra.core.errors.TaskAssignmentError:
The specified worker is already assigned to the given task.
orchestra.core.errors.WorkerCertificationError:
The specified worker is not certified for the given assignment.
"""
worker = Worker.objects.get(id=worker_id)
assignment = TaskAssignment.objects.get(id=assignment_id)
if assignment.assignment_counter > 0:
role = WorkerCertification.Role.REVIEWER
else:
role = WorkerCertification.Role.ENTRY_LEVEL
if not _worker_certified_for_task(worker, assignment.task, role):
raise WorkerCertificationError(
'Worker not certified for this assignment.')
if is_worker_assigned_to_task(worker, assignment.task):
raise TaskAssignmentError('Worker already assigned to this task.')
assignment.worker = worker
assignment.save()
add_worker_to_project_team(worker, assignment.task.project)
return assignment
def assign_task(worker_id, task_id):
"""
Return a given task after assigning or reassigning it to the specified
worker.
Args:
worker_id (int):
The ID of the worker to be assigned.
task_id (int):
The ID of the task to be assigned.
Returns:
task (orchestra.models.Task):
The newly assigned task.
Raises:
orchestra.core.errors.TaskAssignmentError:
The specified worker is already assigned to the given task
or the task status is not compatible with new assignment.
orchestra.core.errors.WorkerCertificationError:
The specified worker is not certified for the given task.
"""
worker = Worker.objects.get(id=worker_id)
task = Task.objects.get(id=task_id)
required_role, requires_reassign = _role_required_to_assign(task)
assignment = current_assignment(task)
if not _worker_certified_for_task(worker, task, required_role):
raise WorkerCertificationError('Worker not certified for this task.')
if is_worker_assigned_to_task(worker, task):
raise TaskAssignmentError('Worker already assigned to this task.')
# If task is currently in progress, reassign it
if requires_reassign:
assignment.worker = worker
assignment.save()
add_worker_to_project_team(worker, task.project)
return task
# Otherwise, create new assignment
assignment_counter = task.assignments.count()
in_progress_task_data = {}
if required_role == WorkerCertification.Role.REVIEWER:
# In-progress task data is the latest
# submission by a previous worker
in_progress_task_data = assignment.in_progress_task_data
previous_status = task.status
if previous_status == Task.Status.AWAITING_PROCESSING:
task.status = Task.Status.PROCESSING
elif previous_status == Task.Status.PENDING_REVIEW:
task.status = Task.Status.REVIEWING
else:
raise TaskAssignmentError('Status incompatible with new assignment')
task.save()
(TaskAssignment.objects
.create(worker=worker,
task=task,
status=TaskAssignment.Status.PROCESSING,
assignment_counter=assignment_counter,
in_progress_task_data=in_progress_task_data,
snapshots=empty_snapshots()))
add_worker_to_project_team(worker, task.project)
notify_status_change(task, previous_status)
return task
def assign_task(worker_id, task_id):
"""
Return a given task after assigning or reassigning it to the specified
worker.
Args:
worker_id (int):
The ID of the worker to be assigned.
task_id (int):
The ID of the task to be assigned.
Returns:
task (orchestra.models.Task):
The newly assigned task.
Raises:
orchestra.core.errors.TaskAssignmentError:
The specified worker is already assigned to the given task
or the task status is not compatible with new assignment.
orchestra.core.errors.WorkerCertificationError:
The specified worker is not certified for the given task.
"""
worker = Worker.objects.get(id=worker_id)
task = Task.objects.get(id=task_id)
roles = {
Task.Status.AWAITING_PROCESSING: WorkerCertification.Role.ENTRY_LEVEL,
Task.Status.PENDING_REVIEW: WorkerCertification.Role.REVIEWER
}
required_role = roles.get(task.status)
if required_role is None:
raise TaskAssignmentError('Status incompatible with new assignment')
assignment = current_assignment(task)
if not _worker_certified_for_task(worker, task, required_role):
raise WorkerCertificationError('Worker not certified for this task.')
if is_worker_assigned_to_task(worker, task):
raise TaskAssignmentError('Worker already assigned to this task.')
assignment_counter = task.assignments.count()
in_progress_task_data = {}
if assignment:
# In-progress task data is the latest
# submission by a previous worker
in_progress_task_data = assignment.in_progress_task_data
previous_status = task.status
if previous_status == Task.Status.AWAITING_PROCESSING:
task.status = Task.Status.PROCESSING
elif previous_status == Task.Status.PENDING_REVIEW:
task.status = Task.Status.REVIEWING
task.save()
(TaskAssignment.objects.create(worker=worker,
task=task,
status=TaskAssignment.Status.PROCESSING,
assignment_counter=assignment_counter,
in_progress_task_data=in_progress_task_data,
snapshots=empty_snapshots()))
add_worker_to_project_team(worker, task.project)
notify_status_change(task, previous_status)
return task
