matfyz-help/discord-bot
2
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

rf: command argument checks in command function

Browse source
This commit is contained in:
Lukáš Nedbálek 2024-09-28 00:33:39 +02:00
parent a4cf15ce1d
commit 16f26119aa
1 changed files with 11 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
main.py
View file

@ -52,7 +52,17 @@ async def on_ready():
@bot.command() @bot.command()
async def grant_role(ctx, study_group_name: str): async def grant_role(ctx, study_group_name: str):
author_id = ctx.author.id # Check study group format
if not words[1].startswith('kruh-') or not words[1][5:].isdigit():
reply_error(ctx, 'Invalid study group format.')
reply(ctx, HELP_MESSAGE)
return False
# Check UKCO format
if not words[2].isdigit() or not len(words[2]) == 8:
reply_error(ctx, 'Invalid UKCO format.')
reply(ctx, HELP_MESSAGE)
return False
for guild_id in server_ids: for guild_id in server_ids:
guild = bot.get_guild(guild_id) guild = bot.get_guild(guild_id)
@ -104,17 +114,6 @@ def check_command(message):
reply(message.channel, HELP_MESSAGE) reply(message.channel, HELP_MESSAGE)
return False return False
# Check study group format
if not words[1].startswith('kruh-') or not words[1][5:].isdigit():
reply_error(message.channel, 'Invalid study group format.')
reply(message.channel, HELP_MESSAGE)
return False
if not words[2].isdigit() or not len(words[2]) == 8:
reply_error(message.channel, 'Invalid UKCO format.')
reply(message.channel, HELP_MESSAGE)
return False
return True return True