diff options
Diffstat (limited to 'pysite/route_manager.py')
| -rw-r--r-- | pysite/route_manager.py | 146 |
1 files changed, 0 insertions, 146 deletions
diff --git a/pysite/route_manager.py b/pysite/route_manager.py deleted file mode 100644 index 79fb67ac..00000000 --- a/pysite/route_manager.py +++ /dev/null @@ -1,146 +0,0 @@ -import importlib -import inspect -import logging -import os - -from flask import Blueprint, Flask, _request_ctx_stack -from flask_dance.contrib.discord import make_discord_blueprint -from flask_sockets import Sockets -from gunicorn_config import _when_ready as when_ready - -from pysite.base_route import APIView, BaseView, ErrorView, RedirectView, RouteView, TemplateView -from pysite.constants import ( - CSRF, DEBUG_MODE, DISCORD_OAUTH_AUTHORIZED, DISCORD_OAUTH_ID, DISCORD_OAUTH_REDIRECT, - DISCORD_OAUTH_SCOPE, DISCORD_OAUTH_SECRET, PREFERRED_URL_SCHEME) -from pysite.database import RethinkDB -from pysite.oauth import OAuthBackend -from pysite.websockets import WS - -TEMPLATES_PATH = "../templates" -STATIC_PATH = "../static" - - -class RouteManager: - def __init__(self): - - # Set up the app and the database - self.app = Flask( - __name__, template_folder=TEMPLATES_PATH, static_folder=STATIC_PATH, static_url_path="/static", - ) - self.sockets = Sockets(self.app) - - self.db = RethinkDB() - self.log = logging.getLogger(__name__) - self.app.secret_key = os.environ.get("WEBPAGE_SECRET_KEY", "super_secret") - self.app.config["SERVER_NAME"] = os.environ.get("SERVER_NAME", "pythondiscord.local:8080") - self.app.config["PREFERRED_URL_SCHEME"] = PREFERRED_URL_SCHEME - self.app.config["WTF_CSRF_CHECK_DEFAULT"] = False # We only want to protect specific routes - - # Trim blocks so that {% block %} statements in templates don't generate blank lines - self.app.jinja_env.trim_blocks = True - self.app.jinja_env.lstrip_blocks = True - - # We make the token valid for the lifetime of the session because of the wiki - you might spend some - # time editing an article, and it seems that session lifetime is a good analogue for how long you have - # to edit - self.app.config["WTF_CSRF_TIME_LIMIT"] = None - - if DEBUG_MODE: - # Migrate the database, as we would in prod - when_ready(output_func=self.db.log.info) - - self.app.before_request(self.db.before_request) - self.app.teardown_request(self.db.teardown_request) - - CSRF.init_app(self.app) # Set up CSRF protection - - # Load the oauth blueprint - self.oauth_backend = OAuthBackend(self) - self.oauth_blueprint = make_discord_blueprint( - DISCORD_OAUTH_ID, - DISCORD_OAUTH_SECRET, - DISCORD_OAUTH_SCOPE, - login_url=DISCORD_OAUTH_REDIRECT, - authorized_url=DISCORD_OAUTH_AUTHORIZED, - redirect_to="main.auth.done", - backend=self.oauth_backend - ) - self.log.debug(f"Loading Blueprint: {self.oauth_blueprint.name}") - self.app.register_blueprint(self.oauth_blueprint) - self.log.debug("") - - # Load the main blueprint - self.main_blueprint = Blueprint("main", __name__) - self.log.debug(f"Loading Blueprint: {self.main_blueprint.name}") - self.load_views(self.main_blueprint, "pysite/views/main") - self.load_views(self.main_blueprint, "pysite/views/error_handlers") - self.app.register_blueprint(self.main_blueprint) - self.log.debug("") - - # Load the subdomains - self.subdomains = ["api", "staff", "wiki"] - - for sub in self.subdomains: - try: - sub_blueprint = Blueprint(sub, __name__, subdomain=sub) - self.log.debug(f"Loading Blueprint: {sub_blueprint.name}") - self.load_views(sub_blueprint, f"pysite/views/{sub}") - self.app.register_blueprint(sub_blueprint) - except Exception: - logging.getLogger(__name__).exception(f"Failed to register blueprint for subdomain: {sub}") - - # Load the websockets - self.ws_blueprint = Blueprint("ws", __name__) - - self.log.debug("Loading websocket routes...") - self.load_views(self.ws_blueprint, "pysite/views/ws") - self.sockets.register_blueprint(self.ws_blueprint, url_prefix="/ws") - - self.app.before_request(self.https_fixing_hook) # Try to fix HTTPS issues - - def https_fixing_hook(self): - """ - Attempt to fix HTTPS issues by modifying the request context stack - """ - - if _request_ctx_stack is not None: - reqctx = _request_ctx_stack.top - reqctx.url_adapter.url_scheme = PREFERRED_URL_SCHEME - - def run(self): - from gevent.pywsgi import WSGIServer - from geventwebsocket.handler import WebSocketHandler - - server = WSGIServer( - ("0.0.0.0", int(os.environ.get("WEBPAGE_PORT", 8080))), # noqa: B104, S104 - self.app, handler_class=WebSocketHandler - ) - server.serve_forever() - - def load_views(self, blueprint, location="pysite/views"): - for filename in os.listdir(location): - if os.path.isdir(f"{location}/{filename}"): - # Recurse if it's a directory; load ALL the views! - self.load_views(blueprint, location=f"{location}/{filename}") - continue - - if filename.endswith(".py") and not filename.startswith("__init__"): - module = importlib.import_module(f"{location}/{filename}".replace("/", ".")[:-3]) - - for cls_name, cls in inspect.getmembers(module): - if ( - inspect.isclass(cls) and - cls is not BaseView and - cls is not ErrorView and - cls is not RouteView and - cls is not APIView and - cls is not WS and - cls is not TemplateView and - cls is not RedirectView and - ( - BaseView in cls.__mro__ or - WS in cls.__mro__ - ) - ): - cls.setup(self, blueprint) - self.log.debug(f">> View loaded: {cls.name: <15} ({module.__name__}.{cls_name})") |