diff --git a/atlas/api.py b/atlas/api.py index dedaf4b8..4ba4fcfd 100644 --- a/atlas/api.py +++ b/atlas/api.py @@ -17,7 +17,7 @@ from ninja.security import django_auth from ninja import NinjaAPI, File from ninja.files import UploadedFile -from generic.models import Modality +from generic.models import Examination, Modality from .models import Case, DuplicateDicom, Series, SeriesImage, UncategorisedDicom @@ -118,7 +118,6 @@ def import_dicoms_helper(request, case_id: int | None = None): data[tags["SeriesInstanceUID"]].append((d, tags)) - logger.debug("IMPORT2") series_list = [] for series_uid in data: modality = get_object_or_404(Modality, short_code=tags["Modality"]) @@ -133,6 +132,12 @@ def import_dicoms_helper(request, case_id: int | None = None): description=tags["SeriesDescription"], series_instance_uid=tags["SeriesInstanceUID"], ) + + if tags["StudyDescription"]: + examination = Examination.objects.get_or_create(examination=tags["StudyDescription"], modality=modality) + series.examination = examination + + series.save() # We might only want to add the author during creation....