thearst3rd/thearstbot-twitch
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Support customizable prefix

Browse source
This commit is contained in:
Terry Hearst 2022-09-12 22:17:49 -04:00
parent d5e1bfbcae
commit 578d067a40
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
bot.py
View file

@ -4,10 +4,12 @@ import os
from twitchio.ext import commands from twitchio.ext import commands
from dotenv import load_dotenv from dotenv import load_dotenv
PREFIX = "!"
class Bot(commands.Bot): class Bot(commands.Bot):
def __init__(self): def __init__(self):
super().__init__(token = os.environ["TMI_TOKEN"], prefix="!", initial_channels=[os.environ["CHANNEL"]]) super().__init__(token = os.environ["TMI_TOKEN"], prefix=PREFIX, initial_channels=[os.environ["CHANNEL"]])
self.custom_commands = {} self.custom_commands = {}
async def event_ready(self): async def event_ready(self):
@ -21,7 +23,7 @@ class Bot(commands.Bot):
# I really want to do this with the add_command function and have no need for this event_message override, but # I really want to do this with the add_command function and have no need for this event_message override, but
# I cannot for the life of me figure out how to make "anonymous" coroutines (like, async lambda or something), # I cannot for the life of me figure out how to make "anonymous" coroutines (like, async lambda or something),
# so I'm just manually handling the command here before passing to the command handler # so I'm just manually handling the command here before passing to the command handler
if message.content[0] == "!": if message.content[0] == PREFIX:
first_token = message.content.split()[0][1:] first_token = message.content.split()[0][1:]
if first_token in self.custom_commands: if first_token in self.custom_commands:
await message.channel.send(self.custom_commands[first_token]) await message.channel.send(self.custom_commands[first_token])
@ -64,7 +66,7 @@ class Bot(commands.Bot):
return return
self.custom_commands[command_name] = command_text self.custom_commands[command_name] = command_text
await ctx.send(f"Adding command: \"!{command_name}\" -> \"{command_text}\"") await ctx.send(f"Adding command: \"{PREFIX}{command_name}\" -> \"{command_text}\"")
def main(): def main():