m100_find_max_power.py

#!/usr/bin/env python3
import argparse
import os
import pandas as pd

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("input_power_timeseries_prefix", help="filepath prefix to the location of the parquet files that contain node power consumption time series")
    parser.add_argument("month", nargs='+', help="the month to aggregate. example value: '22-07'")
    args = parser.parse_args()

    overall_max = -1
    for month in args.month:
        month_prefix = f'{args.input_power_timeseries_prefix}{month}'
        power_file = f'{month_prefix}_power_total.parquet'

        df = pd.read_parquet(power_file)
        max_month_power = df.groupby('timestamp')['value'].sum().max()
        print(f'max power reached during month {month}: {max_month_power} W')
        overall_max = max(overall_max, max_month_power)
    print(f'max power reached over all months: {overall_max} W')