From 2bce2956c62ab19692b4f10ec4c6a677432d76e0 Mon Sep 17 00:00:00 2001
From: Ross <xkjq@ymail.com>
Date: Thu, 7 Apr 2022 18:31:34 +0100
Subject: [PATCH] .

---
 atlas/templates/atlas/series_viewer.html           |  2 +-
 rapids/models.py                                   | 14 ++++++++++++++
 .../templates/rapids/question_display_block.html   |  2 ++
 rapids/urls.py                                     |  5 +++++
 rapids/views.py                                    |  7 +++++++
 5 files changed, 29 insertions(+), 1 deletion(-)

diff --git a/atlas/templates/atlas/series_viewer.html b/atlas/templates/atlas/series_viewer.html
index 71535901..a6146afe 100755
--- a/atlas/templates/atlas/series_viewer.html
+++ b/atlas/templates/atlas/series_viewer.html
@@ -58,7 +58,7 @@
     <summary>Image info</summary>
     <div>
         <a href="{% url 'atlas:series_anonymise_dicom' pk=series.pk %}"
-            title="orders dicom by uploaded filename">Anonymise dicoms</a><br />
+            title="Anonymise dicom images">Anonymise dicoms</a><br />
         <a href="{% url 'atlas:series_order_dicom' pk=series.pk %}" title="orders dicom by slice location">Order dicoms
             by slice location</a>
         <a href="{% url 'atlas:series_order_dicom_instance' pk=series.pk %}"
diff --git a/rapids/models.py b/rapids/models.py
index 3f26658f..e46ead30 100644
--- a/rapids/models.py
+++ b/rapids/models.py
@@ -1,3 +1,5 @@
+import os
+import dicognito
 from rad.settings import REMOTE_URL
 from django.db import models
 from django.http.response import JsonResponse
@@ -474,6 +476,18 @@ class Rapid(models.Model):
 
         return json
 
+    def anonymise_images(self):
+        anonymizer = dicognito.anonymizer.Anonymizer()
+
+        for image in self.images.all():
+            file_path = os.path.join(settings.MEDIA_ROOT, image.image.name)
+
+            try:
+                with pydicom.dcmread(file_path) as dataset:
+                    anonymizer.anonymize(dataset)
+                    dataset.save_as(file_path)
+            except pydicom.errors.InvalidDicomError:
+                pass
 
 # reversion.register(Rapid, follow=["images"])
 # @reversion.register
diff --git a/rapids/templates/rapids/question_display_block.html b/rapids/templates/rapids/question_display_block.html
index 644760df..ecbe5626 100755
--- a/rapids/templates/rapids/question_display_block.html
+++ b/rapids/templates/rapids/question_display_block.html
@@ -95,6 +95,8 @@
     </span>
     {% endfor %}
 </details>
+        <a href="{% url 'rapid:question_anonymise_dicom' pk=question.pk %}"
+            title="Anonymise dicom images">Anonymise dicoms</a><br />
 <script>
     $(document).ready(function () {
         $("#edit-button").click(() => {
diff --git a/rapids/urls.py b/rapids/urls.py
index b480777e..51ba66cb 100755
--- a/rapids/urls.py
+++ b/rapids/urls.py
@@ -45,6 +45,11 @@ urlpatterns = [
         views.GenericViews.question_user_answers,
         name="question_user_answers",
     ),
+    path(
+        "question/<int:pk>/anon",
+        views.question_anonymise_dicom,
+        name="question_anonymise_dicom",
+    ),
     path(
         "exam/<int:pk>/review",
         views.GenericViews.question_review,
diff --git a/rapids/views.py b/rapids/views.py
index bd5e6a08..ee47faad 100755
--- a/rapids/views.py
+++ b/rapids/views.py
@@ -984,6 +984,13 @@ class RapidLateralityViewSet(
 
 GenericViews = GenericViewBase("rapids", Rapid, CidUserAnswer, Exam)
 
+@user_is_author_or_rapid_checker
+def question_anonymise_dicom(request, pk):
+    question = get_object_or_404(Rapid, pk=pk)
+
+    question.anonymise_images()
+
+    return redirect("atlas:series_detail", pk=pk)
 
 @user_is_author_or_rapid_checker
 def question_save_annotation(request, pk):