175 lines
4.6 KiB
Python
175 lines
4.6 KiB
Python
from io import BytesIO
|
|
from typing import List, Dict
|
|
|
|
import matplotlib.pyplot as plt
|
|
from matplotlib.dates import DateFormatter
|
|
from matplotlib.ticker import MultipleLocator
|
|
from reportlab.lib.pagesizes import letter, inch
|
|
from reportlab.lib.styles import (
|
|
ParagraphStyle,
|
|
TA_CENTER, # noqa - not declared in __all__
|
|
)
|
|
from reportlab.platypus import (
|
|
SimpleDocTemplate,
|
|
KeepInFrame,
|
|
Image,
|
|
Paragraph,
|
|
PageBreak,
|
|
Spacer,
|
|
)
|
|
from svglib.svglib import svg2rlg
|
|
|
|
from data import MinerData
|
|
|
|
|
|
async def generate_pdf(data: Dict[str, List[MinerData]], file_loc):
|
|
doc = SimpleDocTemplate(
|
|
file_loc,
|
|
pagesize=letter,
|
|
topMargin=0.25 * inch,
|
|
leftMargin=1 * inch,
|
|
rightMargin=1 * inch,
|
|
bottomMargin=1 * inch,
|
|
title=f"Recorded Data",
|
|
)
|
|
|
|
elements = []
|
|
i = 0
|
|
for item in data.keys():
|
|
i += 1
|
|
if not i == 1:
|
|
elements.append(PageBreak())
|
|
page_elem = await generate_page(data[item])
|
|
for elem in page_elem:
|
|
elements.append(elem)
|
|
|
|
doc.build(
|
|
elements,
|
|
)
|
|
|
|
|
|
async def generate_page(data):
|
|
title_style = ParagraphStyle(
|
|
"Title",
|
|
alignment=TA_CENTER,
|
|
fontSize=25,
|
|
spaceAfter=40,
|
|
spaceBefore=150,
|
|
fontName="Helvetica-Bold",
|
|
)
|
|
|
|
hr_graph = create_hr_graph(data)
|
|
fan_graph = create_fans_graph(data)
|
|
temp_graph = create_temp_graph(data)
|
|
title = Paragraph(data[0].ip, style=title_style)
|
|
|
|
elements = [
|
|
title,
|
|
await hr_graph,
|
|
Spacer(0, 40),
|
|
await temp_graph,
|
|
Spacer(0, 40),
|
|
await fan_graph,
|
|
]
|
|
return elements
|
|
|
|
|
|
async def create_hr_graph(data):
|
|
fig, ax = plt.subplots(figsize=(6, 2))
|
|
xpoints = []
|
|
ypoints = []
|
|
for item in data:
|
|
xpoints.append(item.datetime)
|
|
ypoints.append(item.hashrate)
|
|
for label in ax.get_xticklabels() + ax.get_yticklabels():
|
|
label.set_fontsize(6)
|
|
ax.plot(xpoints, ypoints)
|
|
ylim = max(ypoints) * 1.4
|
|
if ylim == 0:
|
|
ylim = 10
|
|
ax.set_ylim(0, ylim)
|
|
date_form = DateFormatter("%H:%M:%S")
|
|
ax.xaxis.set_major_formatter(date_form)
|
|
ax.yaxis.set_major_formatter("{x:1.1f} TH/s")
|
|
ax.set_title("Hashrate", fontsize=15)
|
|
ax.yaxis.set_major_locator(MultipleLocator(5))
|
|
ax.yaxis.set_minor_locator(MultipleLocator(1))
|
|
|
|
imgdata = BytesIO()
|
|
fig.savefig(imgdata, format="svg")
|
|
imgdata.seek(0) # rewind the data
|
|
drawing = svg2rlg(imgdata)
|
|
imgdata.close()
|
|
plt.close("all")
|
|
|
|
hr_graph = KeepInFrame(375, 375, [Image(drawing)], hAlign="CENTER")
|
|
|
|
return hr_graph
|
|
|
|
|
|
async def create_fans_graph(data):
|
|
fig, ax = plt.subplots(figsize=(6, 2))
|
|
xpoints = []
|
|
ypoints_f1 = []
|
|
ypoints_f2 = []
|
|
ypoints_f3 = []
|
|
ypoints_f4 = []
|
|
for item in data:
|
|
xpoints.append(item.datetime)
|
|
ypoints_f1.append(item.fan_1)
|
|
ypoints_f2.append(item.fan_2)
|
|
ypoints_f3.append(item.fan_3)
|
|
ypoints_f4.append(item.fan_4)
|
|
for label in ax.get_xticklabels() + ax.get_yticklabels():
|
|
label.set_fontsize(6)
|
|
for ypoints in [ypoints_f1, ypoints_f2, ypoints_f3, ypoints_f4]:
|
|
if not ypoints == [-1 for x in range(len(ypoints))]:
|
|
ax.plot(xpoints, ypoints)
|
|
ax.set_ylim(0, 10000)
|
|
date_form = DateFormatter("%H:%M:%S")
|
|
ax.xaxis.set_major_formatter(date_form)
|
|
ax.yaxis.set_major_formatter("{x:1.0f} RPM")
|
|
ax.set_title("Fans", fontsize=15)
|
|
|
|
imgdata = BytesIO()
|
|
fig.savefig(imgdata, format="svg")
|
|
imgdata.seek(0) # rewind the data
|
|
drawing = svg2rlg(imgdata)
|
|
imgdata.close()
|
|
plt.close("all")
|
|
|
|
fans_graph = KeepInFrame(375, 375, [Image(drawing)], hAlign="CENTER")
|
|
|
|
return fans_graph
|
|
|
|
|
|
async def create_temp_graph(data):
|
|
fig, ax = plt.subplots(figsize=(6, 2))
|
|
# plt.figure()
|
|
xpoints = []
|
|
ypoints = []
|
|
for item in data:
|
|
xpoints.append(item.datetime)
|
|
ypoints.append(item.temperature_avg)
|
|
for label in ax.get_xticklabels() + ax.get_yticklabels():
|
|
label.set_fontsize(6)
|
|
ax.plot(xpoints, ypoints)
|
|
ax.set_ylim(0, 130)
|
|
ax.yaxis.set_major_locator(MultipleLocator(20))
|
|
ax.yaxis.set_minor_locator(MultipleLocator(5))
|
|
date_form = DateFormatter("%H:%M:%S")
|
|
ax.xaxis.set_major_formatter(date_form)
|
|
ax.yaxis.set_major_formatter("{x:1.1f} C")
|
|
ax.set_title("Temperature", fontsize=15)
|
|
|
|
imgdata = BytesIO()
|
|
fig.savefig(imgdata, format="svg")
|
|
imgdata.seek(0) # rewind the data
|
|
drawing = svg2rlg(imgdata)
|
|
imgdata.close()
|
|
plt.close("all")
|
|
|
|
temp_graph = KeepInFrame(375, 375, [Image(drawing)], hAlign="CENTER")
|
|
|
|
return temp_graph
|