148 lines
3.4 KiB
Python
148 lines
3.4 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
A Flask-based questing platform.
|
|
"""
|
|
#TODO: look into how cloudflare proxying interacts with nginx
|
|
import os
|
|
import time
|
|
|
|
from flask import Flask
|
|
from flask_paranoid import Paranoid
|
|
from flask_session import Session
|
|
from flask_socketio import SocketIO
|
|
from werkzeug.contrib.fixers import ProxyFix
|
|
|
|
import database as db
|
|
from views import views
|
|
from events import socketio
|
|
|
|
class ReverseProxied(object):
|
|
"""
|
|
Wrap the application in this middleware and configure the
|
|
front-end server to add these headers, to let you quietly bind
|
|
this to a URL other than / and to an HTTP scheme that is
|
|
different than what is used locally.
|
|
|
|
In nginx:
|
|
location /myprefix {
|
|
proxy_pass http://192.168.0.1:5001;
|
|
proxy_set_header Host $host;
|
|
proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
|
|
proxy_set_header X-Scheme $scheme;
|
|
proxy_set_header X-Script-Name /myprefix;
|
|
}
|
|
|
|
:param app: the WSGI application
|
|
"""
|
|
def __init__(self, app):
|
|
self.app = app
|
|
|
|
def __call__(self, environ, start_response):
|
|
script_name = environ.get('HTTP_X_SCRIPT_NAME', '')
|
|
if script_name:
|
|
environ['SCRIPT_NAME'] = script_name
|
|
path_info = environ['PATH_INFO']
|
|
if path_info.startswith(script_name):
|
|
environ['PATH_INFO'] = path_info[len(script_name):]
|
|
|
|
scheme = environ.get('HTTP_X_SCHEME', '')
|
|
if scheme:
|
|
environ['wsgi.url_scheme'] = scheme
|
|
return self.app(environ, start_response)
|
|
|
|
|
|
app = Flask(__name__)
|
|
app.wsgi_app = ReverseProxied(app.wsgi_app)
|
|
app.wsgi_app = ProxyFix(app.wsgi_app)
|
|
app.register_blueprint(views)
|
|
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024
|
|
app.config['SESSION_TYPE'] = 'filesystem'
|
|
app.jinja_env.trim_blocks = True
|
|
app.jinja_env.lstrip_blocks = True
|
|
app.jinja_env.undefined = "StrictUndefined"
|
|
Session(app)
|
|
socketio.init_app(app)
|
|
paranoid = Paranoid(app)
|
|
paranoid.redirect_view = 'views.index'
|
|
|
|
|
|
def init():
|
|
"""
|
|
Initializes the application.
|
|
"""
|
|
# init secret key
|
|
if os.path.exists("secret_key"):
|
|
with open("secret_key", "rb") as file:
|
|
secret_key = file.read()
|
|
else:
|
|
secret_key = os.urandom(64)
|
|
with open("secret_key", "wb") as file:
|
|
file.write(secret_key)
|
|
app.secret_key = secret_key
|
|
|
|
|
|
@app.template_filter("strftime")
|
|
def unix2string(unix):
|
|
"""
|
|
Converts a unix timestamp into a string.
|
|
"""
|
|
form = "%Y-%m-%d %H:%M:%S"
|
|
t = time.localtime(unix)
|
|
return time.strftime(form, t)
|
|
|
|
|
|
@app.template_filter("dice_chal")
|
|
def get_dice_challenge(post_id):
|
|
"""
|
|
Returns the dice challenge for the given post_id.
|
|
"""
|
|
return db.get_dice_call(post_id)[3]
|
|
|
|
|
|
@app.template_filter("num_votes")
|
|
def num_votes(option_id):
|
|
"""
|
|
Returns the number of IPs who voted for this option.
|
|
"""
|
|
return db.get_num_votes(option_id)
|
|
|
|
|
|
@app.template_filter("split_options")
|
|
def split_options(options):
|
|
"""
|
|
Splits a polls options into a list.
|
|
"""
|
|
return options.split("\n")
|
|
|
|
|
|
@app.template_filter("get_rolls")
|
|
def get_rolls(post_id):
|
|
"""
|
|
Gets the dice rolls for the template.
|
|
"""
|
|
return db.get_dice_rolls(post_id=post_id)
|
|
|
|
|
|
@app.template_filter("get_options")
|
|
def get_options(post_id):
|
|
"""
|
|
Gets the poll options for the template.
|
|
"""
|
|
return db.get_poll_options(post_id=post_id)
|
|
|
|
|
|
@app.after_request
|
|
def minify(res):
|
|
"""
|
|
Minify the HTML response to decrease bandwidth usage.
|
|
"""
|
|
if res.content_type == "text/html; charset=utf-8" and False:
|
|
res.set_data(
|
|
res.get_data(as_text=True).replace("\n", "").replace("\t", ""))
|
|
return res
|
|
|
|
|
|
init()
|
|
if __name__ == "__main__":
|
|
app.run(host='0.0.0.0', port=5050)
|