From dd94d8d3c5e32271c02a273386ea3cabdcad0bd1 Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Sat, 25 Jan 2025 07:51:29 -0600
Subject: Add function to retrieve a phenotype's publication data.

---
 uploader/phenotypes/models.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'uploader')

diff --git a/uploader/phenotypes/models.py b/uploader/phenotypes/models.py
index 2765d1e..9b07000 100644
--- a/uploader/phenotypes/models.py
+++ b/uploader/phenotypes/models.py
@@ -54,6 +54,20 @@ def phenotypes_count(conn: mdb.Connection,
         return int(cursor.fetchone()["total_phenos"])
 
 
+def phenotype_publication_data(conn, phenotype_id) -> Optional[dict]:
+    """Retrieve the publication data for a phenotype if it exists."""
+    with conn.cursor(cursorclass=DictCursor) as cursor:
+        cursor.execute(
+            "SELECT DISTINCT pxr.PhenotypeId, pub.* FROM PublishXRef AS pxr "
+            "INNER JOIN Publication as pub ON pxr.PublicationId=pub.Id "
+            "WHERE pxr.PhenotypeId=%s",
+            (phenotype_id,))
+        res = cursor.fetchone()
+        if res is None:
+            return res
+        return dict(res)
+
+
 def dataset_phenotypes(conn: mdb.Connection,
                        population_id: int,
                        dataset_id: int,
-- 
cgit v1.2.3