partitioncloud-server/partitioncloud/__init__.py

#!/usr/bin/python3
"""
Main file
"""
import os

from flask import Flask, g, redirect, render_template, request, send_file, flash
from werkzeug.security import generate_password_hash

from . import albums, auth, partition
from .auth import admin_required
from .db import get_db

app = Flask(__name__)

app.config.from_mapping(
    # a default secret that should be overridden by instance config
    SECRET_KEY="dev",
    # store the database in the instance folder
    DATABASE=os.path.join(app.instance_path, f"{__name__}.sqlite"),
)

app.register_blueprint(auth.bp)
app.register_blueprint(albums.bp)
app.register_blueprint(partition.bp)


@app.route("/")
def home():
    """Redirect to home"""
    return redirect("/albums/")


@app.route("/add-user", methods=["GET", "POST"])
@admin_required
def add_user():
    """
    Ajouter un utilisateur en tant qu'administrateur
    """
    if request.method == "POST":
        username = request.form["username"]
        password = request.form["password"]
        db = get_db()
        error = None

        if not username:
            error = "Un nom d'utilisateur est requis."
        elif not password:
            error = "Un mot de passe est requis."

        if error is None:
            try:
                db.execute(
                    "INSERT INTO user (username, password) VALUES (?, ?)",
                    (username, generate_password_hash(password)),
                )
                db.commit()
            except db.IntegrityError:
                # The username was already taken, which caused the
                # commit to fail. Show a validation error.
                error = f"Le nom d'utilisateur {username} est déjà pris."
            else:
                # Success, go to the login page.
                flash(f"Utilisateur {username} crée")
                return redirect("/albums")

        flash(error)
    return render_template("auth/register.html")


if __name__ == "__main__":
    app.run(host="0.0.0.0")