Prevent NotInAppUsers exception

gbtami · Jan 9, 2025 · e88badb · e88badb
1 parent 01e982a
commit e88badb
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/server/puzzle.py b/server/puzzle.py
@@ -198,13 +198,14 @@ async def puzzle_vote(request):
     good = post_data["vote"] == "true"
     up_or_down = "up" if good else "down"
 
-    users = app_state.users
+    # Who made the request?
     session = await aiohttp_session.get_session(request)
-    try:
-        user = users[session.get("user_name")]
-    except KeyError:
+    session_user = session.get("user_name")
+    if session_user is None:
         return web.json_response({})
 
+    user = await app_state.users.get(session_user)
+
     if user.puzzles.get("puzzleId"):
         return web.json_response({})
     else: