138 lines
3.6 KiB
Python
Executable File
138 lines
3.6 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
from dash.dependencies import Input, Output
|
|
import dash
|
|
import dash_core_components as dcc
|
|
import dash_html_components as html
|
|
|
|
import pandas as pd
|
|
from functools import lru_cache
|
|
|
|
import argparse
|
|
import sqlite3
|
|
import time
|
|
import sys
|
|
import datetime
|
|
|
|
external_stylesheets = ['https://codepen.io/chriddyp/pen/bWLwgP.css']
|
|
|
|
app = dash.Dash(__name__, external_stylesheets=external_stylesheets)
|
|
|
|
|
|
def get_topics(db):
|
|
cursor = db.cursor()
|
|
query = 'SELECT name FROM sqlite_master WHERE type="table"'
|
|
q = cursor.execute(query)
|
|
topics = [t[0] for t in q.fetchall()]
|
|
return topics
|
|
|
|
|
|
update_interval_s = 4
|
|
|
|
# Since we're adding callbacks to elements that don't exist in the app.layout,
|
|
# Dash will raise an exception to warn us that we might be
|
|
# doing something wrong.
|
|
# In this case, we're adding the elements through a callback, so we can ignore
|
|
# the exception.
|
|
app.config.suppress_callback_exceptions = True
|
|
|
|
app.layout = html.Div([
|
|
dcc.Location(id='url', refresh=False),
|
|
html.Div(id='page-content')
|
|
])
|
|
|
|
|
|
@lru_cache(maxsize=10)
|
|
def get_data(topic, timerange, ts):
|
|
conn = sqlite3.connect(db)
|
|
table = topic
|
|
|
|
earliest = datetime.datetime.now() - timerange
|
|
query = f"SELECT * FROM {table} WHERE timestamp > '{earliest}'"
|
|
|
|
df = pd.read_sql(query, conn, parse_dates=[
|
|
'timestamp'], index_col='timestamp')
|
|
return df
|
|
|
|
|
|
plot_page = html.Div([
|
|
dcc.Interval(
|
|
id='interval-component',
|
|
interval=4*1000, # in milliseconds
|
|
n_intervals=0),
|
|
dcc.Dropdown(
|
|
id='timerange',
|
|
options=[
|
|
{'label': 'Last hour', 'value': 'h'},
|
|
{'label': 'Last day', 'value': 'd'},
|
|
{'label': 'Last week', 'value': 'w'},
|
|
{'label': 'All', 'value': 'all'}
|
|
],
|
|
value='d'
|
|
),
|
|
dcc.Graph(id='graph', animate=False),
|
|
dcc.Link('Back', href='/')
|
|
])
|
|
|
|
def get_timedelta(range_str):
|
|
range_to_delta = {
|
|
'h': datetime.timedelta(hours=1),
|
|
'd': datetime.timedelta(days=1),
|
|
'w': datetime.timedelta(weeks=1),
|
|
'all': datetime.timedelta(weeks=10000),
|
|
}
|
|
return range_to_delta[range_str]
|
|
|
|
|
|
@app.callback(Output('graph', 'figure'),
|
|
[Input('interval-component', 'n_intervals'),
|
|
Input('timerange', 'value'),
|
|
Input('url', 'pathname')])
|
|
def update_graph(interval, timerange, pathname):
|
|
topics = pathname.strip('/').split('+')
|
|
if len(topics):
|
|
delta = get_timedelta(timerange)
|
|
now = int(time.time()) // update_interval_s
|
|
def make_data(df, name):
|
|
return {
|
|
'x': df.index,
|
|
'y': df['value'],
|
|
'type': 'scatter',
|
|
'mode': 'lines',
|
|
'name': name
|
|
}
|
|
return {
|
|
'data': [make_data(get_data(topic, delta, now), topic) for topic in topics]}
|
|
else:
|
|
return {}
|
|
|
|
# Update the index
|
|
@app.callback(dash.dependencies.Output('page-content', 'children'),
|
|
[dash.dependencies.Input('url', 'pathname')])
|
|
def display_page(pathname):
|
|
if pathname == '/':
|
|
return index_page
|
|
else:
|
|
return plot_page
|
|
# You could also return a 404 "URL not found" page here
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
options = argparse.ArgumentParser()
|
|
options.add_argument('--debug', '-d', action='store_true')
|
|
options.add_argument('db')
|
|
|
|
args = options.parse_args()
|
|
|
|
db = args.db
|
|
|
|
conn = sqlite3.connect(db)
|
|
topics = get_topics(conn)
|
|
|
|
index_page = html.Div(sum([[
|
|
dcc.Link(f'Go to {page}', href=f'/{page}'),
|
|
html.Br()] for page in topics], []))
|
|
|
|
app.run_server(debug=args.debug, host='0.0.0.0')
|