aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--scripts/insert_data.py17
1 files changed, 16 insertions, 1 deletions
diff --git a/scripts/insert_data.py b/scripts/insert_data.py
index f8b73f6..614ba4e 100644
--- a/scripts/insert_data.py
+++ b/scripts/insert_data.py
@@ -125,6 +125,21 @@ def __format_query__(query, params):
"INSERT INTO ", "INSERT INTO\n\t")
return f"{insert_str}\nVALUES\n\t{values_str};"
+def __xref_params__(dbconn: mdb.Connection, means: list[dict]) -> dict:
+ """Process params for cross-reference table."""
+ xref_names = tuple({mean["ProbeSetId"] for mean in means})
+ with dbconn.cursor(cursorclass=DictCursor) as cursor:
+ params_str = ", ".join(["%s"] * len(xref_names))
+ cursor.execute(
+ f"SELECT Name, Id FROM ProbeSet WHERE Name IN ({params_str})",
+ xref_names)
+ ids = {row["Name"]: row["Id"] for row in cursor.fetchall()}
+ return tuple({
+ **mean,
+ "ProbeSetName": mean["ProbeSetId"],
+ "ProbeSetId": ids[str(mean["ProbeSetId"])]
+ } for mean in means)
+
def insert_means(
filepath: str, speciesid: int, platform_id: int, datasetid: int,
dbconn: mdb.Connection, rconn: Redis) -> int: # pylint: disable=[unused-argument]
@@ -159,7 +174,7 @@ def insert_means(
print(__format_query__(xref_query, means))
cursor.executemany(probeset_query, means)
cursor.executemany(means_query, means)
- cursor.executemany(xref_query, means)
+ cursor.executemany(xref_query, __xref_params__(dbconn, means))
return 0
def insert_se(# pylint: disable = [too-many-arguments]