Merge pull request #40 from deruikong/master

fixed #38
jtyliu · Apr 17, 2023 · eec63b6 · eec63b6
2 parents d3ffaa6 + 16cd7c7
commit eec63b6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/utils/api.py b/utils/api.py
@@ -716,8 +716,12 @@ async def get_judges(self, page: int = None) -> None:
     async def get_pfp(self, username: str) -> str:
         resp = await _query_api(SITE_URL + "user/" + username, "text")
         soup = BeautifulSoup(resp, features="html5lib")
-        pfp = soup.find("div", class_="user-gravatar").find("img")["src"]
-        return pfp
+        try:
+            pfp = soup.find("img", class_="user-gravatar")["src"]
+            return pfp
+        except AttributeError:
+            return None
+
 
     async def get_user_description(self, username: str) -> str:
         resp = await _query_api(SITE_URL + "user/" + username, "text")

diff --git a/utils/jomd_common.py b/utils/jomd_common.py
@@ -100,8 +100,9 @@ async def gimme_common(username, points, types):
         description="Points: %s\nProblem Types: %s" % (points, ", ".join(problem.types)),
         color=0xFCDB05,
     )
-
-    embed.set_thumbnail(await query.get_pfp(username))
+    pfp = await query.get_pfp(username)
+    if pfp is not None:
+        embed.set_thumbnail(pfp)
     embed.add_field(name="Group", value=problem.group, inline=True)
     embed.add_field(name="Time", value="%ss" % problem.time_limit, inline=True)
     embed.add_field(name="Memory", value=memory, inline=True)