Update app.py

app.py

@@ -1,8 +1,10 @@
-from flask import Flask, request, render_template, redirect, url_for, flash, jsonify, abort
+from flask import Flask, request, render_template, redirect, url_for, flash, jsonify, abort, make_response
 from flask_sqlalchemy import SQLAlchemy
 from datetime import datetime, timedelta, timezone
 from apscheduler.schedulers.background import BackgroundScheduler
 import markdown
+import uuid  # Importation du module UUID
+import random # Importation du module random
 
 app = Flask(__name__)
 app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///forum.db'
@@ -40,6 +42,36 @@ class Message(db.Model):
     vote_count = db.Column(db.Integer, default=0)
     reports = db.Column(db.Integer, default=0)
     removed = db.Column(db.Boolean, default=False)
+    user_id = db.Column(db.String(36), nullable=False)  # Ajout de la colonne user_id
+
+
+
+# -------------------------------
+# Fonction pour obtenir ou créer l'ID utilisateur
+# -------------------------------
+def get_user_id():
+    user_id = request.cookies.get('user_id')
+    if not user_id:
+        user_id = str(uuid.uuid4())  # Génère un nouvel UUID
+    return user_id
+
+# -------------------------------
+# Fonction pour obtenir un nom d'utilisateur aléatoire
+# -------------------------------
+def generate_username(user_id):
+    # Liste de couleurs
+    colors = ["Red", "Green", "Blue", "Yellow", "Purple", "Orange", "Pink", "Brown", "Gray", "Cyan"]
+    # Liste d'animaux
+    animals = ["Lion", "Tiger", "Elephant", "Panda", "Monkey", "Giraffe", "Zebra", "Wolf", "Fox", "Bear"]
+
+    # Utilisation de l'UUID pour générer des index "aléatoires" mais constants pour un même utilisateur
+    random.seed(user_id)
+    color = random.choice(colors)
+    animal = random.choice(animals)
+    random.seed() # Reinitialiser le seed pour le reste
+
+    return f"{color} {animal}"
+
 
 # -------------------------------
 # Routes de base et fonctionnalités
@@ -49,11 +81,18 @@ class Message(db.Model):
 @app.route('/')
 def index():
     threads = Thread.query.order_by(Thread.timestamp.desc()).all()
-    return render_template('index.html', threads=threads)
+    user_id = get_user_id()  # Obtenir l'ID utilisateur
+    username = generate_username(user_id) # Obtenir un nom d'utilisateur
+    resp = make_response(render_template('index.html', threads=threads, username=username))  # Créer une réponse
+    resp.set_cookie('user_id', user_id, max_age=60*60*24*365*2)  # Définir le cookie, expire dans 2 ans
+    return resp
 
 # Création d'un nouveau fil de discussion
 @app.route('/new_thread', methods=['GET', 'POST'])
 def new_thread():
+    user_id = get_user_id()  # Obtenir l'ID utilisateur
+    username = generate_username(user_id) # Obtenir un nom d'utilisateur
+
     if request.method == 'POST':
         title = request.form.get('title', '').strip()
         content = request.form.get('content', '').strip()
@@ -64,28 +103,36 @@ def new_thread():
         thread = Thread(title=title)
         db.session.add(thread)
         db.session.commit()
-        # Message initial
-        message = Message(thread_id=thread.id, content=content)
+        # Message initial (avec user_id)
+        message = Message(thread_id=thread.id, content=content, user_id=user_id)
         db.session.add(message)
         db.session.commit()
         flash("Fil de discussion créé.", "success")
         return redirect(url_for('thread', thread_id=thread.id))
-    return render_template('new_thread.html')
+
+    resp = make_response(render_template('new_thread.html', username=username))  # Créer une réponse
+    resp.set_cookie('user_id', user_id, max_age=60*60*24*365*2) # Définir le cookie
+    return resp
 
 # Visualisation d'un fil et réponse
 @app.route('/thread/<int:thread_id>', methods=['GET', 'POST'])
 def thread(thread_id):
     thread_obj = Thread.query.get_or_404(thread_id)
+    user_id = get_user_id()  # Obtenir l'ID utilisateur
+    username = generate_username(user_id) # Obtenir un nom d'utilisateur
+
     if request.method == 'POST':
         content = request.form.get('content', '').strip()
         if not content:
             flash("Le contenu du message ne peut pas être vide.", "error")
             return redirect(url_for('thread', thread_id=thread_id))
-        message = Message(thread_id=thread_id, content=content)
+        # Création du message (avec user_id)
+        message = Message(thread_id=thread_id, content=content, user_id=user_id)
         db.session.add(message)
         db.session.commit()
         flash("Réponse postée.", "success")
         return redirect(url_for('thread', thread_id=thread_id))
+
     # Afficher uniquement les messages de moins de 72h et non supprimés
     expiration_threshold = datetime.now(timezone.utc) - timedelta(hours=72)
     messages = Message.query.filter(
@@ -93,7 +140,16 @@ def thread(thread_id):
         Message.timestamp >= expiration_threshold,
         Message.removed == False
     ).order_by(Message.timestamp.asc()).all()
-    return render_template('thread.html', thread=thread_obj, messages=messages)
+
+    # Préparer un dictionnaire pour mapper les user_id aux noms d'utilisateur
+    user_id_to_username = {}
+    for msg in messages:
+        if msg.user_id not in user_id_to_username:
+             user_id_to_username[msg.user_id] = generate_username(msg.user_id)
+
+    resp = make_response(render_template('thread.html', thread=thread_obj, messages=messages, user_id_to_username=user_id_to_username, username=username)) # Créer une réponse avec le username
+    resp.set_cookie('user_id', user_id, max_age=60*60*24*365*2)  # Définir le cookie
+    return resp
 
 # Système de vote (upvote/downvote)
 @app.route('/vote/<int:message_id>/<action>', methods=['POST'])
@@ -133,6 +189,7 @@ def search():
         flash("Veuillez entrer un terme de recherche.", "error")
         return redirect(url_for('index'))
     threads = Thread.query.filter(Thread.title.ilike(f'%{query}%')).all()
+    # Modification pour inclure l'ID de l'utilisateur dans la recherche des messages
     messages = Message.query.filter(Message.content.ilike(f'%{query}%')).all()
     return render_template('search.html', query=query, threads=threads, messages=messages)
 
@@ -177,4 +234,4 @@ scheduler.start()
 if __name__ == '__main__':
     with app.app_context():
         db.create_all()
-    app.run(debug=True)
+    app.run(debug=True)