135 lines
3.5 KiB
Python
135 lines
3.5 KiB
Python
from functools import lru_cache
|
|
import json
|
|
import os
|
|
import random
|
|
import time
|
|
from sys import argv
|
|
from tempfile import mkdtemp, mkstemp
|
|
|
|
from pyhocon import ConfigFactory
|
|
import requests
|
|
from flask import Flask, render_template, request, redirect
|
|
|
|
from stats import get_data
|
|
|
|
## disgusting hack for python 3.4.0
|
|
import pkgutil
|
|
orig_get_loader = pkgutil.get_loader
|
|
def get_loader(name):
|
|
try:
|
|
return orig_get_loader(name)
|
|
except AttributeError:
|
|
pass
|
|
pkgutil.get_loader = get_loader
|
|
##########
|
|
|
|
SITE_NAME = 'tilde.town'
|
|
|
|
app = Flask('~cgi')
|
|
|
|
@lru_cache(maxsize=32)
|
|
def cfg(k):
|
|
conf = ConfigFactory.parse_file('cfg.conf')
|
|
return conf[k]
|
|
|
|
@lru_cache(maxsize=32)
|
|
def site_context():
|
|
return get_data()
|
|
|
|
def slurp(file_path):
|
|
contents = None
|
|
with open(file_path, 'r') as f:
|
|
contents = f.read()
|
|
return contents
|
|
|
|
def save_post(name, message):
|
|
timestamp = time.time()
|
|
data = {
|
|
'name': name,
|
|
'message': message,
|
|
'timestamp': timestamp,
|
|
}
|
|
_, file_path = mkstemp(dir=app.config['DATA_DIR'])
|
|
with open(file_path, 'w') as f:
|
|
f.write(json.dumps(data))
|
|
|
|
@app.route('/random', methods=['GET'])
|
|
def get_random():
|
|
user = random.choice(site_context()['live_users'])
|
|
return redirect('http://tilde.town/~{}'.format(user['username']))
|
|
|
|
@app.route('/guestbook', methods=['GET'])
|
|
def get_guestbook():
|
|
data_dir = app.config['DATA_DIR']
|
|
# TODO sort by timestamp
|
|
filename_to_json = lambda p: json.loads(slurp(os.path.join(data_dir, p)))
|
|
posts = map(filename_to_json, os.listdir(data_dir))
|
|
sorted_posts = sorted(posts, key=lambda p: p['timestamp'], reverse=True)
|
|
|
|
context = {
|
|
"posts": sorted_posts,
|
|
}
|
|
context.update(site_context())
|
|
return render_template('guestbook.html', **context)
|
|
|
|
@app.route('/guestbook', methods=['POST'])
|
|
def post_guestbook():
|
|
message = request.form['message'][0:400]
|
|
name = request.form['name'][0:140]
|
|
captcha = request.form['hmm']
|
|
if captcha == "scrop":
|
|
save_post(name, message)
|
|
return redirect("/guestbook")
|
|
|
|
@app.route('/helpdesk', methods=['GET'])
|
|
def get_helpdesk():
|
|
status = request.args.get('status', 'unsubmitted')
|
|
desc = request.args.get('desc', '')
|
|
context = {
|
|
'status': status,
|
|
'desc': desc,
|
|
}
|
|
context.update(site_context())
|
|
return render_template('helpdesk.html', **context)
|
|
|
|
def send_email(data):
|
|
name = data.get('name', 'anonymous')
|
|
email = data['email']
|
|
request_type = data['type']
|
|
desc = request.form['desc']
|
|
response = requests.post(
|
|
cfg('mailgun_url'),
|
|
auth=("api", cfg('mailgun_key')),
|
|
data={"from": "root@tilde.town",
|
|
"to": cfg('trello'),
|
|
"subject": "{} from {} <{}>".format(request_type, name, email),
|
|
"text": desc})
|
|
|
|
return response.status_code == 200
|
|
|
|
@app.route('/helpdesk', methods=['POST'])
|
|
def post_helpdesk():
|
|
desc = request.form['desc']
|
|
captcha = request.form['hmm']
|
|
|
|
if captcha == 'scrop':
|
|
status = "success" if send_email(request.form) else "fail"
|
|
else:
|
|
status = "fail"
|
|
|
|
# should we bother restoring other fields beside desc?
|
|
return redirect('/helpdesk?status={}&desc={}'.format(status, desc))
|
|
|
|
if __name__ == '__main__':
|
|
if len(argv) == 2:
|
|
data_dir = argv[1]
|
|
else:
|
|
data_dir = mkdtemp()
|
|
|
|
app.config['DEBUG'] = True
|
|
app.config['DATA_DIR'] = data_dir
|
|
|
|
print("Running with data_dir=", app.config['DATA_DIR'])
|
|
|
|
app.run()
|