app.py

from flask import Flask, jsonify, request, redirect, render_template, session, send_from_directory, url_for
from urllib.parse import unquote
import datetime
import json
import time
import os

# Définir notre application Flask
app = Flask(__name__)

### Fonctions de base
def get_users():
    """Permet de retourner tous les utilisateurs depuis la BDD"""
    with open('db/users.json') as f: return json.load(f)
def get_messages():
    """Permet de retourner tous les messages depuis la BDD"""
    with open('db/messages.json') as f: return json.load(f)

def get_user(id_user = False, name_user = False):
    """Permet de retourner un seul utilisateur depuis la BDD"""
    users = get_users()['users']
    for x in users:
        if id_user and int(x['id']) == id_user: return x
        if name_user and str(x['pseudo']) == name_user: return x
    return None
def update_user(key, value, id_user = False, name_user = False):
    """Permet de mettre à jour un utilisateur"""
    data = get_users()
    for x in data['users']:
        if id_user and int(x['id']) == id_user or str(x['pseudo']) == name_user:
            x[key] = value
            with open('db/users.json', 'w') as f: json.dump(data, f)
            return True
    return False

def get_message(id_msg):
    """Permet de retourner un seul message depuis la BDD"""
    msgs = get_messages()
    for x in msgs:
        if int(x['id']) == id_msg: return x
    return None

def get_emojis():
    return ['😀','😂','😉','😅','😜','🤔','😡','😬','😷','👍','👎']
def get_connected_users():
    users = get_users()['users']
    connected_users = []
    for u in users:
        if 'last_action' in u:
            now = datetime.datetime.now().timestamp()
            date_action = datetime.datetime.strptime(u['last_action'], '%Y-%m-%d %H:%M:%S.%f').timestamp()
            # L'utilisateur n'est pas connecté depuis + de 60 secondes
            if int(now-date_action) <= 60: connected_users.append(u)
    return connected_users

## Routes HTML 
@app.route("/")
def accueuil():
    """Page d'accueil du site"""
    logged = False
    if 'user_id' in session and session['user_id'] is not None: logged = True

    return render_template("messenger.html", **{
            'session': logged,
            'user':get_users()['users'][int(session['user_id'])] if logged else None, 'emojis':get_emojis(),
            'nb_msg': len(get_messages()['messages']) })

@app.route("/favicon.ico")
def favicon():
    """Retourne le favicon du site"""
    return send_from_directory(os.path.join(app.root_path, 'static'),
    'favicon.ico', mimetype='image/vnd.microsoft.icon')           

## Requêtes API
@app.route("/api/auth/signin/<user>/<pw>")
def signin_user(user, pw):
    """Permet d'ajouter un utilisateur"""
    try:
        data = get_users()
        last_id = data['users'][-1]['id']+1
        data['users'].append({"id": last_id, "pseudo": str(user), "passe":str(pw)}) 
        with open('db/users.json', 'w') as f: json.dump(data, f)
        return "True"
    except Exception:
        return "False"

## Requêtes API
@app.route("/api/auth/login/<user>/<pw>")
def login_user(user, pw):
    """Permet de vérifier si un utilisateur peut se connecter"""
    try:
        data = get_users()
        for u in data['users']:
            if u['pseudo'] == str(user) and u['passe']:
                session['user_id'] = u['id']
                # On demande au server web de générer la page de messenger habituel
                return render_template("messenger.html", **{
                    'session': True, 'user':u, 'emojis':get_emojis(),
                    'nb_msg': len(get_messages()['messages']) })
        return "False"
    except Exception:
        return "False"

@app.route("/api/auth/logout")
def logout():
    """Permet de déconnecter un utilisateur"""
    try:
        if "user_id" in session:
            session['user_id'] = None
        return redirect(url_for('accueuil'))
    except Exception:
        return "False"

@app.route("/api/message/new/<message>", methods=['GET', 'POST'])
def add_messages(message):
    """Permet de retourner tous les messages depuis la BDD"""
    data = get_messages()
    user = get_users()['users'][session['user_id']]
    data['messages'].append({"id": int(data['messages'][-1]['id'])+1,
    "id_membre":user['id'], "contenu":unquote(message), "date" : str(datetime.datetime.now())})
    with open('db/messages.json', 'w') as f: json.dump(data, f)
    return data['messages'][-1]['contenu'] # Retourne le dernier message

@app.route("/api/messages")
def list_messages():
    """Permet de retourner tous les messages de la BDD"""
    try:
        data = get_messages()['messages']
        html = ""
        for msg in data:

            date = msg['date'].split(' ')[0]
            heure = str(msg['date'].split(' ')[1].split('.')[0])[:5]
            timestamp = date if not str(datetime.datetime.now()).split(' ')[0] == date else heure
            user = get_users()['users'][msg['id_membre']]
            html += f"<div class='authorbox'>{user['pseudo'].capitalize()}: <div class='msgbox'>{msg['contenu']}</div><small>{timestamp}</small></div>"
        return html
    except json.decoder.JSONDecodeError:
        print("Les messages n'ont pas pu être vérifiés.") 
        return "False"

@app.route("/api/user/status", methods=['POST'])
def mark_user_as_online():
    update_user("last_action", str(datetime.datetime.now()), id_user=session['user_id'])
    return "False"

@app.route("/api/users/online", methods=['GET'])
def list_connected_users():
    try:
        users = get_connected_users()
        if len(users) == 0: return "Aucun utilisateur n'est actuellement en ligne."
        if len(users) == 1: return f"Uniquement {users[0]['pseudo']} est en ligne actuellement."
        return f"Il y a {len(users)} utilisateurs actuellement en ligne : {', '.join([x['pseudo'] for x in users])}"
    except json.decoder.JSONDecodeError:
        print("Les utilisateurs en ligne n'ont pas pu être vérifiés.")
        return "False"

@app.route("/api/stats/messages")
def message_stats():
    try:
        return f"{len(get_messages()['messages'])} messages"
    except Exception:
        print("La quantité totale de messages n'a pas pu être relevé.")
        return "False"

if __name__ == '__main__':
    app.secret_key = "X6DGXy9jGT6z8KjM3sAY"
    app.run(debug=True)