Add helper methods to Db lib

This commit is contained in:
Funky Waddle 2025-06-29 02:18:58 -05:00
parent 881c900e6d
commit 5a81e4b2a1

View file

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