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/') 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