mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-11-04 12:48:00 +00:00
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
import matplotlib.pyplot as plt
|
|
import base64
|
|
from io import BytesIO
|
|
import os
|
|
import datetime
|
|
|
|
time = (datetime.datetime.now().time())
|
|
name = "WIFI_Diag"
|
|
name += str(time)
|
|
name += "/"
|
|
|
|
name = name.replace(":","_")
|
|
if not os.path.exists(str(name)):
|
|
os.makedirs(str(name))
|
|
|
|
|
|
class Plot:
|
|
def __init__(self):
|
|
# print("In Plot")
|
|
pass
|
|
|
|
def bar(self, datax="", datay=" ", title="Temp", xaxis="yaxis", yaxis="xaxis",figname="temp"):
|
|
# fig = plt.figure()
|
|
self.tmpfile = BytesIO()
|
|
|
|
plt.xlabel(xaxis)
|
|
plt.ylabel(yaxis)
|
|
|
|
plt.title(title)
|
|
plt.xticks(rotation=90)
|
|
plt.rc('xtick', labelsize=8)
|
|
plt.rc('ytick', labelsize=8)
|
|
plt.bar(datax, datay)
|
|
path = plt.savefig(str(name)+str(figname)+".png",bbox_inches='tight')
|
|
# plt.savefig(str(title)+".png")
|
|
plt.savefig(self.tmpfile, format='png')
|
|
self.encoded = base64.b64encode(self.tmpfile.getvalue()).decode('utf-8')
|
|
# print("self.encoded",path)
|
|
plt.clf()
|
|
return str(name)+str(figname)+".png"
|
|
|
|
|