loustique-home/flask/main.py

from flask import Flask, render_template, request, jsonify
import requests
from flask_talisman import Talisman
from led import led
import os
import threading
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 lumieres import SystemeLumieres
from board1main import call_board1
import alarme





@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})
dernier_badge_scanne = None

@app.route("/rfid-scan", methods=["POST"])
def rfid_scan():
    global dernier_badge_scanne
    data = request.get_json()
    badge_id = 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:
        return jsonify({"success": False})

@app.route("/check-rfid-login", methods=["GET"])
def check_rfid_login():
    global dernier_badge_scanne
    global current_user
    if dernier_badge_scanne:
        user = dernier_badge_scanne
        current_user = user 
        dernier_badge_scanne = None 
        
        return jsonify({"success": True, "username": user})
    
    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})

@app.route("/alarme_status")
def get_alarme_status_info(): # Nom différent de l'import 'alarme'
    try:
        # On accède à la variable du fichier
        statut = alarme.etat_alarme 
        return jsonify({"success": True, "status": statut})
    except Exception as e:
        return jsonify({"success": False, "message": str(e)}), 500 

@app.route("/api/<action>", methods=["GET"])
def relais_pi2(action):
    url_pi2 = f"https://pi32.local:8000/{action}"
    try:
        reponse = requests.get(url_pi2, timeout=5, verify=False)
        if not reponse.ok:
            return jsonify({"success": False, "message": "Erreur Pi 2"}), reponse.status_code
        
        return jsonify(reponse.json())
            
    except Exception as e:
        # On ne garde que l'erreur si vraiment ça plante
        print(f"[RELAIS] ERREUR : {e}")
        return jsonify({"success": False, "message": str(e)}), 500


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')
        )
    )