kopia lustrzana https://github.com/jbruce12000/kiln-controller
87 wiersze
2.0 KiB
Python
87 wiersze
2.0 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
import websocket
|
||
|
|
import json
|
||
|
|
import time
|
||
|
|
import csv
|
||
|
|
import argparse
|
||
|
|
|
||
|
|
|
||
|
|
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):
|
||
|
|
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()
|
||
|
|
|
||
|
|
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 __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)")
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
logger(args.hostname, args.csvfile, args.noprofilestats, args.pidstats)
|