server_logic.py

from flask import Flask, render_template, redirect, url_for, send_from_directory
from class_logic import ObjectStorage
import os


website = Flask(__name__)
storage = ObjectStorage()
storage.load_all()


@website.route("/", strict_slashes=False)
def homepage():
    return render_template("homepage.html")


@website.route("/warframes/<path:path>", strict_slashes=False)
def warframes(path):
    print("Warframes: User tried to go to /warframes/{}".format(path))
    if path in storage.warframe_list.file_names:
        return render_template("warframe_info.html", warframe_data=storage.warframe_list.warframes[path])
    else:
        return redirect(url_for("warframe_list"))


@website.route("/warframes", strict_slashes=False)
def warframe_list():
    return render_template("warframe_list.html", warframe_names=storage.warframe_list.names,
                           file_names=storage.warframe_list.file_names)


@website.route("/<path:path>", strict_slashes=False)
def catch_all(path):
    print("User tried to go to /{}".format(path))
    return redirect(url_for("homepage"))