From 32edc4c5a43e0e3f774aa34e313e82054ab9fda5 Mon Sep 17 00:00:00 2001
From: Jason Jenkins <jdj0891@yahoo.com>
Date: Thu, 4 Mar 2021 09:53:07 -0800
Subject: [PATCH] Complete

---
 main.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

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('<', '&lt;').replace('>', '&gt;'))
         m.save()
 
     body = """
@@ -25,7 +27,7 @@ def home():
 
 <h2>Wisdom From Your Fellow Classmates</h2>
 """
-    
+
     for m in Message.select():
         body += """
 <div class="message">
@@ -33,7 +35,7 @@ def home():
 </div>
 """.format(m.content)
 
-    return body 
+    return body
 
 
 if __name__ == "__main__":