mirror of
https://github.com/wlinator/luminara.git
synced 2024-10-02 22:23:13 +00:00
76 lines
2.8 KiB
Python
76 lines
2.8 KiB
Python
import mysql.connector
|
|
from discord.ext import commands
|
|
|
|
from db import database
|
|
from lib.const import CONST
|
|
from lib.format import shorten
|
|
from ui.embeds import Builder
|
|
|
|
|
|
class Sql(commands.Cog):
|
|
def __init__(self, bot: commands.Bot):
|
|
self.bot = bot
|
|
|
|
@commands.command(name="sqlselect", aliases=["sqls"])
|
|
@commands.is_owner()
|
|
async def select_cmd(self, ctx: commands.Context[commands.Bot], *, query: str) -> None:
|
|
if query.lower().startswith("select "):
|
|
query = query[7:]
|
|
|
|
try:
|
|
results = database.select_query(f"SELECT {query}")
|
|
embed = Builder.create_embed(
|
|
theme="success",
|
|
user_name=ctx.author.name,
|
|
author_text=CONST.STRINGS["admin_sql_select_title"],
|
|
description=CONST.STRINGS["admin_sql_select_description"].format(
|
|
shorten(query, 200),
|
|
shorten(str(results), 200),
|
|
),
|
|
hide_name_in_description=True,
|
|
)
|
|
except mysql.connector.Error as error:
|
|
embed = Builder.create_embed(
|
|
theme="error",
|
|
user_name=ctx.author.name,
|
|
author_text=CONST.STRINGS["admin_sql_select_error_title"],
|
|
description=CONST.STRINGS["admin_sql_select_error_description"].format(
|
|
shorten(query, 200),
|
|
shorten(str(error), 200),
|
|
),
|
|
hide_name_in_description=True,
|
|
)
|
|
|
|
await ctx.send(embed=embed, ephemeral=True)
|
|
|
|
@commands.command(name="sqlinject", aliases=["sqli"])
|
|
@commands.is_owner()
|
|
async def inject_cmd(self, ctx: commands.Context[commands.Bot], *, query: str) -> None:
|
|
try:
|
|
database.execute_query(query)
|
|
embed = Builder.create_embed(
|
|
theme="success",
|
|
user_name=ctx.author.name,
|
|
author_text=CONST.STRINGS["admin_sql_inject_title"],
|
|
description=CONST.STRINGS["admin_sql_inject_description"].format(
|
|
shorten(query, 200),
|
|
),
|
|
hide_name_in_description=True,
|
|
)
|
|
except mysql.connector.Error as error:
|
|
embed = Builder.create_embed(
|
|
theme="error",
|
|
user_name=ctx.author.name,
|
|
author_text=CONST.STRINGS["admin_sql_inject_error_title"],
|
|
description=CONST.STRINGS["admin_sql_inject_error_description"].format(
|
|
shorten(query, 200),
|
|
shorten(str(error), 200),
|
|
),
|
|
hide_name_in_description=True,
|
|
)
|
|
|
|
await ctx.send(embed=embed, ephemeral=True)
|
|
|
|
|
|
async def setup(bot: commands.Bot) -> None:
|
|
await bot.add_cog(Sql(bot))
|