Commit bdf0221f authored by John-Paul Robinson's avatar John-Paul Robinson
Browse files

Add notebook for generating power reports

parent 2865a20e
{
"cells": [
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Power Stats \n",
"\n",
"Use RestAPI to read power consumption info for cluster nodes and generate usage reports. this is based on the [pandas time series tutorial by Jennifer Walker](https://www.dataquest.io/blog/tutorial-time-series-analysis-with-pandas/)"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"import requests\n",
"import pprint\n",
"import datetime\n",
"import os\n",
"import numpy as np\n",
"import pandas as pd\n",
"import seaborn as sns\n",
"import matplotlib.pyplot as plt\n",
"import matplotlib.dates as mdates"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Set up credentials to query RestAPI. Bright controls access based on the user identity. The user's cert.pem and cert.key are automatically generated but the cacert.pem needs to be constructed from the certs returned by the master."
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"cert_file='~/.cm/cert.pem'\n",
"key_file='~/.cm/cert.key'\n",
"ca_file='cacert.pem'"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"cert=(os.path.expanduser(cert_file), os.path.expanduser(key_file))"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"## Gather Cluster Power Data"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"startdate = '2020/01/01 00:00:00'\n",
"enddate = '2020/07/10 00:00:00'"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"params = (\n",
" ('start', startdate),\n",
" ('measurable', 'Pwr_Consumption'),\n",
" ('indent', '1'),\n",
")"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"response = requests.get('https://master:8081/rest/v1/monitoring/dump', params=params, cert=cert, verify=ca_file)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Simply read the json response into a dataframe for futher parsing."
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"df = pd.DataFrame(response.json()[\"data\"])"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Some of data values report unrealistic power values. Any reading over 10kW is considered invalid."
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"df = df.loc[df['raw'] < 10000]"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Create a datatime type column from the reported sample times."
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"df['datetime'] = pd.to_datetime(df.time, format=\"%Y/%m/%d %H:%M:%S\")"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Create an index for the hourly "
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"hourly_idx=pd.date_range(startdate, enddate, freq='H')"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"debug=False\n",
"\n",
"# prepare data frame to append to, use zeros for default column \n",
"m6_hourly_pwr=pd.DataFrame(np.zeros((1,len(hourly_idx))).T, index=hourly_idx, columns=['sum'])\n",
"\n",
"for num, entity in enumerate(df.entity.unique()):\n",
" if entity not in ['c0108', 'c0009']:\n",
" node_pwr=df[df.entity==entity].set_index(\"datetime\")\n",
" node_pwr=node_pwr[['raw']].resample('H').mean()\n",
" node_pwr=node_pwr[startdate:enddate].fillna(method=\"ffill\")\n",
" if debug:\n",
" print(node_pwr)\n",
" missing = node_pwr['raw'].isnull().sum()\n",
" print(\"{}: {} missing {}\\n\".format(num, entity, missing))\n",
" m6_hourly_pwr[entity]= node_pwr[startdate:enddate]\n"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Plot Power Usage Graph\n",
"\n",
"\n",
"Pick the start and end date for the plots from the data range selected above. Generate the sum and plot only it's values.\n",
"\n",
"We skip over the first month of collection because it is uncommonly noisy."
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"kW = m6_hourly_pwr['2020-02-01':'2020-07-09'].sum(axis=1)/1000"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"ax = kW.plot()\n",
"ax.set_ylabel(\"Power (kW)\")\n",
"ax.set_title(\"Cheaha compute and login node hourly power use\")"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Resample hourly sum to support the seven day average."
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"kW_d = kW.resample('D').mean()"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# Compute the centered 7-day rolling mean\n",
"# https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.rolling.html\n",
"kW_7d = kW_d.rolling(7, center=True).mean()"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# Plot houry, daily, 7-day rolling mean\n",
"fig, ax = plt.subplots()\n",
"ax.plot(kW, marker='.', markersize=2, color='gray', linestyle='None', label='Hourly Average')\n",
"ax.plot(kW_d, color='brown', linewidth=2, label='1-day Average')\n",
"ax.plot(kW_7d, color='black', linewidth=4, label='7-day Rolling Average')\n",
"label='Trend (7 day Rolling Mean)'\n",
"ax.legend()\n",
"ax.set_ylabel('Power (kW)')\n",
"ax.set_title('Cheaha Trends in Electricity Consumption');"
]
}
],
"metadata": {
"language_info": {
"name": "python",
"pygments_lexer": "ipython3"
}
},
"nbformat": 4,
"nbformat_minor": 4
}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment