First commit. Lots of stuff

2 files changed, 28 insertions, 0 deletions

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()