Files
loustique-home/flask/main.py
2026-03-30 18:20:42 +02:00

122 lines
3.3 KiB
Python

from flask import Flask, render_template, request, jsonify
from flask_talisman import Talisman
from led import led
import os
import sys
import log
from add_user import add_user
import auth
import re
app = Flask(__name__)
Talisman(app, force_https=True,
content_security_policy=False )
current_user = None
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
composants = os.path.join(BASE_DIR, "composants", "byPanda")
sys.path.insert(0, composants)
from alarme import SystemeAlarme
from lumiere import SystemeLumieres
from board1main import *
@app.route("/")
def index():
return render_template("index.html")
@app.route("/login", methods=["POST"])
def login():
global current_user
data = request.get_json()
succes = auth.login(data["username"], data["password"])
if succes:
current_user = data["username"]
return jsonify({"success": True})
else:
return jsonify({"success": False})
@app.route("/dashboard")
def dashboard():
return render_template("dashboard.html")
@app.route("/led", methods=["POST"])
def call_led():
etat = SystemeLumieres.mettreAJourEtat()
if (etat == 0):
SystemeLumieres.allumerLumieres
else:
SystemeLumieres.eteindreLumieres()
return jsonify({"success": True})
@app.route("/rfid-scan", methods=["POST"])
def rfid_scan():
global dernier_badge_scanne
data = request.get_json()
badge_id = str(data.get("badge_id"))
username = auth.get_user_by_rfid(badge_id)
if username:
dernier_badge_scanne = username
return jsonify({"success": True, "username": username})
else:
# Badge inconnu dans la BDD
return jsonify({"success": False})
@app.route("/alarme",methods=["POST"])
def armer_alarme():
SystemeAlarme.armer()
return jsonify({"success": True})
@app.route("/admin")
def admin_page():
return render_template("admin.html")
@app.route("/admin/logs")
def logs_page():
return render_template("log.html")
@app.route("/admin/logs/data")
def get_logs():
try:
with open('/var/log/loustique.log', 'r') as f:
lines = f.readlines()
ansi_escape = re.compile(r'\x1b\[[0-9;]*m')
lines = [ansi_escape.sub('', line) for line in lines[-200:]]
return jsonify({"success": True, "logs": lines})
except Exception as e:
return jsonify({"success": False, "message": str(e)})
@app.route("/admin/add_user",methods=["POST"])
@app.route("/admin/add_user", methods=["POST"])
def create_user():
data = request.get_json()
succes = add_user(data["username"], data["password"], data["role"])
if succes:
return jsonify({"success": True})
else:
return jsonify({"success": False, "message": "Utilisateur déjà existant"})
@app.route("/admin/get_users")
def get_users():
users = auth.get_users()
return jsonify({"success": True, "users": users})
if __name__ == "__main__":
print("[*] Démarrage du lecteur RFID et de l'alarme en arrière-plan...")
thread_hardware = threading.Thread(target=call_board1, daemon=True)
thread_hardware.start()
app.run(
host="0.0.0.0",
port=443,
ssl_context=(
os.path.join(BASE_DIR, 'web_secu', 'ssl', 'cert.pem'),
os.path.join(BASE_DIR, 'web_secu', 'ssl', 'key.pem')
)
)