From 528bd9934f8d9cb430264d0714804cfd541b1b6c Mon Sep 17 00:00:00 2001
From: Pim van den Berg <pim.van.den.berg@mendix.com>
Date: Sun, 8 Nov 2020 15:05:13 +0100
Subject: [PATCH] feat(itho-wpu): export to influxdb support

Export getdatalog measurements to InfluxDB by giving the
'--export-to-influxdb' argument. Additionally configure the InfluxDB
credentials via environment variables:
- INFLUXDB_HOST (default: localhost)
- INFLUXDB_PORT (default: 8086)
- INFLUXDB_USERNAME (default: root)
- INFLUXDB_PASSWORD (default: root)
- INFLUXDB_DATABASE
---
 itho-wpu.py | 34 ++++++++++++++++++++++++++++++++--
 1 file changed, 32 insertions(+), 2 deletions(-)

diff --git a/itho-wpu.py b/itho-wpu.py
index d93932f..4df073f 100755
--- a/itho-wpu.py
+++ b/itho-wpu.py
@@ -7,6 +7,8 @@ import pigpio
 import queue
 import sys
 import time
+import os
+import datetime
 from collections import namedtuple
 
 consolelogformatter = logging.Formatter("%(asctime)-15s %(levelname)s: %(message)s")
@@ -17,6 +19,29 @@ stdout_log_handler.setFormatter(consolelogformatter)
 logger.addHandler(stdout_log_handler)
 
 
+def export_to_influxdb(action, measurements):
+    from influxdb import InfluxDBClient
+
+    influx_client = InfluxDBClient(
+        host=os.getenv('INFLUXDB_HOST', 'localhost'),
+        port=os.getenv('INFLUXDB_PORT', 8086),
+        username=os.getenv('INFLUXDB_USERNAME', 'root'),
+        password=os.getenv('INFLUXDB_PASSWORD', 'root'),
+        database=os.getenv('INFLUXDB_DATABASE')
+    )
+    json_body = [
+        {
+            "measurement": action,
+            "time": datetime.datetime.utcnow().replace(microsecond=0).isoformat(),
+            "fields": measurements,
+        }
+    ]
+    try:
+        influx_client.write_points(json_body)
+    except Exception as e:
+        print('Failed to write to influxdb: ', e)
+
+
 def parse_args():
     parser = argparse.ArgumentParser(description='Itho WPU i2c master')
 
@@ -35,6 +60,8 @@ def parse_args():
     parser.add_argument('--slave-only', action='store_true', help="Only run I2C slave")
     parser.add_argument('--slave-timeout', nargs='?', type=int, default=60,
                         help="Slave timeout in seconds when --slave-only")
+    parser.add_argument('--export-to-influxdb', action='store_true',
+                        help="Export results to InfluxDB")
 
     args = parser.parse_args()
     return args
@@ -115,10 +142,13 @@ class I2CMaster:
         self.i.close()
 
 
-def process_response(action, response):
+def process_response(action, response, args):
     if action == "getdatalog" and int(response[1], 0) == 0xA4 and int(response[2], 0) == 0x01:
         measurements = process_datalog(response)
 
+    if args.export_to_influxdb:
+        export_to_influxdb(action, measurements)
+
 
 def process_datalog(response):
     # 0 = Byte
@@ -195,7 +225,7 @@ if __name__ == "__main__":
         master.close()
 
         if result is not None:
-            process_response(args.action, result)
+            process_response(args.action, result, args)
 
     if not args.master_only:
         slave.close()