StarfallBot/starfall/web/blueprints/secure/main.py

26 lines
1.0 KiB
Python

from flask import Blueprint, Flask, redirect, render_template, url_for
from flask_assets import Environment
from starfall.web.blueprints.base import BaseBlueprint
from starfall.web.controllers.secure.login import LoginController
from starfall.web.controllers.secure.register import RegisterController
class AdminBlueprint(BaseBlueprint):
def __init__(self, blueprint: Blueprint, assets: Environment, app: Flask):
super().__init__(blueprint, assets, app)
blueprint.add_url_rule("/secure/", view_func=self.secure)
blueprint.add_url_rule("/secure/login/", methods=["GET", "POST"], view_func=self.login)
blueprint.add_url_rule("/secure/register/", methods=["GET", "POST"], view_func=self.register)
def secure(self):
return redirect(url_for("starfall.login"))
def login(self):
LoginController.apply(self)
return render_template("secure/login.jinja", bp=self)
def register(self):
RegisterController.apply(self)
return render_template("secure/register.jinja", bp=self)