Add the kiln-logger from the Kiln Controller project on Pi

kiln-logger.py

@@ -0,0 +1,138 @@
+#!/usr/bin/env python
+
+import websocket
+import json
+import time
+from datetime import datetime
+import csv
+import argparse
+import sys
+
+from luma.led_matrix.device import max7219
+from luma.core.interface.serial import spi, noop
+from luma.core.virtual import viewport, sevensegment
+
+import requests
+
+
+STD_HEADER = [
+    'stamp',
+    'runtime',
+    'temperature',
+    'target',
+    'state',
+    'heat',
+    'totaltime',
+    'profile',
+]
+
+
+PID_HEADER = [
+    'pid_time',
+    'pid_timeDelta',
+    'pid_setpoint',
+    'pid_ispoint',
+    'pid_err',
+    'pid_errDelta',
+    'pid_p',
+    'pid_i',
+    'pid_d',
+    'pid_kp',
+    'pid_ki',
+    'pid_kd',
+    'pid_pid',
+    'pid_out',
+]
+
+
+def logger(hostname, csvfile, noprofilestats, pidstats, stdout):
+    status_ws = websocket.WebSocket()
+
+    csv_fields = []
+    if not noprofilestats:
+        csv_fields += STD_HEADER
+    if pidstats:
+        csv_fields += PID_HEADER
+
+    out = open(csvfile, 'w')
+    csv_out = csv.DictWriter(out, csv_fields, extrasaction='ignore')
+    csv_out.writeheader()
+
+    if stdout:
+        csv_stdout = csv.DictWriter(sys.stdout, csv_fields, extrasaction='ignore', delimiter='\t')
+        csv_stdout.writeheader()
+    else:
+        csv_stdout = None
+
+    # setup seven segment display
+
+#    serial = spi(port=0, device=0, gpio=noop())
+    serial = spi(port=1, device=0, gpio=noop())
+    device = max7219(serial, cascaded=2)
+    seg = sevensegment(device)
+
+    # end setup seven segment display
+
+    while True:
+        try:
+            msg = json.loads(status_ws.recv())
+
+        except websocket.WebSocketException:
+            try:
+                status_ws.connect(f'ws://{hostname}/status')
+            except Exception:
+                time.sleep(5)
+
+            continue
+
+        if msg.get('type') == 'backlog':
+            continue
+
+        if not noprofilestats:
+            msg['stamp'] = time.time()
+        if pidstats and 'pidstats' in msg:
+            for k, v in msg.get('pidstats', {}).items():
+                msg[f"pid_{k}"] = v
+
+        csv_out.writerow(msg)
+        out.flush()
+
+        if stdout:
+            for k in list(msg.keys()):
+                v = msg[k]
+                if isinstance(v, float):
+                    msg[k] = '{:5.3f}'.format(v)
+            csv_stdout.writerow(msg)
+            sys.stdout.flush()
+        # ADD THE TEMPERATURE HERE
+        try:
+            req = requests.get('http://127.0.0.1:8081/api/stats')
+
+        except KeyError:
+            try:
+                req = requests.get('http://127.0.0.1:8081/api/stats')
+            except Exception:
+                time.sleep(25)
+
+            continue
+
+        temp = round(req.json()['ispoint'])
+        time_rem = msg['totaltime'] - msg['runtime']
+        hours = round(time_rem // 3600)
+        minutes = round((time_rem - (hours * 3600)) // 60)
+        seconds = round((time_rem - (hours * 3600) - (minutes * 60)))
+
+        seg.text = f"TEMP{temp:4.0f}{hours:2.0f}-{minutes:2.0f}-{seconds:2.0f}"
+        # END ADD TEMP
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description='Log kiln data for analysis.')
+    parser.add_argument('--hostname', type=str, default="localhost:8081", help="The kiln-controller hostname:port")
+    parser.add_argument('--csvfile', type=str, default="/tmp/kilnstats.csv", help="Where to write the kiln stats to")
+    parser.add_argument('--pidstats', action='store_true', help="Include PID stats")
+    parser.add_argument('--noprofilestats', action='store_true', help="Do not store profile stats (default is to store them)")
+    parser.add_argument('--stdout', action='store_true', help="Also print to stdout")
+    args = parser.parse_args()
+
+    logger(args.hostname, args.csvfile, args.noprofilestats, args.pidstats, args.stdout)