mirror of
https://github.com/partitioncloud/partitioncloud-server.git
synced 2025-01-23 17:26:26 +01:00
133 lines
3.7 KiB
Python
133 lines
3.7 KiB
Python
#!/usr/bin/python3
|
|
"""
|
|
Main file
|
|
"""
|
|
import os
|
|
import datetime
|
|
import subprocess
|
|
|
|
from flask import Flask, g, redirect, render_template, request, send_file, flash, session, abort
|
|
from werkzeug.security import generate_password_hash
|
|
|
|
from .modules.utils import User, Album, get_all_albums
|
|
from .modules import albums, auth, partition, admin, groupe
|
|
from .modules.auth import admin_required, login_required
|
|
from .modules.db import get_db
|
|
|
|
app = Flask(__name__)
|
|
|
|
app.config.from_mapping(
|
|
DATABASE=os.path.join(app.instance_path, f"{__name__}.sqlite"),
|
|
)
|
|
app.config.from_object('default_config')
|
|
if os.path.exists("instance/config.py"):
|
|
app.config.from_object('instance.config')
|
|
else:
|
|
print("[WARNING] Using default config")
|
|
|
|
app.register_blueprint(auth.bp)
|
|
app.register_blueprint(admin.bp)
|
|
app.register_blueprint(groupe.bp)
|
|
app.register_blueprint(albums.bp)
|
|
app.register_blueprint(partition.bp)
|
|
|
|
|
|
try:
|
|
result = subprocess.run(["git", "describe", "--tags"], stdout=subprocess.PIPE, check=True)
|
|
__version__ = result.stdout.decode('utf8')
|
|
except (FileNotFoundError, subprocess.CalledProcessError):
|
|
# In case git not found or any platform specific weird error
|
|
__version__ = "unknown"
|
|
|
|
|
|
@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
|
|
"""
|
|
current_user = User(user_id=session.get("user_id"))
|
|
|
|
if request.method == "POST":
|
|
username = request.form["username"]
|
|
password = request.form["password"]
|
|
album_uuid = request.form["album_uuid"]
|
|
|
|
error = auth.create_user(username, password)
|
|
|
|
if error is None:
|
|
# Success, go to the login page.
|
|
user = User(name=username)
|
|
try:
|
|
if album_uuid != "":
|
|
user.join_album(album_uuid)
|
|
flash(f"Utilisateur {username} créé")
|
|
return redirect("/albums")
|
|
except LookupError:
|
|
flash(f"Cet album n'existe pas. L'utilisateur {username} a été créé")
|
|
return redirect("/albums")
|
|
|
|
flash(error)
|
|
return render_template("auth/register.html", albums=get_all_albums(), user=current_user)
|
|
|
|
|
|
@app.route("/static/search-thumbnails/<uuid>.jpg")
|
|
@login_required
|
|
def search_thumbnail(uuid):
|
|
"""
|
|
Renvoie l'apercu d'un résultat de recherche
|
|
"""
|
|
db = get_db()
|
|
part = db.execute(
|
|
"""
|
|
SELECT uuid, url FROM search_results
|
|
WHERE uuid = ?
|
|
""",
|
|
(uuid,)
|
|
).fetchone()
|
|
|
|
if part is None:
|
|
abort(404)
|
|
if not os.path.exists(os.path.join(app.static_folder, "search-thumbnails", f"{uuid}.jpg")):
|
|
os.system(
|
|
f'/usr/bin/convert -thumbnail\
|
|
"178^>" -background white -alpha \
|
|
remove -crop 178x178+0+0 \
|
|
partitioncloud/search-partitions/{uuid}.pdf[0] \
|
|
partitioncloud/static/search-thumbnails/{uuid}.jpg'
|
|
)
|
|
|
|
return send_file(os.path.join(app.static_folder, "search-thumbnails", f"{uuid}.jpg"))
|
|
|
|
|
|
@app.before_request
|
|
def before_request():
|
|
"""Set cookie max age to 31 days"""
|
|
session.permanent = True
|
|
app.permanent_session_lifetime = datetime.timedelta(days=int(app.config["MAX_AGE"]))
|
|
|
|
@app.context_processor
|
|
def inject_default_variables():
|
|
"""Inject the version number in the template variables"""
|
|
if __version__ == "unknown":
|
|
return {"version": ''}
|
|
return {"version": __version__}
|
|
|
|
|
|
@app.after_request
|
|
def after_request(response):
|
|
"""Automatically close db after each request"""
|
|
if ('db' in g) and (g.db is not None):
|
|
g.db.close()
|
|
return response
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host="0.0.0.0")
|