privatecoffee-website/main.py

from flask import Flask, render_template, send_from_directory
from jinja2 import TemplateNotFound

import json
import pathlib
import os

from argparse import ArgumentParser

from helpers.finances import generate_transparency_table, get_transparency_data

app = Flask(__name__)


@app.route("/assets/<path:path>")
def send_assets(path):
    return send_from_directory("assets", path)


@app.route("/", defaults={"path": "index"})
@app.route("/<path:path>.html")
def catch_all(path):
    try:
        services = json.loads(
            (pathlib.Path(__file__).parent / "services.json").read_text()
        )

        warning = None

        if app.development_mode:
            warning = render_template("prod-warning.html")

        kwargs = {
            "services": services,
            "warning": warning,
        }

        if path == "membership":
            finances = json.loads(
                (pathlib.Path(__file__).parent / "finances.json").read_text()
            )

            finances_table = generate_transparency_table(
                get_transparency_data(finances)
            )

            kwargs.update(
                {
                    "finances": finances_table,
                }
            )

        return render_template(f"{path}.html", **kwargs)
    except TemplateNotFound:
        return "404 Not Found", 404


app.development_mode = False

if os.environ.get("PRIVATECOFFEE_DEV"):
    app.development_mode = True


if __name__ == "__main__":
    parser = ArgumentParser(description="Run the private.coffee web server.")
    parser.add_argument("--port", type=int, default=9810)
    parser.add_argument("--debug", action="store_true")
    parser.add_argument("--dev", action="store_true")
    args = parser.parse_args()

    app.development_mode = args.dev or app.development_mode

    app.run(port=args.port, debug=args.debug)