anonkun/anonkun.py
2018-06-14 14:40:35 -04:00

89 lines
2.3 KiB
Python

#!/usr/bin/env python3
"""
Simple file host using Flask.
"""
import os
import time
from flask import Flask, session, request, abort, redirect, url_for, g, \
render_template
from flask_socketio import SocketIO, emit
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.config['SECRET_KEY'] = 'secret!'
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024
socketio = SocketIO(app)
@socketio.on('joined', namespace='/chat')
def joined(message):
"""
Sent by clients when they enter a room.
"""
print("Client connected.")
@socketio.on('text', namespace='/chat')
def text(data):
"""
Sent by a client when the user entered a new message.
"""
message = data["msg"]
emit('message', {'msg': message})
@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)
messages = [{"name":"Anonymous", "date":1528998539, "message":"lol"}, {"name":"Namefag", "date":1528998521, "message":"kek"}]
@app.route("/", methods=["GET", "POST"])
def index():
if request.method == "GET":
return render_template('index.html', messages=messages)
message = request.form.get("msg")
messages.append(message)
#print(message)
return redirect(url_for("index"))