From 0151913e618e1d53d08f639cb20b5bb1663d1466 Mon Sep 17 00:00:00 2001
From: stijndcl <declercq.stijn@outlook.com>
Date: Thu, 31 Mar 2022 20:12:18 +0200
Subject: [PATCH] Update compbio leaderboard to assignment 4

---
 .gitignore                 |  1 +
 cogs/slash/school_slash.py |  5 +++--
 data/menus/leaderboards.py | 18 ++++++------------
 3 files changed, 10 insertions(+), 14 deletions(-)

diff --git a/.gitignore b/.gitignore
index df503ea..4784e9c 100644
--- a/.gitignore
+++ b/.gitignore
@@ -6,6 +6,7 @@ files/lost.json
 files/locked.json
 files/ufora_notifications.json
 files/compbio_benchmarks_2.json
+files/compbio_benchmarks_4.json
 .idea/
 __pycache__
 .env
diff --git a/cogs/slash/school_slash.py b/cogs/slash/school_slash.py
index 3b6df67..c26f61b 100644
--- a/cogs/slash/school_slash.py
+++ b/cogs/slash/school_slash.py
@@ -91,9 +91,10 @@ class SchoolSlash(commands.Cog):
 
     @_compbio_group.command(name="leaderboard", description="Gesorteerd en ingevuld leaderboard")
     async def _compbio_lb_slash(self, ctx: ApplicationContext,
-                                benchmark: Option(int, "De specifieke benchmark om op te halen (default k=600)", choices=[6, 10, 50, 600], default=600)):
+                                benchmark: Option(str, "De specifieke benchmark om op te halen (default 10000-10)", choices=["100-10", "100-100", "1000-100", "10000-10"], default="10000-10")):
         await ctx.response.defer()
-        lb = leaderboards.CompbioLeaderboard(ctx, kmer=benchmark)
+        size, amount = benchmark.split("-")
+        lb = leaderboards.CompbioLeaderboard(ctx, size=size, amount=amount)
         await lb.respond()
 
     @_compbio_group.command(name="submit", description="Link een Dodona-submission aan jouw username")
diff --git a/data/menus/leaderboards.py b/data/menus/leaderboards.py
index af97e18..1680568 100644
--- a/data/menus/leaderboards.py
+++ b/data/menus/leaderboards.py
@@ -123,16 +123,17 @@ class BitcoinLeaderboard(Leaderboard):
 @dataclass
 class CompbioLeaderboard(Leaderboard):
     colour: discord.Colour = field(default=discord.Colour.green())
-    title: str = field(default="Leaderboard Computationele Biologie #2")
+    title: str = field(default="Leaderboard Computationele Biologie #4")
     reverse: bool = False
-    kmer: int = 600
+    size: int = 10000
+    amount: int = 10
 
     def __post_init__(self):
-        self.title += f" (k = {self.kmer})"
+        self.title += f" ({self.size}-{self.amount})"
         super().__post_init__()
 
     def get_submission_user(self, submission_id: str) -> str:
-        with open("files/compbio_benchmarks_2.json", "r") as fp:
+        with open("files/compbio_benchmarks_4.json", "r") as fp:
             file = json.load(fp)
 
         if submission_id in file:
@@ -142,14 +143,7 @@ class CompbioLeaderboard(Leaderboard):
         return f"[# {submission_id}]"
 
     def get_data(self) -> list[tuple]:
-        files = {
-            6: "J02459.1",
-            10: "J02459.1",
-            50: "J02459.1",
-            600: "AF033819.3"
-        }
-
-        url = f"https://github.ugent.be/raw/computationele-biologie/benchmarks-2022/main/reconstruction/{files[self.kmer]}.{self.kmer}mers.md"
+        url = f"https://github.ugent.be/raw/computationele-biologie/benchmarks-2022/tree/main/profile_hmm/size{self.size}-amount{self.amount}.md"
         headers = {"Authorization": f"token {settings.UGENT_GH_TOKEN}"}
         result = requests.get(url, headers=headers).text