From 5217a366e7904142fc443594facb126d28cd2b25 Mon Sep 17 00:00:00 2001
From: BonfaceKilz
Date: Tue, 27 Apr 2021 11:58:54 +0300
Subject: db: traits: Add extra method for looking up webqtl datasets

---
 gn3/db/traits.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

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

diff --git a/gn3/db/traits.py b/gn3/db/traits.py
index 3134d31..0cebe2c 100644
--- a/gn3/db/traits.py
+++ b/gn3/db/traits.py
@@ -34,6 +34,19 @@ class webqtlCaseData:
         return _str
 
 
+def lookup_webqtldataset_name(riset_name: str, conn: Any):
+    """Given a group name(riset), return it's name e.g. BXDPublish,
+HLCPublish."""
+    with conn.cursor() as cursor:
+        cursor.execute(
+            "SELECT PublishFreeze.Name FROM "
+            "PublishFreeze, InbredSet WHERE "
+            "PublishFreeze.InbredSetId = InbredSet.Id "
+            "AND InbredSet.Name = '%s'" % riset_name)
+        _result, *_ = cursor.fetchone()
+        return _result
+
+
 def get_riset(data_type: str, name: str, conn: Any):
     query, _name, _id = None, None, None
     if data_type == "Publish":
-- 
cgit v1.2.3