nex/college-bot-v2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Activity Actions

Add the refresh button
Some checks failed
Build and Publish / build_and_publish (push) Has been cancelled
Details

Browse source
This commit is contained in:
Nexus 2024-07-05 00:34:14 +01:00
parent a8f5ae145a
commit dbd06bfe38
1 changed files with 52 additions and 33 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

43
src/cogs/election.py
View file

@ -2,6 +2,8 @@
This module is only meant to be loaded during election times. This module is only meant to be loaded during election times.
""" """
import asyncio import asyncio
from calendar import c
from contextlib import asynccontextmanager
import logging import logging
import random import random
import datetime import datetime
@ -148,18 +150,11 @@ class ElectionCog(commands.Cog):
# ] # ]
return results return results
@commands.slash_command(name="election") async def _get_embed(self) -> discord.Embed | None:
async def get_election_results(self, ctx: discord.ApplicationContext):
"""Gets the current election results"""
await ctx.defer()
async with httpx.AsyncClient(headers=self.HEADERS) as client: async with httpx.AsyncClient(headers=self.HEADERS) as client:
response = await client.get(self.SOURCE, follow_redirects=True) response = await client.get(self.SOURCE, follow_redirects=True)
if response.status_code != 200: if response.status_code != 200:
return await ctx.respond( raise RuntimeError(f"HTTP {response.status_code} while fetching results from BBC")
"Sorry, I can't do that right now (HTTP %d while fetching results from BBC)" % response.status_code
)
# noinspection PyTypeChecker
soup = await asyncio.to_thread(BeautifulSoup, response.text, "html.parser") soup = await asyncio.to_thread(BeautifulSoup, response.text, "html.parser")
results = await self.bot.loop.run_in_executor(None, self.process_soup, soup) results = await self.bot.loop.run_in_executor(None, self.process_soup, soup)
if results: if results:
@ -185,9 +180,33 @@ class ElectionCog(commands.Cog):
top_party = list(sorted(colour_scores.keys(), key=lambda k: colour_scores[k], reverse=True))[0] top_party = list(sorted(colour_scores.keys(), key=lambda k: colour_scores[k], reverse=True))[0]
embed.colour = discord.Colour(results[top_party][2]) embed.colour = discord.Colour(results[top_party][2])
embed.description = "\n".join(description_parts) embed.description = "\n".join(description_parts)
return await ctx.respond(embed=embed) return embed
else:
return await ctx.respond("Unable to get election results at this time.") @commands.slash_command(name="election")
async def get_election_results(self, ctx: discord.ApplicationContext):
"""Gets the current election results"""
class RefreshView(discord.ui.View):
@discord.ui.button(label="Refresh", style=discord.ButtonStyle.primary, emoji="\U0001f501")
async def refresh(_self, _btn, interaction):
await interaction.response.defer(invisible=True)
try:
embed = await self._get_embed()
except Exception as e:
self.log.exception("Failed to get election results.")
return await interaction.followup.send(f"Sorry, I cannot contact the BBC at this time: {e}")
if embed is None:
return await interaction.followup.send("Sorry, I could not find any election results.")
await interaction.edit_original_response(embed=embed)
await ctx.defer()
try:
embed = await self._get_embed()
except Exception as e:
self.log.exception("Failed to get election results.")
return await ctx.respond(f"Sorry, I cannot contact the BBC at this time: {e}")
if embed is None:
return await ctx.respond("Sorry, I could not find any election results.")
await ctx.respond(embed=embed, view=RefreshView())
def setup(bot): def setup(bot):