def add_trashtalk(guild_id: int, added_on: str, added_by_user_id: int,
message: str):
sql = "INSERT INTO TRASHTALK (SERVER_ID, ADDED_ON, ADDED_BY_USER_ID, MESSAGE) VALUES ((SELECT ID FROM CONFIG WHERE GUILD_ID=%s), %s, %s, %s)"
val = (guild_id, added_on, added_by_user_id, message)
cur_main.execute(sql, val)
conn_main.commit()
def change_msg_welcome_channel(guild_id: int, main_channel_id: int,
welcome_channel_id: int):
sql = "UPDATE CONFIG SET MESSAGE_CHANNEL_ID=%s, WELCOME_CHANNEL_ID=%s WHERE GUILD_ID=%s"
val_1 = (main_channel_id, welcome_channel_id, guild_id)
cur_main.execute(sql, val_1)
conn_main.commit()
def log_trashtalk(guild_id: int, datum: str, from_user_id: int,
to_user_id: int):
sql = "INSERT INTO TRASHTALK_LOG (SERVER_ID, DATE, FROM_USER_ID, TO_USER_ID) VALUES ((SELECT ID FROM CONFIG WHERE GUILD_ID=%s), %s, %s, %s)"
val = (guild_id, datum, from_user_id, to_user_id)
cur_main.execute(sql, val)
conn_main.commit()
def change_welcome_message(guild_id: int, welcome_msg: str):
sql = "UPDATE CONFIG SET WELCOME_MESSAGE=%s WHERE GUILD_ID=%s"
val = (welcome_msg, guild_id)
cur_main.execute(sql, val)
conn_main.commit()
def create_command(category: str, command: str, parameters: str,
description: str):
sql = "INSERT INTO COMMANDS (CATEGORY_ID, COMMAND, PARAMETERS, DESCRIPTION) VALUES ((SELECT ID FROM COMMAND_CATEGORIES WHERE CATEGORY=%s), %s, %s, %s)"
val = (category, command, parameters, description)
cur_main.execute(sql, val)
conn_main.commit()
def insert_user_xp(guild_id: int, user_id: int, xp: int):
sql = "INSERT INTO LEVEL (SERVER_ID, USER_ID, XP) VALUES ((SELECT ID FROM CONFIG WHERE GUILD_ID=%s), %s, %s)"
val_1 = (guild_id, user_id, xp)
try:
cur_main.execute(sql, val_1)
conn_main.commit()
except IntegrityError:
raise InvalidGuild
def update_user_xp(guild_id: int, user_id: int, xp: int):
cur_main.execute("DELETE FROM LEVEL WHERE SERVER_ID=(SELECT ID FROM CONFIG WHERE GUILD_ID=%s) AND USER_ID=%s", (guild_id, user_id))
conn_main.commit()
sql = "INSERT INTO LEVEL (SERVER_ID, USER_ID, XP) VALUES ((SELECT ID FROM CONFIG WHERE GUILD_ID=%s), %s, %s)"
val = (guild_id, user_id, xp)
cur_main.execute(sql, val)
conn_main.commit()
def enable_command(guild_id: int, command: str):
if check_command_status_for_guild(guild_id, command):
return
sql = "DELETE FROM DISABLED_COMMANDS WHERE COMMAND_ID=(SELECT ID FROM COMMANDS WHERE COMMAND=%s) AND SERVER_ID=(SELECT ID FROM CONFIG WHERE GUILD_ID=%s)"
val = (command, guild_id)
cur_main.execute(sql, val)
conn_main.commit()
def disable_command(guild_id: int, command: str):
cur_main.execute("SELECT ID FROM COMMANDS WHERE COMMAND=%s", (command, ))
command_in_db = cur_main.fetchone()
if command_in_db:
sql = "INSERT INTO DISABLED_COMMANDS (SERVER_ID, COMMAND_ID) VALUES ((SELECT ID FROM CONFIG WHERE GUILD_ID=%s), %s)"
val = (guild_id, command_in_db[0])
cur_main.execute(sql, val)
conn_main.commit()
def create_admin_command(command: str, parameters: str, description: str):
sql = "INSERT INTO ADMIN_COMMANDS (COMMAND, PARAMETERS, DESCRIPTION) VALUES (%s, %s, %s);"
val = (command, parameters, description)
try:
cur_main.execute(sql, val)
conn_main.commit()
return 0
except Exception as e:
return e
def delete_command(command: str = "", command_id: int = 0):
if command:
sql = "DELETE FROM COMMANDS WHERE COMMAND=%s"
val = (command, )
elif command_id:
sql = "DELETE FROM COMMANDS WHERE ID=%s"
val = (command_id, )
else:
# TODO Create Error
raise Exception
cur_main.execute(sql, val)
conn_main.commit()
def reset_trashtalk(guild_id: int, user_id: int):
sql = "DELETE FROM TRASHTALK_LOG WHERE SERVER_ID=(SELECT ID FROM CONFIG WHERE GUILD_ID=%s) AND FROM_USER_ID=%s"
val = (guild_id, user_id)
cur_main.execute(sql, val)
conn_main.commit()
def setup_config(guild_id: int, main_channel_id: int, welcome_channel_id: int):
sql = "INSERT INTO CONFIG (ACTIVE, GUILD_ID, LANGUAGE, PREFIX, MESSAGE_CHANNEL_ID, WELCOME_CHANNEL_ID) VALUES (%s, %s, %s, %s, %s, %s)"
val = (True, guild_id, "german", "$", main_channel_id, welcome_channel_id)
cur_main.execute(sql, val)
conn_main.commit()
def log_invite(guild_id: int, date: str, from_user_id: int, to_user_id: int):
sql = "INSERT INTO INVITES (SERVER_ID, DATE, FROM_USER_ID, TO_USER_ID) VALUES ((SELECT ID FROM CONFIG WHERE GUILD_ID=%s), %s, %s, %s)"
val = (guild_id, date, from_user_id, to_user_id)
cur_main.execute(sql, val)
conn_main.commit()
def delete_admin_command(command: str):
sql = "DELETE FROM ADMIN_COMMANDS WHERE COMMAND=%s"
val = (command, )
cur_main.execute(sql, val)
conn_main.commit()
def log_message(guild_id: int, date: str, user_id: int, message: str):
sql = "INSERT INTO MESSAGE (SERVER_ID, DATE, USER_ID, MESSAGE) VALUES ((SELECT ID FROM CONFIG WHERE GUILD_ID=%s), %s, %s, %s)"
val = (guild_id, date, user_id, message)
cur_main.execute(sql, val)
conn_main.commit()
def add_user_voice_time(guild_id: int, user_id: int, minutes: int):
sql = "INSERT INTO VOICE (SERVER_ID, USER_ID, MINUTES) VALUES ((SELECT ID FROM CONFIG WHERE GUILD_ID=%s), %s, %s)"
val = (guild_id, user_id, minutes)
cur_main.execute(sql, val)
conn_main.commit()
def update_twitch_username(guild_id: int, twitch_username: str):
cur_main.execute(
'UPDATE CONFIG SET TWITCH_USERNAME = %s WHERE GUILD_ID = %s',
(twitch_username, guild_id))
conn_main.commit()
def delete_all_configs():
cur_main.execute("TRUNCATE TABLE config;")
conn_main.commit()
def change_auto_role(guild_id: int, role_id: int):
sql = "UPDATE CONFIG SET WELCOME_ROLE_ID = %s WHERE GUILD_ID = %s"
val = (role_id, guild_id)
cur_main.execute(sql, val)
conn_main.commit()
def activate_guild(guild_id: int):
sql = "UPDATE CONFIG SET ACTIVE = TRUE WHERE GUILD_ID = %s"
val = (guild_id, )
cur_main.execute(sql, val)
conn_main.commit()
def change_prefix(guild_id: int, prefix: str):
sql = "UPDATE CONFIG SET PREFIX = %s WHERE GUILD_ID = %s"
val = (prefix, guild_id)
cur_main.execute(sql, val)
conn_main.commit()