diff --git a/main.py b/main.py index fd03988..b6fefeb 100644 --- a/main.py +++ b/main.py @@ -2,7 +2,9 @@ import base64 from flask import Flask, request -from model import Message +from model import Message + + app = Flask(__name__) @@ -10,7 +12,7 @@ def home(): if request.method == 'POST': - m = Message(content=request.form['content']) + m = Message(content=request.form['content'].replace('<', '<').replace('>', '>')) m.save() body = """ @@ -25,7 +27,7 @@ def home():

Wisdom From Your Fellow Classmates

""" - + for m in Message.select(): body += """
@@ -33,7 +35,7 @@ def home():
""".format(m.content) - return body + return body if __name__ == "__main__":