xevion/contest-assistant
1
0
Fork 0
mirror of https://github.com/Xevion/contest-assistant.git synced 2025-12-09 10:06:51 -06:00
Code Issues Packages Projects Releases Wiki Activity

submission channel config command, basic submission filtering

Browse Source
This commit is contained in:
Xevion
2021-02-07 14:29:15 -06:00
parent 9cb226b502
commit b748db0866
2 changed files with 30 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
contest/cogs/contest.py
View File

@@ -1,5 +1,6 @@
import discord
from discord.ext import commands
from discord.ext.commands import Context
from contest import checks
from contest.bot import ContestBot
@@ -17,20 +18,43 @@ class ContestCog(commands.Cog):
cur_prefix = await self.bot.db.get_prefix(ctx.guild.id)
if 1 <= len(new_prefix) <= 2:
if cur_prefix == new_prefix:
return await ctx.send(f'The prefix is already `{new_prefix}`')
return await ctx.send(f':no_entry_sign: The prefix is already `{new_prefix}`.')
else:
await self.bot.db.set_prefix(ctx.guild.id, new_prefix)
return await ctx.send(f':white_check_mark: Prefix changed to `{new_prefix}`')
return await ctx.send(f':white_check_mark: Prefix changed to `{new_prefix}`.')
else:
return await ctx.send(':no_entry_sign: Invalid argument. Prefix must be 1 or 2 characters long.')
@commands.command()
@commands.guild_only()
@checks.privileged()
async def submission(self, ctx, new_submission: discord.TextChannel):
async def submission(self, ctx: Context, new_submission: discord.TextChannel):
"""Changes the bot's saved submission channel."""
pass
cur_submission = await self.bot.db.get_submission_channel(ctx.guild.id)
if cur_submission == new_submission.id:
return await ctx.send(
f':no_entry_sign: The submission channel is already set to {new_submission.mention}.')
else:
await self.bot.db.set_submission_channel(ctx.guild.id, new_submission.id)
return await ctx.send(f':white_check_mark: Submission channel changed to {new_submission.mention}.')
@commands.Cog.listener()
async def on_message(self, message: discord.Message):
if message.author == self.bot.user or message.author.bot or not message.guild: return
cur_submission = await self.bot.db.get_submission_channel(message.guild.id)
if message.channel.id == cur_submission:
attachments = message.attachments
if len(attachments) == 0:
await message.delete(delay=1)
warning = await message.channel.send(
f':no_entry_sign: {message.author.mention} Each submission must contain exactly one image.')
await warning.delete(delay=5)
elif len(attachments) > 1:
await message.delete(delay=1)
warning = await message.channel.send(
f':no_entry_sign: {message.author.mention} Each submission must contain exactly one image.')
await warning.delete(delay=5)
@commands.Cog.listener()
async def on_raw_reaction_add(self, payload: discord.RawReactionActionEvent) -> None:

4
contest/db.py
View File

@@ -76,7 +76,7 @@ class ContestDatabase(object):
finally:
await cur.close()
async def get_submission(self, guild_id: int) -> int:
async def get_submission_channel(self, guild_id: int) -> int:
cur = await self.conn.cursor()
try:
await cur.execute('''SELECT submission FROM guild WHERE id = ?''', [guild_id])
@@ -84,7 +84,7 @@ class ContestDatabase(object):
finally:
await cur.close()
async def set_submission(self, guild_id: int, new_submission: int) -> None:
async def set_submission_channel(self, guild_id: int, new_submission: int) -> None:
"""Updates the submission channel for a specific guild in the database"""
await self.conn.execute('''UPDATE guild SET submission = ? WHERE id = ?''', [new_submission, guild_id])
await self.conn.commit()