Files
loustique-home/flask/main.py
2026-04-01 22:23:25 +02:00

156 lines
4.8 KiB
Python

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 alarme import SystemeAlarme
from lumieres 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})
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("/api/<action>", methods=["GET"])
def relais_pi2(action):
url_pi2 = f"https://pi32.local:8000/{action}"
print(f"\n[RELAIS] 1. Tentative de contact avec le Pi 2 : {url_pi2}")
try:
reponse = requests.get(url_pi2, timeout=5, verify=False)
print(f"[RELAIS] 2. Code HTTP reçu du Pi 2 : {reponse.status_code}")
print(f"[RELAIS] 3. Texte brut reçu du Pi 2 : {reponse.text}")
if not reponse.ok:
return jsonify({
"success": False,
"message": f"Le Pi 2 a refusé la requête (Code {reponse.status_code})"
}), reponse.status_code
# Si tout va bien, on tente d'extraire le JSON
try:
data = reponse.json()
return jsonify(data)
except ValueError:
print("[RELAIS] 4. ERREUR : Le Pi 2 n'a pas renvoyé de JSON valide.")
return jsonify({"success": False, "message": "Réponse invalide du Pi 2"}), 502
except Exception as e:
print(f"[RELAIS] ERREUR CRITIQUE : Impossible de joindre le Pi 2. Raison : {e}")
return jsonify({"success": False, "message": f"Erreur de connexion : {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')
)
)