From b5099937f84e2fb58b69d97b4b8fff17363fe7d9 Mon Sep 17 00:00:00 2001 From: Parker Date: Wed, 2 Apr 2025 18:52:00 -0500 Subject: First commit. Lots of stuff --- src/utils/command_tree.py | 19 +++++++++++++++++++ src/utils/db.py | 9 +++++++++ 2 files changed, 28 insertions(+) create mode 100644 src/utils/command_tree.py create mode 100644 src/utils/db.py (limited to 'src/utils') diff --git a/src/utils/command_tree.py b/src/utils/command_tree.py new file mode 100644 index 0000000..31ec2c1 --- /dev/null +++ b/src/utils/command_tree.py @@ -0,0 +1,19 @@ +import discord +from discord import app_commands +from discord.ext.commands.errors import * + +from config import create_embed + + +class Tree(app_commands.CommandTree): + async def on_error( + self, + interaction: discord.Interaction, + error: app_commands.AppCommandError, + ): + + if isinstance(error, CommandNotFound): + return + + else: + raise error diff --git a/src/utils/db.py b/src/utils/db.py new file mode 100644 index 0000000..a6734ea --- /dev/null +++ b/src/utils/db.py @@ -0,0 +1,9 @@ +from database import SessionLocal + + +def get_db(): + db = SessionLocal() + try: + yield db + finally: + db.close() -- cgit v1.2.3-70-g09d2