mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-11-02 03:37:55 +00:00
lf_report.py : added support of moving graph.png files to results directory
Signed-off-by: Chuck SmileyRekiere <chuck.smileyrekiere@candelatech.com>
This commit is contained in:
@@ -34,9 +34,10 @@ import pdfkit
|
||||
# base report class
|
||||
class lf_report():
|
||||
def __init__(self,
|
||||
_path = "/home/lanforge/report-data", # /home/lanforge/report-data , /home/lanforge/html-reports/ , /home/user/git/lanforge-scripts/report-data
|
||||
#_path the report directory under which the report directories will be created.
|
||||
_path = "/home/lanforge/report-data",
|
||||
_alt_path = "",
|
||||
_output_format = 'html', # pass in on the write functionality
|
||||
_output_format = 'html', # pass in on the write functionality, current not used
|
||||
_dataframe="",
|
||||
_title="LANForge Test Run Heading",
|
||||
_table_title="LANForge Table Heading",
|
||||
@@ -46,14 +47,16 @@ class lf_report():
|
||||
_date="",
|
||||
_output_html="outfile.html",
|
||||
_output_pdf="outfile.pdf",
|
||||
_path_date_time=""):
|
||||
_path_date_time=""): # this is where the final report is placed.
|
||||
#other report paths,
|
||||
|
||||
# _path is where the directory with the data time will be created
|
||||
if _path == "local" or _path == "here":
|
||||
self.path = os.path.abspath(__file__)
|
||||
print("path set to file path: {}".format(self.path))
|
||||
elif _alt_path != "":
|
||||
self.path = _alt_path
|
||||
print("path set to alt path: {}".format(self.path))
|
||||
else:
|
||||
self.path = _path
|
||||
print("path set: {}".format(self.path))
|
||||
@@ -91,29 +94,27 @@ class lf_report():
|
||||
self.copy_banner()
|
||||
self.copy_logo()
|
||||
|
||||
|
||||
|
||||
def copy_banner(self):
|
||||
src_file = str(self.current_path)+'/'+str(self.banner_directory)+'/'+str(self.banner_file_name)
|
||||
dst_file = str(self.path_date_time)+'/'+ str(self.banner_file_name)
|
||||
print("src_file: {}".format(src_file))
|
||||
print("dst_file: {}".format(dst_file))
|
||||
shutil.copy(src_file,dst_file)
|
||||
|
||||
def copy_banner(self):
|
||||
banner_src_file = str(self.current_path)+'/'+str(self.banner_directory)+'/'+str(self.banner_file_name)
|
||||
banner_dst_file = str(self.path_date_time)+'/'+ str(self.banner_file_name)
|
||||
print("banner src_file: {}".format(banner_src_file))
|
||||
print("dst_file: {}".format(banner_dst_file))
|
||||
#print("banner src_file: {}".format(banner_src_file))
|
||||
#print("dst_file: {}".format(banner_dst_file))
|
||||
shutil.copy(banner_src_file,banner_dst_file)
|
||||
|
||||
def copy_logo(self):
|
||||
logo_src_file = str(self.current_path)+'/'+str(self.logo_directory)+'/'+str(self.logo_file_name)
|
||||
logo_dst_file = str(self.path_date_time)+'/'+ str(self.logo_file_name)
|
||||
print("src_file: {}".format(logo_src_file))
|
||||
print("dst_file: {}".format(logo_dst_file))
|
||||
#print("logo_src_file: {}".format(logo_src_file))
|
||||
#print("logo_dst_file: {}".format(logo_dst_file))
|
||||
shutil.copy(logo_src_file,logo_dst_file)
|
||||
|
||||
def move_graph_image(self,):
|
||||
graph_src_file = str(self.graph_image)
|
||||
graph_dst_file = str(self.path_date_time)+'/'+ str(self.graph_image)
|
||||
print("graph_src_file: {}".format(graph_src_file))
|
||||
print("graph_dst_file: {}".format(graph_dst_file))
|
||||
shutil.move(graph_src_file,graph_dst_file)
|
||||
|
||||
|
||||
def set_path(self,_path):
|
||||
self.path = _path
|
||||
@@ -132,16 +133,17 @@ class lf_report():
|
||||
def build_date_time_directory(self):
|
||||
if self.date_time_directory == "":
|
||||
self.set_date_time_directory()
|
||||
try:
|
||||
self.path_date_time = os.path.join(self.path, self.date_time_directory)
|
||||
print("path_date_time {}".format(self.path_date_time))
|
||||
#try:
|
||||
self.path_date_time = os.path.join(self.path, self.date_time_directory)
|
||||
print("path_date_time {}".format(self.path_date_time))
|
||||
try:
|
||||
if not os.path.exists(self.path_date_time):
|
||||
os.mkdir(self.path_date_time)
|
||||
except:
|
||||
self.path_date_time = os.path.join(self.current_path, self.date_time_directory)
|
||||
if not os.path.exists(self.path_date_time):
|
||||
os.mkdir(self.path_date_time)
|
||||
#print("report path : {}".format(self.path_date_time))
|
||||
print("report path : {}".format(self.path_date_time))
|
||||
|
||||
def set_title(self,_title):
|
||||
self.title = _title
|
||||
@@ -170,10 +172,13 @@ class lf_report():
|
||||
|
||||
def get_path(self):
|
||||
return self.path
|
||||
|
||||
# get_path_date_time, get_report_path and need to be the same ()
|
||||
def get_path_date_time(self):
|
||||
return self.path_date_time
|
||||
|
||||
def get_report_path(self):
|
||||
return self.path_date_time
|
||||
|
||||
def write_html(self):
|
||||
self.write_output_html = str(self.path_date_time)+'/'+ str(self.output_html)
|
||||
print("write_output_html: {}".format(self.write_output_html))
|
||||
|
||||
@@ -54,6 +54,7 @@ if __name__ == "__main__":
|
||||
|
||||
report.set_graph_title("Graph Title")
|
||||
report.build_graph_title()
|
||||
|
||||
graph = lf_bar_graph(_data_set=dataset,
|
||||
_xaxis_name="stations",
|
||||
_yaxis_name="Throughput 2 (Mbps)",
|
||||
@@ -69,6 +70,8 @@ if __name__ == "__main__":
|
||||
print("graph name {}".format(graph_png))
|
||||
|
||||
report.set_graph_image(graph_png)
|
||||
# need to move the graph image to the results
|
||||
report.move_graph_image()
|
||||
|
||||
report.build_graph()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user