prettify code

This commit is contained in:
Thomas Ruoff
2020-01-10 23:46:52 +01:00
parent 1aebb3e2c6
commit 028f25f299

View File

@@ -51,40 +51,41 @@ import sys
import csv import csv
from datetime import datetime, timedelta from datetime import datetime, timedelta
def get_timestamp(timestr): def get_timestamp(timestr):
"""Converts CSV time value to a UTC timestamp in seconds""" """Converts CSV time value to a UTC timestamp in seconds"""
naive_dt = datetime.strptime(timestr, '%Y/%m/%d %H:%M:%S') naive_dt = datetime.strptime(timestr, "%Y/%m/%d %H:%M:%S")
utc = (naive_dt - datetime(1970, 1, 1)) / timedelta(seconds=1) utc = (naive_dt - datetime(1970, 1, 1)) / timedelta(seconds=1)
return int(utc) return int(utc)
SENSOR_ID = os.environ.get('SENSOR_ID', '16229960')
DATABASE = os.environ.get('INFLUXDB_DATABASE', 'sensors')
NODE = 'esp8266-' + SENSOR_ID SENSOR_ID = os.environ.get("SENSOR_ID", "16229960")
DATABASE = os.environ.get("INFLUXDB_DATABASE", "sensors")
NODE = "esp8266-" + SENSOR_ID
NAME_MAP = { NAME_MAP = {
'Humidity': "Humidity": "humidity",
'humidity', "Max_cycle": "max_micro",
'Max_cycle': 'max_micro', "Samples": "samples",
'Samples': 'samples', "Min_cycle": "min_micro",
'Min_cycle': 'min_micro', "Signal": "signal",
'Signal': 'signal', "Temp": "temperature",
'Temp': 'temperature'
} }
READER = csv.DictReader(sys.stdin, delimiter=';') READER = csv.DictReader(sys.stdin, delimiter=";")
for row in READER: for row in READER:
measurements = [] measurements = []
for header, value in row.items(): for header, value in row.items():
if header == 'Time' or not value: if header == "Time" or not value:
continue continue
measurements.append('{0}={1}'.format(NAME_MAP.get(header, header), value)) measurements.append("{0}={1}".format(NAME_MAP.get(header, header), value))
values = { values = {
'database': DATABASE, "database": DATABASE,
'node': NODE, "node": NODE,
'measurements': ','.join(measurements), "measurements": ",".join(measurements),
'time': get_timestamp(row['Time']) "time": get_timestamp(row["Time"]),
} }
print('{database},node={node} {measurements} {time}'.format(**values)) print("{database},node={node} {measurements} {time}".format(**values))