From d44d87fce2a2daa03269ffb634a85854464cc78a Mon Sep 17 00:00:00 2001 From: Matthew Stidham Date: Fri, 23 Jul 2021 09:36:06 -0700 Subject: [PATCH] Check to make sure that column variables exist in table before grabbing data Signed-off-by: Matthew Stidham --- py-scripts/csv_to_influx.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/py-scripts/csv_to_influx.py b/py-scripts/csv_to_influx.py index c26908f6..c69735e4 100755 --- a/py-scripts/csv_to_influx.py +++ b/py-scripts/csv_to_influx.py @@ -77,9 +77,10 @@ class CSVtoInflux(): date = row[columns['Date']] date = datetime.datetime.utcfromtimestamp(int(date) / 1000).isoformat() #convert to datetime so influx can read it, this is required for variable in csv_variables: - index = columns[variable] - influx_variable = csv_vs_influx[variable] - tags[influx_variable] = row[index] + if variable in columns.keys(): + index = columns[variable] + influx_variable = csv_vs_influx[variable] + tags[influx_variable] = row[index] self.influxdb.post_to_influx(short_description, numeric_score, tags, date) def script_name(self):