thearstbot-twitch/bot.py

# Basic twitch bot

import os
from twitchio.ext import commands
from dotenv import load_dotenv

PREFIX = "!"


class Bot(commands.Bot):
	def __init__(self):
		super().__init__(token = os.environ["TMI_TOKEN"], prefix=PREFIX, initial_channels=[os.environ["CHANNEL"]])
		self.custom_commands = {}

	async def event_ready(self):
		print(f"Logged in as | {self.nick}")
		print(f"User id is | {self.user_id}")

	async def event_message(self, message):
		if message.echo:
			return

		# 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),
		# so I'm just manually handling the command here before passing to the command handler
		if message.content[0] == PREFIX:
			first_token = message.content.split()[0][1:]
			if first_token in self.custom_commands:
				await message.channel.send(self.custom_commands[first_token])
				return

		await self.handle_commands(message)

	@commands.command()
	async def hello(self, ctx: commands.Context):
		# Basic hello world command, executed with "!hello". Reply hello to whoever made the command
		response = f"Hello {ctx.author.name}! "
		if ctx.author.is_broadcaster:
			response += "👑"
		else:
			if ctx.author.is_mod:
				response += "⚔"
			elif ctx.author.is_vip:
				reponse += "💎"
			if ctx.author.is_subscriber:
				response += "🤑"
		await ctx.send(response.strip())

	@commands.command()
	async def addcommand(self, ctx: commands.Context):
		# Lets moderators add custom text responses
		if not ctx.author.is_mod:
			await ctx.send("lmao nice try ur not a mod")
			return
		# Not sure if ctx.args is supposed to work but it seems like it doesn't...
		# I want the last arg to not get split anyway, so I do it myself
		args = ctx.message.content.split(" ", 2)
		if len(args) < 3:
			await ctx.send("lmao plz supply 3 args")
			return
		command_name = args[1]
		command_text = args[2]

		if command_name in self.custom_commands or self.get_command(command_name) is not None:
			await ctx.send("lmao that command already exists")
			return

		self.custom_commands[command_name] = command_text
		await ctx.send(f"Adding command: \"{PREFIX}{command_name}\" -> \"{command_text}\"")

	@commands.command()
	async def removecommand(self, ctx: commands.Context):
		if not ctx.author.is_mod:
			await ctx.send("lmao nice try ur not a mod")
			return

		args = ctx.message.content.split(" ", 1)
		if len(args) < 2:
			await ctx.send("lmao wut command")
			return

		command_name = args[1]
		if self.get_command(command_name) is not None:
			await ctx.send("lmao u cant delet this")
			return

		if command_name not in self.custom_commands:
			await ctx.send("lmao wut is that command")
			return

		del self.custom_commands[command_name]
		await ctx.send(f"Deleted command \"{PREFIX}{command_name}\"")

	@commands.command()
	async def help(self, ctx: commands.Context):
		message = "Available commands:"
		for command in self.commands:
			message += f" {PREFIX}{command}"
		for command in self.custom_commands.keys():
			message += f" {PREFIX}{command}"
		await ctx.send(message)


def main():
	load_dotenv()
	bot = Bot()
	bot.run()
	# bot.run() is blocking and will stop execution of any below code here until stopped or closed.


if __name__ == "__main__":
	main()