mirror of
https://github.com/Doctorado-ML/beflask.git
synced 2025-08-15 23:25:51 +00:00
102 lines
3.0 KiB
Python
102 lines
3.0 KiB
Python
import os
|
|
import shutil
|
|
from pathlib import Path
|
|
from flask import Blueprint, render_template, url_for, current_app
|
|
from benchmark.ResultsFiles import Benchmark
|
|
from flask_login import current_user, login_required
|
|
|
|
interactive = Blueprint("interactive", __name__, template_folder="templates")
|
|
|
|
|
|
@interactive.route("/ranking")
|
|
@login_required
|
|
def ranking():
|
|
os.chdir(current_user.benchmark.folder)
|
|
return render_template("ranking.html")
|
|
|
|
|
|
@current_app.socket.on("client")
|
|
def handle_client(message):
|
|
current_app.logger.info(message)
|
|
if message.get("action") == "ReadyToRock!":
|
|
get_benchmark(
|
|
score=message.get("score"),
|
|
excel=message.get("excel", False),
|
|
html=message.get("html", False),
|
|
)
|
|
current_app.socket.emit("server", {"message": "Ready!", "percentage": 0})
|
|
|
|
|
|
def send_message(message, percentage, status="Ok", payload={}):
|
|
output = {
|
|
"message": message,
|
|
"percentage": percentage,
|
|
"status": status,
|
|
"payload": payload,
|
|
}
|
|
current_app.socket.emit("server", output)
|
|
|
|
|
|
def get_benchmark(score, excel=False, html=False):
|
|
def move_exreport():
|
|
src = os.path.join(
|
|
current_user.benchmark.folder, "exreport", "exreport_output"
|
|
)
|
|
dst = os.path.join(
|
|
current_app.static_folder, "exreport", "exreport_output"
|
|
)
|
|
shutil.copytree(src, dst, dirs_exist_ok=True)
|
|
|
|
def progress(step):
|
|
values = [0, 20, 40, 60, 80]
|
|
if excel:
|
|
values = [0, 40, 60, 80, 100]
|
|
return values[step]
|
|
|
|
benchmark = Benchmark(score=score, visualize=html)
|
|
send_message("Start", 0)
|
|
try:
|
|
send_message("Generating ranking...", 0)
|
|
benchmark.compile_results()
|
|
send_message("Saving results...", progress(1))
|
|
benchmark.save_results()
|
|
send_message("Generating report...", progress(2))
|
|
benchmark.report(tex_output=False)
|
|
send_message("Generating exreport...", progress(3))
|
|
benchmark.exreport()
|
|
if excel:
|
|
send_message("Generating excel...", progress(4))
|
|
benchmark.excel()
|
|
except ValueError as e:
|
|
send_message(
|
|
"Error: Couldn't generate ranking. " + str(e),
|
|
percentage=100,
|
|
status="Error",
|
|
)
|
|
return
|
|
except KeyError as e:
|
|
send_message(
|
|
"Couldn't generate ranking. It seems that there are "
|
|
"partial results for some classifierss. "
|
|
f"Key not found {str(e)}",
|
|
percentage=100,
|
|
status="Error",
|
|
)
|
|
return
|
|
# copy the results to the static folder
|
|
move_exreport()
|
|
excel_payload = (
|
|
"" if not excel else str(Path(benchmark.get_excel_file_name()))
|
|
)
|
|
current_app.logger.info("excel_payload:" + excel_payload)
|
|
send_message(
|
|
"Done!",
|
|
100,
|
|
payload={
|
|
"excel": excel_payload,
|
|
"html": url_for(
|
|
"static", filename="exreport/exreport_output/report.html"
|
|
),
|
|
},
|
|
)
|