service_netatmo/netatmo_service/app.py

from __future__ import annotations

import logging
import os

from flask import Flask, Response, abort, jsonify, render_template, url_for

from .collector import Collector
from .config import Config
from .netatmo import NetatmoClient
from .rrd import ALIASES, PERIODS, RRD_ERROR, RRDStore, SCHEMAS


class PrefixMiddleware:
    """Mount a WSGI application below a fixed URL path."""

    def __init__(self, application, prefix: str):
        self.application = application
        self.prefix = prefix

    def __call__(self, environ, start_response):
        path = environ.get("PATH_INFO", "")
        if path == self.prefix or path.startswith(f"{self.prefix}/"):
            environ["SCRIPT_NAME"] = environ.get("SCRIPT_NAME", "") + self.prefix
            environ["PATH_INFO"] = path[len(self.prefix):] or "/"
            return self.application(environ, start_response)
        start_response("404 Not Found", [("Content-Type", "text/plain; charset=utf-8")])
        return [b"Not Found\n"]


def create_app(test_config: dict | None = None) -> Flask:
    app = Flask(__name__)
    app.config.from_object(Config)
    if test_config:
        app.config.update(test_config)

    prefix = app.config["URL_PREFIX"]
    app.config["APPLICATION_ROOT"] = prefix or "/"
    if prefix:
        app.wsgi_app = PrefixMiddleware(app.wsgi_app, prefix)

    logging.basicConfig(level=app.config.get("LOG_LEVEL", "INFO"))
    store = app.config.get("RRD_STORE") or RRDStore(
        app.config["RRD_FOLDER"], app.config["GRAPH_WIDTH"], app.config["GRAPH_HEIGHT"]
    )
    store.ensure_all()
    client = app.config.get("NETATMO_CLIENT") or NetatmoClient(app.config)
    names = {key: app.config[f"MODULE_{key.upper()}"] for key in SCHEMAS}
    collector = Collector(client, store, names, app.config["POLL_INTERVAL"])
    app.extensions["rrd_store"] = store
    app.extensions["netatmo_collector"] = collector

    # Flask's debug reloader imports twice. Only its serving child starts a poller.
    should_start = app.config["START_COLLECTOR"] and client.configured
    if should_start and (not app.debug or os.environ.get("WERKZEUG_RUN_MAIN") == "true"):
        collector.start()
    elif app.config["START_COLLECTOR"] and not client.configured:
        app.logger.warning("Collector disabled: configure NETATMO_ACCESS_TOKEN or OAuth refresh credentials")

    @app.get("/")
    def dashboard():
        return render_template(
            "dashboard.html",
            rrd_names=list(SCHEMAS),
            periods=list(PERIODS),
            graph_url_template=url_for("graph", rrd_name="__name__", period="__period__"),
        )

    @app.get("/health")
    def health():
        return jsonify(status="ok", collector_configured=client.configured)

    @app.get("/last/<rrd_name>/<data_point>")
    def last(rrd_name: str, data_point: str):
        try:
            return jsonify(store.last(rrd_name.lower(), data_point.lower()))
        except KeyError:
            abort(404, description="Unknown RRD or data point")
        except (OSError, RuntimeError, RRD_ERROR):
            app.logger.exception("Could not read RRD")
            abort(503, description="RRD data is unavailable")

    @app.get("/graph/<rrd_name>/<period>")
    def graph(rrd_name: str, period: str):
        try:
            image = store.graph(rrd_name.lower(), ALIASES.get(period.lower(), period.lower()))
            return Response(image, mimetype="image/png", headers={"Cache-Control": "no-cache, max-age=0"})
        except KeyError:
            abort(404, description="Unknown RRD or period")
        except (OSError, RuntimeError, RRD_ERROR):
            app.logger.exception("Could not render graph")
            abort(503, description="RRD graph is unavailable")

    # Compatibility with the compact form requested as /graph/[rrdname][period].
    @app.get("/graph/<rrd_and_period>")
    def compact_graph(rrd_and_period: str):
        for name in SCHEMAS:
            if rrd_and_period.startswith(name):
                return graph(name, rrd_and_period[len(name):].lstrip("-_"))
        abort(404, description="Unknown RRD or period")

    @app.cli.command("collect-now")
    def collect_now():
        """Fetch Netatmo and update every RRD immediately."""
        count = collector.collect_once()
        print(f"Updated {count} RRDs")

    return app