pm.py

import argparse
import datetime
import serial
import time


class WattsUp:
    def __init__(self, ports, interval):
        self.sockets = [serial.Serial(port, 115200) for port in ports]
        self.interval = interval

    def fetch_power(self):
        powers = []
        for s in self.sockets:
            line = s.readline()
            if line.startswith(b'#d'):
                fields = line.split(b',')
                if len(fields) > 5:
                    powers.append(float(fields[3]) / 10)
        return powers

    def log(self, logfile):
        with open(logfile, 'w') as f:
            f.write('time,id,' + ','.join(f'pm{i + 1}' for i in range(len(self.sockets))) + ',sum\n')
            f.flush()
            n = 0
            while True:
                powers = self.fetch_power()
                if powers:
                    total_power = sum(powers)
                    timestamp = datetime.datetime.now()
                    f.write(f"{timestamp},{n}," + ','.join(f"{p:.1f}" for p in powers) + f",{total_power:.1f}\n")
                    f.flush()
                    n += self.interval
                time.sleep(self.interval)


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--ports", required=True, help="Comma-separated list of serial ports")
    parser.add_argument("--interval", type=float, required=True, help="Sampling interval in seconds")
    parser.add_argument("-o", "--output", required=True, help="Output log file")
    args = parser.parse_args()

    ports = args.ports.split(",")
    meter = WattsUp(ports, args.interval)
    meter.log(args.output)


if __name__ == "__main__":
    main()