fix: Handle invalid locale

seriaati · Oct 17, 2024 · b8b2e71 · b8b2e71
1 parent 46ef9e0
commit b8b2e71
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/hoyo_buddy/web_app/app.py b/hoyo_buddy/web_app/app.py
@@ -283,7 +283,11 @@ async def _handle_login_routes(self, route: str, parsed_params: dict[str, str])
 
         if route == "/login":
             user_data = await self.fetch_user_data()
-            return pages.LoginPage(user_data, translator=self._translator, locale=Locale(locale))
+            try:
+                locale_ = Locale(locale)
+            except ValueError:
+                return pages.ErrorPage(code=422, message="Invalid locale")
+            return pages.LoginPage(user_data, translator=self._translator, locale=locale_)
 
         if route == "/geetest":
             query: str | None = await page.client_storage.get_async(f"hb.{parsed_params['user_id']}.params")