From c3707f1e8045c72b5d0686c0a9ef239a396997b3 Mon Sep 17 00:00:00 2001
From: zsloan
Date: Fri, 2 May 2025 18:28:12 +0000
Subject: Add function for retrieving group_id given group_name

---
 gn3/db/datasets.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'gn3/db/datasets.py')

diff --git a/gn3/db/datasets.py b/gn3/db/datasets.py
index f3b4f9f..fea207b 100644
--- a/gn3/db/datasets.py
+++ b/gn3/db/datasets.py
@@ -79,6 +79,21 @@ def retrieve_mrna_group_name(connection: Any, probeset_id: int, dataset_name: st
             return res[0]
         return None
 
+def retrieve_group_id(connection: Any, group_name: str):
+    """
+    Given the group name, retrieve the group ID
+    """
+    query = (
+        "SELECT iset.Id "
+        "FROM InbredSet AS iset "
+        "WHERE iset.Name = %(group_name)s")
+    with connection.cursor() as cursor:
+        cursor.execute(query, {"group_name": group_name})
+        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
-- 
cgit 1.4.1