From c16815b16d006384a88001d9938ff5a44aa57f0e Mon Sep 17 00:00:00 2001
From: CaitSith2 <d_good@caitsith2.com>
Date: Thu, 9 Jun 2022 22:14:31 -0700
Subject: [PATCH] Fix Room log

---
 WebHostLib/__init__.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/WebHostLib/__init__.py b/WebHostLib/__init__.py
index 522bbb27..79c90499 100644
--- a/WebHostLib/__init__.py
+++ b/WebHostLib/__init__.py
@@ -170,8 +170,11 @@ def _read_log(path: str):
 
 @app.route('/log/<suuid:room>')
 def display_log(room: UUID):
+    room = Room.get(id=room)
+    if room is None:
+        return abort(404)
     if room.owner == session["_id"]:
-        return Response(_read_log(os.path.join("logs", str(room) + ".txt")), mimetype="text/plain;charset=UTF-8")
+        return Response(_read_log(os.path.join("logs", str(room.id) + ".txt")), mimetype="text/plain;charset=UTF-8")
     return "Access Denied", 403