diff options
| -rw-r--r-- | scripts/insert_data.py | 8 |
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" |