✨

aula/Threads/routes.py

@@ -1,6 +1,7 @@
-from flask import render_template, Blueprint
+from flask import redirect, render_template, Blueprint
 from flask_login import login_required, current_user
-from aula.models import get_thread
+from aula.models import get_thread, insert_message
+from aula.forms import SendMessageForm
 
 Threads = Blueprint('Threads', __name__)
 
@@ -10,7 +11,7 @@ def threads():
     threads_data = current_user.get_threads()
     return render_template("threads.html", threads=threads_data)
 
-@Threads.route("/threads/<int:thread_id>", methods=['GET'])
+@Threads.route("/threads/<int:thread_id>", methods=['GET', 'POST'])
 @login_required
 def show(thread_id):
     thread = get_thread(thread_id)
@@ -19,4 +20,10 @@ def show(thread_id):
     elif not current_user.in_thread(thread_id):
         return f"Du har ikke adgang til tråden med id {thread_id}."
     else:
-        return render_template("thread_show.html", thread=thread, messages=thread.get_messages())
+        form = SendMessageForm()
+
+        if form.validate_on_submit():
+            insert_message(form.besked.data, thread_id, current_user.user_id)
+            return redirect(f"/threads/{thread_id}")
+
+        return render_template("thread_show.html", thread=thread, messages=thread.get_messages(), form=form)