def create_groups(groups):
"""Add groups to db"""
for group in groups:
grp = Group(
code=group['code'],
active=bool(group['active']),
)
db.session.add(grp)
db.session.commit()
for module in group['modules']:
m = Module.query.filter_by(code=module).first()
grp.add_module_to_group(m)
db.session.commit()
def import_groups_to_db(data):
"""
Writes the group data from the json file to the db
:param data: <dict> Dictionary containing group data
:return: Number of imported groups
"""
for row in data:
grp = Group(code=row['code'], active=row['active'])
db.session.add(grp)
db.session.commit()
for module in row['modules']:
grp.add_module_to_group(
Module.query.filter_by(code=module).first())
db.session.commit()
return str(len(data))
