def get_linked_task(self, *, validated_data):
if "algorithm" in validated_data:
return create_algorithm_jobs_for_session.signature(
kwargs={
"algorithm_image_pk":
validated_data["algorithm"].latest_ready_image.pk
},
immutable=True,
)
elif "archive" in validated_data:
return add_images_to_archive.signature(
kwargs={"archive_pk": validated_data["archive"].pk},
immutable=True,
)
elif "reader_study" in validated_data:
return add_images_to_reader_study.signature(
kwargs={"reader_study_pk": validated_data["reader_study"].pk},
immutable=True,
)
elif "answer" in validated_data:
return add_image_to_answer.signature(
kwargs={"answer_pk": validated_data["answer"].pk},
immutable=True,
)
else:
raise RuntimeError(
"Algorithm image, archive or reader study must be set")
def get_form_kwargs(self):
kwargs = super().get_form_kwargs()
kwargs.update({
"linked_task":
create_algorithm_jobs_for_session.signature(
kwargs={
"algorithm_image_pk": self.algorithm.latest_ready_image.pk
},
immutable=True,
)
})
return kwargs