1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
|
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
# 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})")
|
