start integrating cimar

generic/models.py

@@ -7,6 +7,7 @@ from django.contrib.auth.mixins import LoginRequiredMixin
 from django.db import models
 from django.forms import ValidationError
 from django.http import Http404, HttpRequest
+from django.shortcuts import render
 from django.utils import timezone
 from django.core.validators import MaxValueValidator, MinValueValidator
 from smtplib import SMTPException
@@ -29,7 +30,8 @@ from django.db.models.signals import post_save
 from django.dispatch import receiver
 from generic.mixins import AuthorMixin, QuestionMixin
 from helpers.images import get_image_hash, pretty_print_dicom
-from rad.settings import REMOTE_URL
+from helpers.cimar import CimarAPI, STORAGE_API_URL
+from rad.settings import REMOTE_URL, CIMAR_USERNAME, CIMAR_PASSWORD
 from django.utils.html import format_html
 
 from easy_thumbnails.files import get_thumbnailer
@@ -45,6 +47,34 @@ from django.forms.utils import from_current_timezone, to_current_timezone
 
 from loguru import logger
 
+from pygments import highlight
+from pygments.lexers import JsonLexer
+from pygments.formatters import HtmlFormatter
+
+from django.contrib import admin
+from django.utils.safestring import mark_safe
+from django.template.loader import render_to_string
+
+
+def get_pretty_json(data):
+    """Function to display pretty version of our data"""
+    if not data:
+        return "No data"
+
+    # Convert the data to sorted, indented JSON
+    response = json.dumps(data, sort_keys=True, indent=2)
+
+    # Get the Pygments formatter
+    formatter = HtmlFormatter(style='colorful')
+
+    # Highlight the data
+    response = highlight(response, JsonLexer(), formatter)
+
+    # Get the stylesheet
+    style = "<style>" + formatter.get_style_defs() + "</style><br>"
+
+    # Safe the output
+    return mark_safe(style + response)
 
 def findMiddle(input_list):
     """Returns the middle element of a list"""
@@ -1690,6 +1720,8 @@ class UserProfile(models.Model):
 
     username = property(getusername)
 
+    cimar_sid = models.CharField(max_length=100, blank=True)
+
     def get_exams(self):
         available_exams = defaultdict(list)
 
@@ -1826,4 +1858,89 @@ class ExamCollection(models.Model, AuthorMixin):
 #        settings.AUTH_USER_MODEL,
 #        blank=True,
 #        help_text="Author/Manager(s) of the exam collection",
-#    )
+#    )
+
+class CimarCase(models.Model):
+    uuid = models.CharField(max_length=255, unique=True)
+
+    viewer_link = models.URLField(blank=True)
+
+    study_details = models.JSONField(blank=True, default=dict)
+
+    study_schema = models.JSONField(blank=True, default=dict)
+
+    series_data = models.JSONField(blank=True, default=dict)
+
+    valid_uuid = models.BooleanField(default=False)
+
+    def load_series_data(self, api=None):
+        if api is None:
+            api = CimarAPI()
+            api.login(username=CIMAR_USERNAME, password=CIMAR_PASSWORD)
+        
+
+
+
+    def refresh_study(self):
+        api = CimarAPI()
+        api.login(username=CIMAR_USERNAME, password=CIMAR_PASSWORD)
+
+        self.study_details = api.get_study_by_uuid(self.uuid)
+        self.study_schema = api.get_study_schema_by_uuid(self.uuid)
+
+        self.viewer_link = api.get_study_viewer_url_by_uuid(self.uuid)
+
+        series_data = []
+
+        for series in self.study_schema["series"]:
+            series_data.append({
+                "series_id" : series["id"],
+                "series_description" : series["description"],
+                "image_count": len(series["images"]),
+                "thumbnail": self.get_series_image_thumbnail_link(series)
+
+            }
+            )
+
+        self.series_data = {}
+        self.series_data["series"] = series_data
+
+        self.valid_uuid = True
+        self.save()
+
+    def __str__(self):
+        return f"{self.uuid} ({self.valid_uuid})"
+
+    def get_user_viewer_link(self, user):
+        return self.viewer_link +  f"?sid={user.userprofile.cimar_sid}"
+
+
+    def get_pretty_study_details(self):
+        """Function to display pretty version of our data"""
+        return get_pretty_json(self.study_details)
+
+    def get_pretty_study_schema(self):
+        """Function to display pretty version of our data"""
+        return get_pretty_json(self.study_schema)
+
+    def get_pretty_study_series(self):
+        """Function to display pretty version of our data"""
+        return get_pretty_json(self.series_data)
+
+    #def get_series_details_block(self, cimar_sid=None):
+    #    if "series" in self.series_data:
+    #        return render_to_string("cimar_series.html", {"series_data": self.series_data["series"], cimar_sid: cimar_sid})
+    #    
+    #    return "No series data available"
+
+    #    return render_to_string("cimar_series.html", {"series_data": self.series_data["series"]})
+
+
+
+    def get_series_image_thumbnail_link(self,series):
+
+        middle_point = len(series["images"]) // 2
+
+        image = series["images"][middle_point]
+
+        return STORAGE_API_URL + f"/study/{self.study_details["storage_namespace"]}/{self.study_details["study_uid"]}/image/{image["id"]}/version/{image["version"]}/frame/0/thumbnail"