Add function for retrieving mRNA Assay group name

1 files changed, 18 insertions, 0 deletions

diff --git a/gn3/db/datasets.py b/gn3/db/datasets.py
index 9121ec2..58bbcf5 100644
--- a/gn3/db/datasets.py
+++ b/gn3/db/datasets.py
@@ -38,6 +38,24 @@ def retrieve_sample_list(group: str):
                 samplelist = headers[3:]
     return samplelist
 
+def retrieve_mrna_group_name(connection: Any, probeset_id: int):
+    """
+    Given the trait id (ProbeSet.Id in the database), retrieve the name
+    of the group the dataset belongs to.
+    """
+    query = (
+        "SELECT iset.Name "
+        "FROM ProbeSet AS ps "
+        "INNER JOIN ProbeSetXRef AS px ON ps.Id = px.ProbeSetId "
+        "INNER JOIN InbredSet AS id ON px.InbredSetId = is.Id "
+        "WHERE ps.Id = %(probeset_id)s")
+    with connection.cursor() as cursor:
+        cursor.execute(query, {"probeset_id": probeset_id})
+        res = cursor.fetchone()
+        if res:
+            return res[0]
+        return None
+
 def retrieve_phenotype_group_name(connection: Any, dataset_id: int):
     """
     Given the dataset id (PublishFreeze.Id in the database), retrieve the name