From c6f9aca5917d9838680349dc91db215c9a38ecda Mon Sep 17 00:00:00 2001
From: BonfaceKilz
Date: Tue, 13 Apr 2021 08:18:19 +0300
Subject: Add method to fetch riset name and id

---
 gn3/db/__init__.py |  0
 gn3/db/traits.py   | 31 +++++++++++++++++++++++++++++++
 2 files changed, 31 insertions(+)
 create mode 100644 gn3/db/__init__.py
 create mode 100644 gn3/db/traits.py

diff --git a/gn3/db/__init__.py b/gn3/db/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/gn3/db/traits.py b/gn3/db/traits.py
new file mode 100644
index 0000000..b0d1831
--- /dev/null
+++ b/gn3/db/traits.py
@@ -0,0 +1,31 @@
+"""This contains all the necessary functions that are required to add traits
+to the published database"""
+from collections import namedtuple
+from typing import Any
+
+
+riset = namedtuple('riset', ['name', 'id'])
+
+
+def get_riset(data_type: str, name: str, conn: Any):
+    query, _name, _id = None, None, None
+    if data_type == "Publish":
+        query = ("SELECT InbredSet.Name, InbredSet.Id FROM InbredSet, "
+                 "PublishFreeze WHERE PublishFreeze.InbredSetId = "
+                 "InbredSet.Id AND PublishFreeze.Name = '%s'" % name)
+    elif data_type == "Geno":
+        query = ("SELECT InbredSet.Name, InbredSet.Id FROM InbredSet, "
+                 "GenoFreeze WHERE GenoFreeze.InbredSetId = "
+                 "InbredSet.Id AND GenoFreeze.Name = '%s'" % name)
+    elif data_type == "ProbeSet":
+        query = ("SELECT InbredSet.Name, InbredSet.Id FROM "
+                 "InbredSet, ProbeSetFreeze, ProbeFreeze WHERE "
+                 "ProbeFreeze.InbredSetId = InbredSet.Id AND "
+                 "ProbeFreeze.Id = ProbeSetFreeze.ProbeFreezeId AND "
+                 "ProbeSetFreeze.Name = '%s'" % name)
+    if query:
+        with conn.cursor() as cursor:
+            _name, _id = cursor.fetchone()
+            if _name == "BXD300":
+                _name = "BXD"
+    return riset(_name, _id)
-- 
cgit v1.2.3