From 971d29e82391cc269405f59b5aad5f31061b633d Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Thu, 22 Jun 2023 16:31:18 +0300
Subject: Update calls to `database_connection` function

Pass in the database URI at call time.
---
 wqflask/db/webqtlDatabaseFunction.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'wqflask/db')

diff --git a/wqflask/db/webqtlDatabaseFunction.py b/wqflask/db/webqtlDatabaseFunction.py
index 878792b4..b09cf377 100644
--- a/wqflask/db/webqtlDatabaseFunction.py
+++ b/wqflask/db/webqtlDatabaseFunction.py
@@ -22,12 +22,14 @@
 
 from wqflask.database import database_connection
 
+from utility.tools import get_setting
+
 
 def retrieve_species(group):
     """Get the species of a group (e.g. returns string "mouse" on "BXD"
 
     """
-    with database_connection() as conn, conn.cursor() as cursor:
+    with database_connection(get_setting("SQL_URI")) as conn, conn.cursor() as cursor:
         cursor.execute(
             "SELECT Species.Name FROM Species, InbredSet WHERE InbredSet.Name = %s AND InbredSet.SpeciesId = Species.Id",
             (group,))
@@ -37,7 +39,7 @@ def retrieve_species(group):
 
 
 def retrieve_species_id(group):
-    with database_connection() as conn, conn.cursor() as cursor:
+    with database_connection(get_setting("SQL_URI")) as conn, conn.cursor() as cursor:
         cursor.execute("SELECT SpeciesId FROM InbredSet WHERE Name = %s",
                        (group,))
         return cursor.fetchone()[0]
-- 
cgit v1.2.3