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/base/data_set/probably_unused.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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

diff --git a/wqflask/base/data_set/probably_unused.py b/wqflask/base/data_set/probably_unused.py
index 4e54bcff..abd3ad07 100644
--- a/wqflask/base/data_set/probably_unused.py
+++ b/wqflask/base/data_set/probably_unused.py
@@ -3,6 +3,7 @@
 import pickle as pickle
 
 from wqflask.database import database_connection
+from utility.tools import get_setting
 
 def create_datasets_list():
     if USE_REDIS:
@@ -19,7 +20,7 @@ def create_datasets_list():
                      'Geno': 'GenoFreeze'}
 
         for dataset_type in type_dict:
-            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 Name FROM %s",
                                (type_dict[dataset_type],))
                 results = cursor.fetchall(query)
-- 
cgit v1.2.3