forked from Ugric/Komodo-Hub
72 lines
1.9 KiB
Python
72 lines
1.9 KiB
Python
from flask import Flask, request, g, render_template
|
|
import os
|
|
import importlib.util
|
|
from database.db_init import init_database
|
|
from database.db_commands import get_user_from_token
|
|
from werkzeug.exceptions import HTTPException
|
|
|
|
|
|
def import_pages(route, app):
|
|
for root, _, files in os.walk(route):
|
|
for file in files:
|
|
if not file.endswith(".py"):
|
|
continue
|
|
|
|
path = os.path.join(root, file)
|
|
module_name = os.path.splitext(os.path.relpath(path, route))[0]
|
|
module_name = module_name.replace(os.sep, ".")
|
|
|
|
spec = importlib.util.spec_from_file_location(module_name, path)
|
|
if spec is None or spec.loader is None:
|
|
continue
|
|
|
|
module = importlib.util.module_from_spec(spec)
|
|
|
|
try:
|
|
spec.loader.exec_module(module)
|
|
except Exception as e:
|
|
print(f"Failed to import {path}: {e}")
|
|
continue
|
|
|
|
if hasattr(module, "page"):
|
|
print(f"{module_name}.page =", module.page)
|
|
app.register_blueprint(module.page)
|
|
|
|
|
|
app = Flask(__name__, static_folder="static", static_url_path="")
|
|
app.secret_key = "9279dc6c-d2bc-4f89-a1b3-fc1438a6910c"
|
|
init_database()
|
|
|
|
@app.before_request
|
|
def load_user():
|
|
token = request.cookies.get("token")
|
|
g.user = token and get_user_from_token(token)
|
|
|
|
|
|
@app.context_processor
|
|
def inject_user():
|
|
return {"user": g.user}
|
|
|
|
@app.errorhandler(404)
|
|
def page_not_found(e):
|
|
return render_template(
|
|
"error.jinja",
|
|
extra_information=e,
|
|
), 404
|
|
|
|
@app.errorhandler(Exception)
|
|
def handle_bad_request(e):
|
|
print(e)
|
|
if isinstance(e, HTTPException):
|
|
return e
|
|
return render_template(
|
|
"error.jinja",
|
|
extra_information=e,
|
|
), 500
|
|
|
|
|
|
import_pages("routes", app)
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host="0.0.0.0", port=3000, debug=True)
|