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'):
|
for filename in os.listdir('./data/models'):
|
||||||
if filename.endswith('.py'):
|
if filename.endswith('.py'):
|
||||||
importlib.import_module('.' + filename[:-3], 'data.models')
|
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