diff --git a/src/ntp_routes.py b/src/ntp_routes.py index 3093988..865b738 100644 --- a/src/ntp_routes.py +++ b/src/ntp_routes.py @@ -3,6 +3,7 @@ from loguru import logger import json import plotly.express as px import plotly.graph_objects as po +import matplotlib.pyplot as plt import pandas as pd import psycopg import sqlalchemy @@ -118,3 +119,45 @@ def ntpserver(): dbh.close() +def get_dataframe(): + +@app.route('/plot.png') +def plot_png(): + dbh = psycopg.connect() + engine = sqlalchemy.create_engine("postgresql+psycopg://", creator=lambda: dbh) + query = """ + select time_bucket('5 minutes', time) as bucket, + device, + avg(cast(values->'rootdisp'->>'value' as float)) as rootdisp, + max(cast(values->'stratum'->>'value' as int)) as stratum + from measurements + where time >= date_trunc('day', now()) AND time < date_trunc('day', now()) + '1 day'::interval and + application = 'TSM' and attributes->>'Label' = 'david' + group by bucket, device + order by bucket, device + """ + df = pd.read_sql(query, con=engine) + + fig, ax1 = plt.subplots() + + ax1.plot(df['bucket'], df['rootdisp'], 'r-', label='Root Dispersion') + ax1.set_xlabel('Time') + ax1.set_ylabel('Root Dispersion (ms)', color='r') + ax1.tick_params(axis='y', labelcolor='r') + + ax2 = ax1.twinx() + ax2.plot(df['bucket'], df['stratum'], 'b-', label='Stratum') + ax2.set_ylabel('Stratum', color='b') + ax2.tick_params(axis='y', labelcolor='b') + ax2.set_yticks(range(int(df['stratum'].min()), int(df['stratum'].max()) + 1)) + + fig.suptitle('NTP Server Numbers') + fig.tight_layout() + + img_io = io.BytesIO() + plt.savefig(img_io, format='png') + img_io.seek(0) + plt.close(fig) + + return Response(img_io, mimetype='image/png') +