diff options
Diffstat (limited to 'wqflask/db/webqtlDatabaseFunction.py')
| -rw-r--r-- | wqflask/db/webqtlDatabaseFunction.py | 6 |
1 files changed, 4 insertions, 2 deletions
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] |