didier/functions/database/custom_commands.py

182 lines
4.6 KiB
Python
Raw Normal View History

2021-05-17 18:14:33 +02:00
from data.database_classes.custom_commands import CustomCommand
2021-05-17 20:10:07 +02:00
import discord
2021-05-17 18:14:33 +02:00
from functions.database import utils
2021-05-17 20:10:07 +02:00
def is_custom_command(message: discord.Message) -> CustomCommand:
2021-05-17 18:14:33 +02:00
"""
Check if a message triggers a custom command
These use "?" as a prefix
"""
2021-05-17 20:10:07 +02:00
content = message.content
2021-05-17 18:14:33 +02:00
# Message didn't call a custom command
2021-05-17 20:10:07 +02:00
if not content.startswith("?"):
return CustomCommand()
# Can't be invoked by bots to prevent spam (@RPS)
if message.author.bot:
2021-05-17 18:14:33 +02:00
return CustomCommand()
# Ignore capitals & spaces, strip off prefix
2021-05-17 20:10:07 +02:00
content = content.lower().replace(" ", "")[1:]
2021-05-17 18:14:33 +02:00
2021-05-17 20:10:07 +02:00
by_name = _find_by_name(content)
2021-05-17 18:14:33 +02:00
# Command was found by its name
if by_name:
return CustomCommand(*by_name)
# Check if a command exists with this alias instead
2021-05-17 20:10:07 +02:00
return CustomCommand(*_find_by_alias(content), alias_used=content)
2021-05-17 18:14:33 +02:00
def _find_by_name(message):
"""
Find a command by its name
"""
connection = utils.connect()
cursor = connection.cursor()
2021-05-17 20:10:07 +02:00
cursor.execute("SELECT * FROM custom_commands WHERE name = %s", (message,))
2021-05-17 18:14:33 +02:00
return cursor.fetchone()
def _find_by_alias(message):
"""
Find a command by one of its aliases
"""
connection = utils.connect()
cursor = connection.cursor()
2021-05-17 20:10:07 +02:00
cursor.execute("SELECT command FROM custom_command_aliases WHERE alias = %s", (message,))
2021-05-17 18:14:33 +02:00
res = cursor.fetchone()
# No command matched this alias
if not res:
return ()
2021-05-17 20:10:07 +02:00
cursor.execute("SELECT * FROM custom_commands WHERE id = %s", (res,))
2021-05-17 18:14:33 +02:00
return cursor.fetchone()
def is_name_free(name) -> bool:
"""
Check if a name is already in use by a command
Includes aliases
"""
connection = utils.connect()
cursor = connection.cursor()
2021-05-17 20:10:07 +02:00
cursor.execute("SELECT id from custom_commands WHERE name = %s", (name,))
2021-05-17 18:14:33 +02:00
if cursor.fetchone():
return False
2021-05-17 20:10:07 +02:00
cursor.execute("SELECT id FROM custom_command_aliases WHERE alias = %s", (name,))
return cursor.fetchone() is None
def _clean(inp: str):
"""
Strip markdown and other stuff out of a command name
"""
return "".join(filter(lambda x: x.isalnum(), inp))
def add_command(name: str, response: str):
"""
Add a new custom command
"""
name = _clean(name.lower())
if not is_name_free(name):
return "Er is al een commando met deze naam."
connection = utils.connect()
cursor = connection.cursor()
cursor.execute("INSERT INTO custom_commands(name, response) VALUES (%s, E%s)", (name, response,))
connection.commit()
def add_alias(command: str, alias: str):
"""
Add an alias for a command
Assumes the command exists
"""
command = _clean(command.lower())
alias = _clean(alias.lower())
# Base command doesn't exist
if is_name_free(command):
return "Er is geen commando met deze naam."
# Alias already exists
if not is_name_free(alias):
return "Er is al een commando met deze naam."
# Find the id of the base command
command_id = CustomCommand(*_find_by_name(command)).id
connection = utils.connect()
cursor = connection.cursor()
2021-05-17 18:14:33 +02:00
2021-05-17 20:10:07 +02:00
cursor.execute("INSERT INTO custom_command_aliases(command, alias) VALUES(%s, %s)", (command_id, alias,))
connection.commit()
2021-06-19 17:02:42 +02:00
def get_all():
"""
Return a list of all registered custom commands
"""
connection = utils.connect()
cursor = connection.cursor()
cursor.execute("SELECT * FROM custom_commands")
commands = cursor.fetchall()
ret = []
# Create a list of all entries
for command in commands:
dic = {"id": command[0], "name": command[1], "response": command[2]}
# Find and add aliases
cursor.execute("SELECT id, alias FROM custom_command_aliases WHERE command = %s", (command[0],))
aliases = cursor.fetchall()
if aliases:
dic["aliases"] = list(map(lambda x: {"id": x[0], "alias": x[1]}, aliases))
ret.append(dic)
return ret
def get_by_id(command_id: int):
"""
Return a command that matches a given id
"""
connection = utils.connect()
cursor = connection.cursor()
cursor.execute("SELECT * FROM custom_commands WHERE id = %s", (command_id,))
command = cursor.fetchone()
# Nothing found
if not command:
return None
dic = {"id": command[0], "name": command[1], "response": command[2]}
cursor.execute("SELECT id, alias FROM custom_command_aliases WHERE command = %s", (command_id,))
aliases = cursor.fetchall()
if aliases:
dic["aliases"] = list(map(lambda x: {"id": x[0], "alias": x[1]}, aliases))
return dic