powerlog.py

import time
import adpower
import energystat
import argparse


interval = 5
timeout = 120

parser = argparse.ArgumentParser()
parser.add_argument("-i", help="interval", type=int, default=interval)
parser.add_argument("-t", help="timeout in secs", type=int, default=timeout)

args = parser.parse_args()

adp = adpower.AdPower()
energy_stat = energystat.EnergyStat(int(args.i))

current_time = time.time()
next_start_time = current_time + (args.i - (current_time % args.i))
time_to_wait = next_start_time - current_time
if time_to_wait > 0:
    time.sleep(time_to_wait)

start_time = time.time()

adp.start()

try:
    while True:
        elapsed_time = time.time() - start_time
        if elapsed_time > args.t:
            break
        res = adp.probe()
        if res is None:
            print("No data received")
            break

        energy_stat.add_power(res[0])
        energy_info = energy_stat.get_energy()
        if energy_info:
            ts = time.strftime('%m-%d %H:%M:%S')
            print(f"{ts} {energy_info[0]:.4f} {energy_info[1]:.4f}", flush=True)
        time.sleep(0.1)
except KeyboardInterrupt:
    pass

adp.stop()