Merge pull request #28 from Franco1010/see_contest

Command: see

listenerServers/discordListener.py

@@ -1,9 +1,7 @@
 import discord
 import aiohttp
-import urllib.parse
 import os
 from dotenv import load_dotenv
-import json
 
 load_dotenv()
 
@@ -33,9 +31,8 @@ async def send_message(message):
     async with aiohttp.ClientSession() as session:
         async with session.post(url, json=payload) as response:
             response_json = await response.json()
-            channel = message.channel
             print(response_json)
-            await channel.send(response_json["response"])
+            await message.reply(response_json["response"])
 
 
 @client.event

polybotCommand/clickMenu/polygon/contest.py

@@ -28,10 +28,7 @@ async def help(ctx):
 async def contest_list(ctx):
     result = contestsDB.query_contests(ctx.obj["groupId"])
     if len(result):
-        data = []
-        for i in result:
-            if i["source"] == "polygon":
-                data.append([i["name"], i["id"]])
+        data = [[r["name"], r["id"]] for r in result if r["source"] == "polygon"]
         table = (
             "```\n"
             + tabulate(data, headers=["Name", "Id"], tablefmt="pretty")
@@ -88,6 +85,7 @@ async def add(ctx, contestid, done):
 
 @contest.command()
 @click.argument("contestid")
+@click.pass_context
 async def see(ctx, contestid):
     mine = is_mine(ctx.obj["groupId"], contestid)
     if mine != 2:
@@ -97,6 +95,9 @@ async def see(ctx, contestid):
     if contest == None:
         click.echo("Contest is missing write access.")
         return
+    data = [[c.letter, c.name] for c in contest]
+    table = "```\n" + tabulate(data, headers=["#", "Name"], tablefmt="pretty") + "\n```"
+    click.echo(table)
 
 
 @contest.command()

src_layer/polygon/polygonApi.py

@@ -219,6 +219,7 @@ async def contest(contestId):
     if resp is None:
         return None
     problems = []
-    for _, problem in resp.items():
+    for letter, problem in resp.items():
+        problem["letter"] = letter
         problems.append(make_from_dict(utils.Problem, problem))
     return problems

src_layer/polygon/utils.py

@@ -13,6 +13,7 @@
         "revision",
         "latestPackage",
         "modified",
+        "letter",
     ],
 )