aboutsummaryrefslogtreecommitdiff
path: root/scripts/insert_data.py
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/insert_data.py')
-rw-r--r--scripts/insert_data.py8
1 files changed, 5 insertions, 3 deletions
diff --git a/scripts/insert_data.py b/scripts/insert_data.py
index 63e85ef..0ac3799 100644
--- a/scripts/insert_data.py
+++ b/scripts/insert_data.py
@@ -67,9 +67,10 @@ def strains_info(
def read_datavalues(filepath, headings, strain_info):
"""Read numerical, data values from the file."""
+ id_key = headings[0]
return {
- str(row["ProbeSetID"]): tuple({
- "ProbeSetName": str(row["ProbeSetID"]),
+ str(row[id_key]): tuple({
+ "ProbeSetName": str(row[id_key]),
"StrainId": strain_info[sname]["Id"],
"DataValue": float(row[sname])
} for sname in headings[1:])
@@ -79,9 +80,10 @@ def read_datavalues(filepath, headings, strain_info):
def read_probesets(filepath, headings):
"""Read the ProbeSet names."""
+ id_key = headings[0]
for row in (dict(zip(headings, line))
for line in read_file_contents(filepath)):
- yield {"Name": str(row["ProbeSetID"])}
+ yield {"Name": str(row[id_key])}
def last_data_id(dbconn: mdb.Connection) -> int:
"Get the last id from the database"