Paperwall/web/app.py

"""Flask app: status/preview page + one settings form per screen + rotation settings."""

import requests
from flask import Flask, redirect, render_template, request, url_for
from PIL import Image
from werkzeug.utils import secure_filename

from screens.picture import UPLOAD_DIR

ALLOWED_IMAGE_EXTENSIONS = {"png", "jpg", "jpeg", "gif", "bmp", "webp"}


def create_app(config, screens: dict, scheduler) -> Flask:
    app = Flask(__name__)
    app.config["MAX_CONTENT_LENGTH"] = 20 * 1024 * 1024  # 20 MB, generous for a photo

    @app.route("/")
    def index():
        status = scheduler.get_status()
        return render_template(
            "index.html",
            status=status,
            screens=screens,
            rotation=config.get_rotation(),
            navidrome_priority=config.get_navidrome_priority(),
        )

    @app.route("/actions/refresh", methods=["POST"])
    def refresh_now():
        scheduler.request_manual_refresh()
        return redirect(url_for("index"))

    @app.route("/actions/switch/<screen_id>", methods=["POST"])
    def switch_now(screen_id):
        if screen_id in screens:
            scheduler.request_switch(screen_id)
        return redirect(url_for("index"))

    @app.route("/settings/navidrome", methods=["GET", "POST"])
    def settings_navidrome():
        if request.method == "POST":
            config.update_screen_settings("navidrome", {
                "url": request.form.get("url", "").strip(),
                "username": request.form.get("username", "").strip(),
                "password": request.form.get("password", ""),
                "filter_user": request.form.get("filter_user", "").strip(),
                "show_progress": "show_progress" in request.form,
                "show_genre": "show_genre" in request.form,
                "show_source": "show_source" in request.form,
                "boost_cover": "boost_cover" in request.form,
                "grayscale_cover": "grayscale_cover" in request.form,
            })
            return redirect(url_for("settings_navidrome", saved=1))
        return render_template("settings_navidrome.html", settings=config.get_screen_settings("navidrome"))

    @app.route("/settings/weather", methods=["GET", "POST"])
    def settings_weather():
        error = None
        if request.method == "POST":
            city = request.form.get("city", "").strip()
            lat_raw = request.form.get("latitude", "").strip()
            lon_raw = request.form.get("longitude", "").strip()

            if city:
                try:
                    result = screens["weather"].geocode(city)
                except requests.exceptions.RequestException:
                    result = None
                if result is None:
                    error = f'Could not find "{city}" -- try a different spelling, or fill in coordinates manually below instead.'
                else:
                    config.update_screen_settings("weather", {
                        "city": result["label"],
                        "latitude": result["latitude"],
                        "longitude": result["longitude"],
                    })
                    return redirect(url_for("settings_weather", saved=1))
            else:
                config.update_screen_settings("weather", {
                    "city": "",
                    "latitude": float(lat_raw) if lat_raw else None,
                    "longitude": float(lon_raw) if lon_raw else None,
                })
                return redirect(url_for("settings_weather", saved=1))

        return render_template("settings_weather.html", settings=config.get_screen_settings("weather"), error=error)

    @app.route("/settings/picture", methods=["GET", "POST"])
    def settings_picture():
        error = None
        if request.method == "POST":
            url = request.form.get("image_url", "").strip()
            common_updates = {
                "fit": request.form.get("fit", "cover"),
                "boost_contrast": "boost_contrast" in request.form,
                "grayscale": "grayscale" in request.form,
            }
            uploaded = request.files.get("image")

            if uploaded and uploaded.filename:
                filename = secure_filename(uploaded.filename)
                ext = filename.rsplit(".", 1)[-1].lower() if "." in filename else ""
                if ext not in ALLOWED_IMAGE_EXTENSIONS:
                    error = "Unsupported file type -- use PNG, JPG, GIF, BMP, or WebP."
                else:
                    UPLOAD_DIR.mkdir(parents=True, exist_ok=True)
                    # Only one picture is ever active -- clear any previous
                    # upload first so a different extension doesn't leave an
                    # orphaned file behind.
                    for old in UPLOAD_DIR.glob("current.*"):
                        old.unlink()
                    dest = UPLOAD_DIR / f"current.{ext}"
                    uploaded.save(dest)
                    try:
                        Image.open(dest).verify()
                    except Exception:
                        dest.unlink(missing_ok=True)
                        error = "That file doesn't look like a valid image."
                    else:
                        config.update_screen_settings("picture", {
                            "mode": "upload",
                            "uploaded_filename": dest.name,
                            **common_updates,
                        })
                        return redirect(url_for("settings_picture", saved=1))
            elif url:
                config.update_screen_settings("picture", {"mode": "url", "image_url": url, **common_updates})
                return redirect(url_for("settings_picture", saved=1))
            else:
                # Neither a new upload nor a URL change -- just update fit/boost/grayscale.
                config.update_screen_settings("picture", common_updates)
                return redirect(url_for("settings_picture", saved=1))

        return render_template("settings_picture.html", settings=config.get_screen_settings("picture"), error=error)

    @app.route("/settings/rotation", methods=["GET", "POST"])
    def settings_rotation():
        if request.method == "POST":
            raw = request.form.get("screens", "")
            selected = [s.strip() for s in raw.split(",") if s.strip() in screens]
            interval_minutes = int(request.form.get("interval_minutes", 5) or 5)
            enabled = "rotation_enabled" in request.form
            config.set_rotation(selected, interval_minutes, enabled)
            config.set_navidrome_priority("navidrome_priority" in request.form)
            return redirect(url_for("settings_rotation", saved=1))
        return render_template(
            "settings_rotation.html",
            screens=screens,
            rotation=config.get_rotation(),
            navidrome_priority=config.get_navidrome_priority(),
        )

    return app