From 5a81e4b2a1126f2d113c9611347d29c154c36460 Mon Sep 17 00:00:00 2001
From: Funky Waddle <funkywaddle@funkywaddle.com>
Date: Sun, 29 Jun 2025 02:18:58 -0500
Subject: [PATCH] Add helper methods to Db lib

---
 libs/Db.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/libs/Db.py b/libs/Db.py
index e2797f8..d99939b 100644
--- a/libs/Db.py
+++ b/libs/Db.py
@@ -27,3 +27,14 @@ class Db:
         for filename in os.listdir('./data/models'):
             if filename.endswith('.py'):
                 importlib.import_module('.' + filename[:-3], 'data.models')
+
+    def role_exists(self, model: Model, role: str):
+        does_exist = False
+        language = self.session.query(model).filter(model.label == role).first()
+        if language is not None:
+            does_exist = True
+        return does_exist
+
+    def add_record(self, model):
+        self.session.add(model)
+        self.session.commit()
\ No newline at end of file