haldis/app/views/general.py

"Script to generate the general views of Haldis"
import os
from datetime import datetime, timedelta

import yaml
from typing import Optional

from flask import Flask, render_template, make_response
from flask import request, jsonify
from flask import Blueprint, abort
from flask import current_app as app
from flask import send_from_directory, url_for
from flask_login import login_required

from utils import first
from hlds.definitions import location_definitions
from hlds.models import Location
from models import Order

# import views
from views.order import get_orders

import json
from flask import jsonify

general_bp = Blueprint("general_bp", __name__)


@general_bp.route("/")
def home() -> str:
    "Generate the home view"
    prev_day = datetime.now() - timedelta(days=1)
    recently_closed = get_orders(
        ((Order.stoptime > prev_day) & (Order.stoptime < datetime.now()))
    )
    return render_template(
        "home.html", orders=get_orders(), recently_closed=recently_closed
    )


def get_css_dict(css_path):
    "Generate the dictionary with all the currently available themes and their names"
    themes_dict = dict()

    # Open the YAML file with all the themes.
    path = os.path.join(app.root_path, "views/themes.yml")
    with open(path, "r") as stream:
        data = yaml.safe_load(stream)
    # Build a dictionary from the YAML file with all the themes and their attributes.
    themes = {}
    for item in data:
        key = list(item.keys())[0]
        themes[key] = item[key]

    # Get the current date.
    current_date = datetime.now()
    current_year = current_date.year

    # Check each theme in the dictionary and return the first one that is "correct"
    for key, theme in themes.items():
        if theme["type"] == "static-date":
            start_day, start_month = theme["start"].split("/")
            start_date = datetime(year=current_year, day=int(start_day), month=int(start_month))

            end_day, end_month = theme["end"].split("/")
            if int(start_month) > int(end_month):
                current_year += 1
            end_date = datetime(year=current_year, day=int(end_day), month=int(end_month))

            if start_date <= current_date <= end_date:
                path = os.path.join(app.root_path, css_path, theme["file"])
                themes_dict[key] = path
    themes_dict["darkmode"] = os.path.join(
        app.root_path, "static/css/themes/lowPerformance/darkmode.css"
    )
    themes_dict["lightmode"] = os.path.join(
        app.root_path, "static/css/themes/lowPerformance/lightmode.css"
    )

    return themes_dict


def css_list():
    "Generate the list of names of all the currently available themes"
    if request.cookies.get("performance", "") == "highPerformance":
        css_path = "static/css/themes/highPerformance/"
    else:
        css_path = "static/css/themes/lowPerformance/"
    return list(get_css_dict(css_path).keys())


@general_bp.route("/css")
def css():
    "Generate the css"
    if request.cookies.get("performance", "") == "highPerformance":
        css_path = "static/css/themes/highPerformance/"
    else:
        css_path = "static/css/themes/lowPerformance/"

    cookie_theme = request.cookies.get("theme", "")

    themes_dict = get_css_dict(css_path)

    # TODO: Fix to work with default cookie value [customTheme]
    if cookie_theme == "customTheme":
        path = f"{css_path}ligtmode.css"
    else:
        path = themes_dict[cookie_theme]

    f = open(path)
    response = make_response(f.read())
    response.headers["Content-Type"] = "text/css"
    f.close()
    return response


@general_bp.route("/map")
def map_view() -> str:
    "Generate the map view"
    return render_template("maps.html", locations=location_definitions)


@general_bp.route("/location")
def locations() -> str:
    "Generate the location view"
    return render_template("locations.html", locations=location_definitions)


@general_bp.route("/location/<location_id>")
def location(location_id) -> str:
    "Generate the location view given an id"
    loc = first(filter(lambda l: l.id == location_id, location_definitions))
    if loc is None:
        abort(404)
    return render_template("location.html", location=loc, title=loc.name)


@general_bp.route("/location/<location_id>/<dish_id>")
def location_dish(location_id, dish_id) -> str:
    loc: Optional[Location] = first(
        filter(lambda l: l.id == location_id, location_definitions)
    )
    if loc is None:
        abort(404)
    dish = loc.dish_by_id(dish_id)
    if dish is None:
        abort(404)
    return jsonify([
        {
            "type": c[0],
            "id": c[1].id,
            "name": c[1].name,
            "description": c[1].description,
            "options": [
                {
                    "id": o.id,
                    "name": o.name,
                    "description": o.description,
                    "price": o.price,
                    "tags": o.tags,
                }
                for o in c[1].options
            ],
        }
        for c in dish.choices
    ])


@general_bp.route("/about/")
def about() -> str:
    "Generate the about view"
    return render_template("about.html")


@general_bp.route("/profile/")
@login_required
def profile() -> str:
    "Generate the profile view"
    return render_template("profile.html", themes_list=css_list())


@general_bp.route("/favicon.ico")
def favicon() -> str:
    "Generate the favicon"
    # pylint: disable=R1705
    if not get_orders((Order.stoptime > datetime.now())):
        return send_from_directory(
            os.path.join(app.root_path, "static"),
            "favicon.ico",
            mimetype="image/x-icon",
        )
    else:
        return send_from_directory(
            os.path.join(app.root_path, "static"),
            "favicon_orange.ico",
            mimetype="image/x-icon",
        )