Fix crash in case of vanity code and not found

2023-11-07 02:12:55 +01:00 · 2023-11-07 02:12:55 +01:00 · f7779ace60
commit f7779ace60
parent f49f69cc79
1 changed files with 8 additions and 2 deletions
--- a/divent/bot.py
+++ b/divent/bot.py
@ -221,7 +221,10 @@ async def subscribe(entity_id: str):
            entity_id=guild.vanity_url_code or guild.id,
        )

-    user = await client.get_or_fetch_user(int(entity_id))
+    try:
+        user = await client.get_or_fetch_user(int(entity_id))
+    except ValueError:
+        return redirect(url_for(".login"))

    if user and str(user.id) == entity_id:
        return await render_template(
@ -281,7 +284,10 @@ async def ical(entity_id: str):

        return calendar.serialize()

-    user = await client.get_or_fetch_user(int(entity_id))
+    try:
+        user = await client.get_or_fetch_user(int(entity_id))
+    except ValueError:
+        return redirect(url_for(".login"))

    if user:
        calendar = Calendar()