58 lines
2.1 KiB
Python
58 lines
2.1 KiB
Python
"""
|
|
Bot command to post the highscores
|
|
"""
|
|
|
|
from nextcord.ext import commands
|
|
import nextcord
|
|
|
|
from cogs.submission.submission_views import SubmissionCreateButton # pylint: disable=import-error
|
|
from highscores import highscores_data # pylint: disable=import-error
|
|
from utilities.utils import send_highscore_message # pylint: disable=import-error
|
|
|
|
|
|
class PostHighscores(commands.Cog, name="Post Highscores"):
|
|
"""Posts Highscores."""
|
|
|
|
def __init__(self, bot: commands.Bot):
|
|
self.bot = bot
|
|
|
|
@commands.command(name="post-scores")
|
|
@commands.has_permissions(administrator=True)
|
|
async def post_highscores(self, ctx: commands.Context):
|
|
"""
|
|
A command which posts the highscore in the designated channel.
|
|
Sends warning if no channel has been registered.
|
|
Usage:
|
|
```
|
|
?post-scores
|
|
```
|
|
"""
|
|
guild = highscores_data.get_guild(ctx.guild.id)
|
|
print(f"Printing out highscores information in channel {guild.highscore_channel}")
|
|
channel = ctx.guild.get_channel(guild.highscore_channel)
|
|
|
|
error_response = "Registered Highscores channel does not exist or was never registered. "
|
|
error_response += 'Register with "?register" command.'
|
|
assert channel is not None, await ctx.send(error_response)
|
|
|
|
if len(highscores_data.get_guild(ctx.guild.id).tables) < 1:
|
|
await channel.send(
|
|
"No tables have been created for this guild. Please create some tables."
|
|
)
|
|
return
|
|
|
|
for table in highscores_data.get_guild(ctx.guild.id).tables:
|
|
message: nextcord.Message = await send_highscore_message(channel, guild, table)
|
|
await message.edit(view=SubmissionCreateButton(self.bot))
|
|
|
|
# have to save data here in case message id was changed. rather save here after all
|
|
# messages have been sent.
|
|
highscores_data.save_guild_data(guild)
|
|
|
|
|
|
# This function will be called when this extension is loaded.
|
|
# It is necessary to add these functions to the bot.
|
|
def setup(bot: commands.Bot):
|
|
"""Adds function to bot"""
|
|
bot.add_cog(PostHighscores(bot))
|