Update contest URLs and add duration property

rtk-rnjn · Mar 29, 2024 · 5f4b23d · 5f4b23d
1 parent 17e8366
commit 5f4b23d
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 6 deletions.
diff --git a/cogs/rtfm/_kontests/codechef.py b/cogs/rtfm/_kontests/codechef.py
@@ -27,12 +27,16 @@ def name(self) -> str:
 
     @property
     def start_time(self) -> datetime.datetime:
-        return datetime.datetime.strptime(self.__data["start"], TIME_FORMAT)
+        return datetime.datetime.strptime(self.__data["start_in"], TIME_FORMAT)
 
     @property
     def url(self) -> str:
         return f"{API}/{self.code}"
 
+    @property
+    def duration(self) -> str:
+        return self.__data["duration"]
+
 
 class CodeChef:
     def __init__(self) -> None:

diff --git a/cogs/rtfm/_kontests/csacademy.py b/cogs/rtfm/_kontests/csacademy.py
@@ -19,10 +19,6 @@ class CSAcademyContestData:
     def __init__(self, data: dict) -> None:
         self.__data = data
 
-    @property
-    def id(self) -> int:
-        return self.__data["id"]
-
     @property
     def name(self) -> str:
         return self.__data["longName"]
@@ -60,6 +56,10 @@ def description(self) -> str:
     def rated(self) -> bool:
         return self.__data["rated"]
 
+    @property
+    def url(self) -> str:
+        return f"{BASE_URL}/{self.__data['name']}"
+
 
 class CSAcademy:
     def __init__(self) -> None:

diff --git a/cogs/rtfm/rtfm.py b/cogs/rtfm/rtfm.py
@@ -1477,7 +1477,7 @@ async def kontest_codeforces(self) -> list:
         contests = self.kontests_cache["codeforces"]
 
         return [
-            f"""ID: *{contest.id}* | **[{contest.name}]({contest.website_url})** | {contest.phase} | {contest.difficulty}
+            f"""ID: *{contest.id}* | **[{contest.name}]({contest.website_url})** | {contest.phase}
 {contest.description}
 `Start:` {discord.utils.format_dt(contest.start_time, "R") if contest.start_time else 'TBA'}
 `Time :` {contest.duration_seconds // 60} Minutes
@@ -1489,6 +1489,7 @@ async def kontest_atcoder(self) -> list:
         if "atcoder" not in self.kontests_cache:
             atcoder = AtCoder()
             await atcoder.fetch(self.bot.http_session)
+            await atcoder.get_contests(self.bot.http_session)
             self.kontests_cache["atcoder"] = atcoder.contests
 
         contests = self.kontests_cache["atcoder"]
@@ -1497,6 +1498,23 @@ async def kontest_atcoder(self) -> list:
             f"""ID: NA | **[{contest.name}]({contest.url})**
 *Description not available*
 `Start:` {discord.utils.format_dt(contest.start_time, "R")}
+"""
+            for contest in contests
+        ]
+
+    async def kontest_csacademy(self) -> list:
+        if "csacademy" not in self.kontests_cache:
+            csacademy = CSAcademy()
+            await csacademy.fetch(self.bot.http_session)
+            self.kontests_cache["csacademy"] = csacademy.contests
+
+        contests = self.kontests_cache["csacademy"]
+
+        return [
+            f"""ID: NA | **[{contest.name}]({contest.url})**
+{contest.description}
+`Start:` {discord.utils.format_dt(contest.start_time, "R") if contest.start_time else 'TBA'}
+`End  :` {discord.utils.format_dt(contest.end_time, "R") if contest.end_time else 'TBA'}
 """
             for contest in contests
         ]