aboutsummaryrefslogtreecommitdiff
path: root/gn3/db
diff options
context:
space:
mode:
Diffstat (limited to 'gn3/db')
-rw-r--r--gn3/db/species.py20
1 files changed, 8 insertions, 12 deletions
diff --git a/gn3/db/species.py b/gn3/db/species.py
index 1e5015f..abcbf64 100644
--- a/gn3/db/species.py
+++ b/gn3/db/species.py
@@ -47,17 +47,13 @@ def translate_to_mouse_gene_id(species: str, geneid: int, conn: Any) -> int:
return geneid
with conn.cursor as cursor:
- if species == "rat":
- cursor.execute(
- "SELECT mouse FROM GeneIDXRef WHERE rat = %s", geneid)
- rat_geneid = cursor.fetchone()
- if rat_geneid:
- return rat_geneid[0]
-
- cursor.execute(
- "SELECT mouse FROM GeneIDXRef WHERE human = %s", geneid)
- human_geneid = cursor.fetchone()
- if human_geneid:
- return human_geneid[0]
+ query = {
+ "rat": "SELECT mouse FROM GeneIDXRef WHERE rat = %s"
+ "human": "SELECT mouse FROM GeneIDXRef WHERE human = %s"
+ }
+ cursor.execute(query[species], geneid)
+ translated_gene_id = cursor.fetchone()
+ if translated_gene_id:
+ return translated_gene_id[0]
return 0 # default if all else fails