Add helper methods to Db lib
This commit is contained in:
parent
881c900e6d
commit
5a81e4b2a1
11
libs/Db.py
11
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()
|
||||
Loading…
Reference in a new issue