jr/discord
1
0
Fork 0
mirror of https://github.com/runyanjake/discord.git synced 2025-10-04 15:27:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

Refactor some code to a message_handler to clean up bot definition

Browse Source
This commit is contained in:
whitney 2024-05-16 14:24:26 -07:00
parent 4ce5873711
commit 426ab7bf89
2 changed files with 44 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
pwsBot/bot.py
View File

@ -1,17 +1,9 @@
import discord
import os
import re
import sys
from resource.credentials import get_token
from resource.intents import get_intents
from commands.echo import echo
from commands.hello_world import hello_world
pws_command_prefix = '/pws '
pws_command_regex = r'/pws (\S+)'
pws_command_with_args_regex = r'/pws \S+(.*)'
from commands.message_handler import handle_message
intents = get_intents()
client = discord.Client(intents=intents)
@ -23,38 +15,6 @@ async def on_ready():
@client.event
async def on_message(message):
print(message.content)
if message.author == client.user:
return
if message.content.startswith(pws_command_prefix):
msg = [3]
msg.append(pws_command_prefix)
msg.append(re.match(pws_command_regex, message.content)[1])
msg.append(re.match(pws_command_with_args_regex, message.content)[1])
command = msg[2]
args = msg[3]
print("command: " + command + ", args: " + args)
match command:
case 'helloworld':
if args:
print('Command ' + command + ' got unexpected arguments: ' + message.content + '.')
await message.author.send('Unexpected arguments for command: `' + message.content + '`. Check list of commands for valid usage.')
else:
await hello_world(message)
case 'echo':
if not args:
print('Command ' + command + ' did not specify argument(s): ' + message.content + '.')
await message.author.send('Did not specify argument(s) for command: `' + message.content + '`. Check list of commands for valid usage.')
else:
await echo(message, args)
case _:
print('Invalid command ' + command + ' received.')
await message.author.send('Invalid command `' + command + '`. Check list of commands for valid usage.')
else:
print('Error parsing command')
await message.author.send('Error parsing command:\n`' + message.content + '`\nMessage format is "`' + pws_command_prefix + '` command (args...)"')
await handle_message(client, message)
client.run(client_token)

42
pwsBot/commands/message_handler.py Normal file
View File

@ -0,0 +1,42 @@
import re
from commands.echo import echo
from commands.hello_world import hello_world
pws_command_prefix = '/pws '
pws_command_regex = r'/pws (\S+)'
pws_command_with_args_regex = r'/pws \S+(.*)'
async def handle_message(client, message):
print(message.content)
#Ignore messages this bot has sent to avoid answering itself.
if message.author == client.user:
return
#Respond only to messages starting with the PWS prefix.
if message.content.startswith(pws_command_prefix):
msg = [3]
msg.append(pws_command_prefix)
msg.append(re.match(pws_command_regex, message.content)[1])
msg.append(re.match(pws_command_with_args_regex, message.content)[1])
command = msg[2]
args = msg[3]
print("command: " + command + ", args: " + args)
match command:
case 'helloworld':
if args:
print('Command ' + command + ' got unexpected arguments: ' + message.content + '.')
await message.author.send('Unexpected arguments for command: `' + message.content + '`. Check list of commands for valid usage.')
else:
await hello_world(message)
case 'echo':
if not args:
print('Command ' + command + ' did not specify argument(s): ' + message.content + '.')
await message.author.send('Did not specify argument(s) for command: `' + message.content + '`. Check list of commands for valid usage.')
case _:
print('Invalid command ' + command + ' received.')
await message.author.send('Invalid command `' + command + '`. Check list of commands for valid usage.')