Skip to content

Commit 83308d6

Browse files
committed
print csv for easier pr
1 parent 5529ffb commit 83308d6

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

discord_bot/util/views.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,17 @@
33

44
from util.embeds import Embeds
55

6+
# ya this is duplicated in commands but whatever
7+
NAMES = ["Souren", "Kevin", "Ken", "Artom", "Andrew"]
8+
IDS = [
9+
"290550631591182336",
10+
"183383346569543681",
11+
"290630197458501633",
12+
"161921661346643968",
13+
"196069761082327041",
14+
]
15+
ID_NAMES_MAP = {IDS[i]: NAMES[i] for i in range(len(NAMES))}
16+
617

718
class JudgeView(discord.ui.View):
819
def __init__(self, songs: dict[str, str], userIds: list[str]):
@@ -25,4 +36,17 @@ async def callback(self, interaction: discord.Interaction):
2536
)
2637
)
2738
await interaction.channel.send(content=f"{self.songs[self.index]}")
39+
# generate 4 lines of csv as a string to be added to the winners.csv file
40+
# csv should have the format submitter_name, song_link, date YYYY-MM-DD, winner (0 or 1)
41+
csv_string = ""
42+
from datetime import datetime
43+
44+
today = datetime.now().date()
45+
for key in self.songs:
46+
name = ID_NAMES_MAP[str(key)] if str(key) in ID_NAMES_MAP else "Unknown"
47+
if key == self.index:
48+
csv_string += f"{name},{self.songs[key]},{today},1\n"
49+
else:
50+
csv_string += f"{name},{self.songs[key]},{today},0\n"
51+
await interaction.channel.send(content=f"```csv\n{csv_string}```")
2852
self.view.stop()

0 commit comments

Comments
 (0)