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

Persistently store custom commands

Browse source
This commit is contained in:
Terry Hearst 2022-09-13 23:00:59 -04:00
parent 59a65f778e
commit 98a02adc2f
2 changed files with 22 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
.gitignore vendored
View file

@ -1,3 +1,6 @@
# SQLite database
storage.db
# Byte-compiled / optimized / DLL files
__pycache__/
*.py[cod]

19
bot.py
View file

@ -1,16 +1,31 @@
# Basic twitch bot
import os
import sqlite3
from twitchio.ext import commands
from dotenv import load_dotenv
PREFIX = "!"
DB_PATH = "storage.db"
class Bot(commands.Bot):
def __init__(self):
super().__init__(token = os.environ["TMI_TOKEN"], prefix=PREFIX, initial_channels=[os.environ["CHANNEL"]])
self.custom_commands = {}
self.db_con = sqlite3.connect(DB_PATH)
self.db_cur = self.db_con.cursor()
if self.db_cur.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='custom_commands'").fetchone() == None:
print("Created database table")
self.db_cur.execute("CREATE TABLE custom_commands(command, output_text)")
commands = self.db_cur.execute("SELECT * FROM custom_commands").fetchall()
for command in commands:
self.custom_commands[command[0]] = command[1]
if len(self.custom_commands) > 0:
print("Loaded custom commands:")
print(self.custom_commands)
async def event_ready(self):
print(f"Logged in as | {self.nick}")
@ -66,6 +81,8 @@ class Bot(commands.Bot):
return
self.custom_commands[command_name] = command_text
self.db_cur.execute("INSERT INTO custom_commands VALUES (?, ?)", [command_name, command_text])
self.db_con.commit()
await ctx.send(f"Adding command: \"{PREFIX}{command_name}\" -> \"{command_text}\"")
@commands.command()
@ -89,6 +106,8 @@ class Bot(commands.Bot):
return
del self.custom_commands[command_name]
self.db_cur.execute("DELETE FROM custom_commands WHERE command=?", [command_name])
self.db_con.commit()
await ctx.send(f"Deleted command \"{PREFIX}{command_name}\"")
@commands.command()