Skip to content
Snippets Groups Projects
start_client.py 618 B
Newer Older
import os
from flask import Flask, request, Response
Xin-Hao Zhu's avatar
Xin-Hao Zhu committed
from multiprocessing import Process
from generate_html import generate_html
from utils import TOKEN, CLIENT_PORT, CLIENT_PATH, TEMPLATE_DIR
Xin-Hao Zhu's avatar
Xin-Hao Zhu committed

app = Flask(__name__)

Xin-Hao Zhu's avatar
Xin-Hao Zhu committed
@app.route(CLIENT_PATH)
Xin-Hao Zhu's avatar
Xin-Hao Zhu committed
def get_stat():
    if request.args.get("token") != TOKEN:
        return "Unauthorized", 401
    with open(os.path.join(TEMPLATE_DIR, "stat.html"), "r", encoding="utf-8") as f:
        html = f.read()
    return Response(html, mimetype="text/html")

Xin-Hao Zhu's avatar
Xin-Hao Zhu committed

if __name__ == "__main__":
    p = Process(target=generate_html)
    p.start()
Xin-Hao Zhu's avatar
Xin-Hao Zhu committed
    app.run(host="0.0.0.0", port=CLIENT_PORT)