diff --git a/main.py b/main.py index fd03988..0da1720 100644 --- a/main.py +++ b/main.py @@ -1,17 +1,22 @@ import os import base64 +import random -from flask import Flask, request +from flask import Flask, request, session from model import Message app = Flask(__name__) +app.secret_key = '12345678910' @app.route('/', methods=['GET', 'POST']) def home(): + if 'csrf_token' not in session: + session['csrf_token'] = str(random.randint(10000000, 99999999)) if request.method == 'POST': - m = Message(content=request.form['content']) - m.save() + if request.form.get('csrf_token', None) == session['csrf_token']: + m = Message(content=request.form['content']) + m.save() body = """ @@ -19,19 +24,23 @@ def home():

Class Message Board

Contribute to the Knowledge of Others

+ +

Wisdom From Your Fellow Classmates

-""" +""".format(session['csrf_token']) + for m in Message.select(): body += """
{}
-""".format(m.content) +""".format(m.content.replace('<', '<').replace('>', '>')) + return body