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