www-main/main.py

95 lines
3.7 KiB
Python

# -*- coding: utf-8 -*-
import feedparser
import requests
import string
import random
from flask import Flask, render_template, request, flash, session, abort
from time import mktime
from datetime import datetime
app = Flask('main')
app.config.from_pyfile('main.cfg')
def pull_feed_entries():
all_entries = []
for tag, url in app.config['FEEDS']:
entries = feedparser.parse(url).entries
for e in entries:
e.tag = tag
dt = datetime.fromtimestamp(mktime(e.updated_parsed))
e.updated_display = dt.strftime(app.config['DATE_FORMAT'])
all_entries.extend(entries)
all_entries.sort(key=lambda e: e.updated_parsed, reverse=True)
return all_entries[:app.config['MAX_ENTRIES']]
def mailman_subscribe(email, mailing_list):
password = "".join(random.choice(string.letters) for _ in range(8))
post = {"email": email, "fullname": "", "pw": password, "pw-conf": password, "email-button": "Zapisz", "digest": 0}
headers = {"X-Forwarded-For": request.remote_addr}
r = requests.post("https://lists.hackerspace.pl/subscribe/" + mailing_list, post, headers=headers)
if r.status_code != 200:
return False
return True
def parse_subscribe_requests():
if "email" in request.form:
email = request.form["email"].strip()
if len(email) > 0:
if "@" in email:
lists = []
if "mail-waw" in request.form:
lists.append("waw")
if "mail-proj" in request.form:
lists.append("waw-proj")
if "mail-offtopic" in request.form:
lists.append("waw-ot")
subscribed = []
failed = []
for maillist in lists:
if mailman_subscribe(email, maillist):
subscribed.append(maillist)
else:
failed.append(maillist)
if len(subscribed) == 1:
flash(u"Pomyślnie zasubskrybowano na listę %s@lists.hackerspace.pl.\
W celu aktywacji subskrypcji odwiedź odnośnik wysłany mailem na adres %s." % (subscribed[0], email))
elif len(subscribed) > 1:
flash(u"Pomyślnie zasubskrybowano na listy %s. \
W celu aktywacji subskrypcji odwiedź odnośniki wysłane mailem na adres %s." % \
(", ".join(l + "@lists.hackerspace.pl" for l in subscribed), email))
if len(failed) > 0:
flash(u"Wystąpił problem z zapisaniem na następujące listy: %s. Ups! Napisz na bofh@hackerspace.pl, spróbujemy to naprawić." % ", ".join(failed), "error")
else:
flash(u"Podano nieprawidłowy adres email.", "error")
else:
flash(u"Nie podano adresu email.", "error")
@app.route('/', methods=["GET", "POST"])
def main():
if request.method == "POST":
parse_subscribe_requests()
return render_template('main.html', entries=pull_feed_entries())
@app.route('/about', methods=["GET", "POST"])
def about():
if request.method == "POST":
parse_subscribe_requests()
return render_template('about.html', entries=pull_feed_entries())
@app.before_request
def csrf_protect():
if request.method == "POST":
token = session.pop('_csrf_token', None)
if not token or token != request.form.get('_csrf_token'):
abort(403)
generate_csrf_token()
def generate_csrf_token():
if '_csrf_token' not in session:
session['_csrf_token'] = "".join(random.choice(string.letters) for _ in range(32))
return session['_csrf_token']
app.jinja_env.globals['csrf_token'] = generate_csrf_token
if __name__ == '__main__':
app.run('0.0.0.0', 8080, debug=True)