Fix LookupError in notification loop

Signed-off-by: alfred richardsn <rchrdsn@protonmail.ch>
fincubator · Aug 2, 2020 · 7a1a029 · 7a1a029
1 parent 549e062
commit 7a1a029
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/handlers/base.py b/src/handlers/base.py
@@ -253,7 +253,10 @@ async def show_order(
 
     new_edit_msg = None
     if invert is None:
-        user = database_user.get()
+        try:
+            user = database_user.get()
+        except LookupError:
+            user = await database.users.find_one({"id": user_id})
         invert = user.get("invert_order", False)
     else:
         user = await database.users.find_one_and_update(