48 lines
1.8 KiB
Python
48 lines
1.8 KiB
Python
from flask_restful import Resource
|
|
from pygal.style import Style
|
|
from master import ctx
|
|
import pygal
|
|
import timeago
|
|
from math import ceil
|
|
|
|
class HistoryGraph(Resource):
|
|
def get(self, history_count):
|
|
try:
|
|
custom_style = Style(
|
|
background='transparent',
|
|
plot_background='transparent',
|
|
foreground='#6c757d',
|
|
foreground_strong='#6c757d',
|
|
foreground_subtle='#6c757d',
|
|
opacity='0.1',
|
|
opacity_hover='0.2',
|
|
transition='0ms',
|
|
colors=('#749363','#007acc'),
|
|
)
|
|
|
|
graph = pygal.Line(
|
|
stroke_style={'width': 0.4},
|
|
#show_dots=False,
|
|
show_legend=False,
|
|
fill=True,
|
|
style=custom_style,
|
|
disable_xml_declaration=True)
|
|
|
|
instance_count = [history['time'] for history in ctx.history.instance_history][-history_count:]
|
|
|
|
if len(instance_count) > 0:
|
|
graph.x_labels = [ timeago.format(instance_count[0])]
|
|
|
|
instance_counts = [history['count'] for history in ctx.history.instance_history][-history_count:]
|
|
client_counts = [history['count'] for history in ctx.history.client_history][-history_count:]
|
|
|
|
graph.add('Client Count', client_counts)
|
|
graph.add('Instance Count', instance_counts)
|
|
return { 'message' : graph.render().replace("<title>Pygal</title>", ""),
|
|
'data_points' : len(instance_count),
|
|
'instance_count' : 0 if len(instance_counts) is 0 else instance_counts[-1],
|
|
'client_count' : 0 if len(client_counts) is 0 else client_counts[-1]
|
|
}, 200
|
|
except Exception as e:
|
|
return { 'message' : str(e) }, 500
|