From 16cd7c7eef618f50bd119a2759c3a3272cac9317 Mon Sep 17 00:00:00 2001 From: Ray Kong Date: Sun, 16 Apr 2023 18:05:51 -0400 Subject: [PATCH] fixed #38 --- utils/api.py | 8 ++++++-- utils/jomd_common.py | 5 +++-- 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/utils/api.py b/utils/api.py index de8814b..d399e79 100644 --- 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 index f7c183a..ee10273 100644 --- 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)