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