From c13eaeae7fc63195e2ee9aa071aa2282a583a96b Mon Sep 17 00:00:00 2001
From: Sean Kelly <kelly@seankelly.biz>
Date: Thu, 14 Mar 2024 13:07:44 -0500
Subject: [PATCH] Support #348

---
 ...rotocol_outcome_protocol_secure_outcome.py | 26 +++++++++++++++++++
 .../src/eke/knowledge/protocols.py            | 16 ++++++++++++
 .../templates/eke.knowledge/protocol.html     | 10 +++++++
 3 files changed, 52 insertions(+)
 create mode 100644 src/eke.knowledge/src/eke/knowledge/migrations/0018_protocol_outcome_protocol_secure_outcome.py

diff --git a/src/eke.knowledge/src/eke/knowledge/migrations/0018_protocol_outcome_protocol_secure_outcome.py b/src/eke.knowledge/src/eke/knowledge/migrations/0018_protocol_outcome_protocol_secure_outcome.py
new file mode 100644
index 00000000..81e52081
--- /dev/null
+++ b/src/eke.knowledge/src/eke/knowledge/migrations/0018_protocol_outcome_protocol_secure_outcome.py
@@ -0,0 +1,26 @@
+# Generated by Django 4.2.10 on 2024-03-14 16:46
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("ekeknowledge", "0017_protocol_estimated_finish_date_protocol_start_date"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="protocol",
+            name="outcome",
+            field=models.TextField(
+                blank=True, help_text="What this protocol's net result was"
+            ),
+        ),
+        migrations.AddField(
+            model_name="protocol",
+            name="secure_outcome",
+            field=models.TextField(
+                blank=True, help_text="What this protocol's secret net result was"
+            ),
+        ),
+    ]
diff --git a/src/eke.knowledge/src/eke/knowledge/protocols.py b/src/eke.knowledge/src/eke/knowledge/protocols.py
index 2088107a..048d838a 100644
--- a/src/eke.knowledge/src/eke/knowledge/protocols.py
+++ b/src/eke.knowledge/src/eke/knowledge/protocols.py
@@ -50,6 +50,7 @@ def compute_new_value(self, modelField: Field, value: str, predicates: dict) ->
             rdflib.URIRef('http://edrn.nci.nih.gov/rdf/schema.rdf#objective'),
             rdflib.URIRef('http://edrn.nci.nih.gov/rdf/schema.rdf#aims'),
             rdflib.URIRef('http://edrn.nci.nih.gov/rdf/schema.rdf#outcome'),
+            rdflib.URIRef('http://edrn.nci.nih.gov/rdf/schema.rdf#secureOutcome')
         ):
             text = predicates.get(pred, [''])[0]
             if text:
@@ -113,6 +114,8 @@ class Protocol(KnowledgeObject):
     # 🤬 Get bent    # collaborativeGroupsDeNormalized = models.CharField(max_length=400, blank=True, null=False, help_text='🙄')
     collaborativeGroup = models.CharField(max_length=400, blank=True, null=False, help_text='Collaborative Group')
     kind = models.CharField(max_length=80, blank=True, null=False, help_text='Protocol type')
+    outcome = models.TextField(null=False, blank=True, help_text="What this protocol's net result was")
+    secure_outcome = models.TextField(null=False, blank=True, help_text="What this protocol's secret net result was")
     content_panels = KnowledgeObject.content_panels + [
         FieldPanel('coordinatingInvestigatorSite'),
         FieldPanel('leadInvestigatorSite'),
@@ -132,6 +135,8 @@ class Protocol(KnowledgeObject):
         FieldPanel('collaborativeGroup'),
         FieldPanel('kind'),
         FieldPanel('publications'),
+        FieldPanel('outcome'),
+        FieldPanel('secure_outcome')
     ]
     search_fields = KnowledgeObject.search_fields + [
         index.SearchField('abbreviation'),
@@ -140,6 +145,8 @@ class Protocol(KnowledgeObject):
         index.FilterField('piName'),
         index.FilterField('collaborativeGroup'),
         index.FilterField('cancer_types'),
+        index.FilterField('outcome'),
+        index.FilterField('secure_outcome'),
     ]
     class Meta:
         pass
@@ -164,14 +171,18 @@ class RDFMeta:
             _internalIDPredicate: RDFAttribute('protocolID', scalar=True),
             str(rdflib.DCTERMS.title): RDFAttribute('title', scalar=True),
             str(rdflib.DCTERMS.description): _ComplexDescriptionRDFAttribute('description', scalar=True),
+            esu('outcome'): _ComplexDescriptionRDFAttribute('outcome', scalar=True),
+            esu('secureOutcome'): _ComplexDescriptionRDFAttribute('secure_outcome', scalar=True),
             esu('protocolType'): RDFAttribute('kind', scalar=True)
         }
+
     def _authentication(self, request: HttpRequest) -> dict:
         '''Given a request, determine if the user is authenticated and what the login link would be if not.'''
         params = {'authenticated': request.user.is_authenticated}
         if not params['authenticated']:
             params['login'] = reverse('wagtailcore_login') + '?next=' + request.path
         return params
+
     def get_context(self, request: HttpRequest, *args, **kwargs) -> dict:
         '''Get the context for the page template.'''
         context = super().get_context(request, args, kwargs)
@@ -192,7 +203,12 @@ def get_context(self, request: HttpRequest, *args, **kwargs) -> dict:
         context['biomarkers'] = bms
         context['cancer_types'] = [i for i in self.cancer_types.all().order_by('title').values_list('title', flat=True)]
         context['publications'] = [i for i in self.publications.order_by(Lower('title'))]
+
+        # Show the new secure outcome only for logged in users
+        context['show_secure_outcome'] = request and request.user.is_authenticated
+
         return context
+
     def data_table(self) -> dict:
         if self.leadInvestigatorSite:
             if self.leadInvestigatorSite.pi:
diff --git a/src/eke.knowledge/src/eke/knowledge/templates/eke.knowledge/protocol.html b/src/eke.knowledge/src/eke/knowledge/templates/eke.knowledge/protocol.html
index edfdd49e..f3c2f81f 100644
--- a/src/eke.knowledge/src/eke/knowledge/templates/eke.knowledge/protocol.html
+++ b/src/eke.knowledge/src/eke/knowledge/templates/eke.knowledge/protocol.html
@@ -74,6 +74,16 @@ <h3>Comments</h3>
                 <p>{{page.comments|safe}}</p>
             {% endif %}
 
+            {% if page.outcome %}
+                <h3>Outcome</h3>
+                <p>{{page.outcome|safe}}</p>
+            {% endif %}
+
+            {% if page.secure_outcome and show_secure_outcome %}
+                <h3>Secure Outcome</h3>
+                <p>{{page.secure_outcome|safe}}</p>
+            {% endif %}
+
             <h2>Publications</h2>
             <ul>
                 {% for publication in publications %}