From 18f4682a27630bd81eb0b1514476c2cf0f663a63 Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Mon, 12 Sep 2022 06:45:59 +0300
Subject: Check db resultset before indexing into it

The assumption that the resultset will never be empty was leading to a
lot of errors.
---
 wqflask/base/data_set.py | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

(limited to 'wqflask/base/data_set.py')

diff --git a/wqflask/base/data_set.py b/wqflask/base/data_set.py
index dda5e8f9..47c03b06 100644
--- a/wqflask/base/data_set.py
+++ b/wqflask/base/data_set.py
@@ -328,8 +328,11 @@ class DatasetGroup:
                     "InbredSet.InbredSetCode "
                     "FROM InbredSet WHERE Name = %s",
                     (dataset.name,))
-            (self.name, self.id,
-             self.genetic_type, self.code) = cursor.fetchone()
+            results = cursor.fetchone()
+            if results:
+                (self.name, self.id, self.genetic_type, self.code) = results
+            else:
+                self.name = name or dataset.name
         if self.name == 'BXD300':
             self.name = "BXD"
 
@@ -353,7 +356,9 @@ class DatasetGroup:
                 "SELECT MappingMethodId FROM "
                 "InbredSet WHERE Name= %s",
                 (self.name,))
-            mapping_id = cursor.fetchone()[0]
+            results = cursor.fetchone()
+            if results and results[0]:
+                mapping_id = results[0]
         if mapping_id == "1":
             mapping_names = ["GEMMA", "QTLReaper", "R/qtl"]
         elif mapping_id == "2":
@@ -1227,8 +1232,8 @@ def geno_mrna_confidentiality(ob):
             f"AuthorisedUsers FROM {ob.type}Freeze WHERE Name = %s",
             (ob.name,)
         )
-        result = cursor.fetchall()[0]
-        if result:
+        result = cursor.fetchall()
+        if len(result) > 0 and result[0]:
             return True
 
 
-- 
cgit v1.2.3