diff --git a/WebHostLib/__init__.py b/WebHostLib/__init__.py index 847e99c8..b42ac6a0 100644 --- a/WebHostLib/__init__.py +++ b/WebHostLib/__init__.py @@ -136,8 +136,7 @@ def view_seed(seed: UUID): seed = Seed.get(id=seed) if not seed: abort(404) - return render_template("viewSeed.html", seed=seed, - rooms=[room for room in seed.rooms if room.owner == session["_id"]]) + return render_template("viewSeed.html", seed=seed) @app.route('/new_room/') diff --git a/WebHostLib/templates/viewSeed.html b/WebHostLib/templates/viewSeed.html index 62763629..deb62617 100644 --- a/WebHostLib/templates/viewSeed.html +++ b/WebHostLib/templates/viewSeed.html @@ -31,12 +31,16 @@ Rooms:  - {% call macros.list_rooms(rooms) %} + {% call macros.list_rooms(seed.rooms | selectattr("owner", "eq", session["_id"])) %}
  • Create New Room
  • {% endcall %} + {% if seed.rooms %} + There are a total of {{ seed.rooms | length }} Rooms, only those created by you are linked above. + {% endif %} +