def test_partial_update(self): """ Simiulate a partial update process. """ old_recipe = copy.copy(good_recipe) sc = SousChef(**sous_chef) db_session.add(sc) db_session.commit() old_recipe = recipe_schema.validate(old_recipe, sc.to_dict()) old_id = old_recipe['options']['set_content_items'][0]['id'] old_recipe['slug'] += "-{}".format(gen_short_uuid()) r = Recipe(sc, **old_recipe) db_session.add(r) db_session.commit() new_recipe = { 'owner_screen_name': 'johnoliver', 'last_job': { 'foo': 'bar' }, 'status': 'stable', "set_content_items": [{ 'id': 2, 'title': 'foobar' }] } new_recipe = recipe_schema.update(r, new_recipe, sc.to_dict()) assert (new_recipe['options']['owner_screen_name'] == 'johnoliver') assert (new_recipe['last_job']['foo'] == 'bar') assert (new_recipe['status'] == 'stable') assert (new_recipe['options']['set_content_items'][0]['id'] != old_id) db_session.delete(r) db_session.delete(sc) db_session.commit()
def test_partial_update(self): """ Simiulate a partial update process. """ old_recipe = copy.copy(good_recipe) sc = SousChef(**sous_chef) db_session.add(sc) db_session.commit() old_recipe = recipe_schema.validate(old_recipe, sc.to_dict()) old_id = old_recipe['options']['set_content_items'][0]['id'] old_recipe['slug'] += "-{}".format(gen_short_uuid()) r = Recipe(sc, **old_recipe) db_session.add(r) db_session.commit() new_recipe = { 'owner_screen_name': 'johnoliver', 'last_job': {'foo': 'bar'}, 'status': 'stable', "set_content_items": [{'id': 2, 'title': 'foobar'}] } new_recipe = recipe_schema.update( r, new_recipe, sc.to_dict()) assert(new_recipe['options']['owner_screen_name'] == 'johnoliver') assert(new_recipe['last_job']['foo'] == 'bar') assert(new_recipe['status'] == 'stable') assert(new_recipe['options']['set_content_items'][0]['id'] != old_id) db_session.delete(r) db_session.delete(sc) db_session.commit()
def sous_chefs(org): """ (Re)load all sous chefs. """ # load all sous-chefs for sc, fp in init.load_sous_chefs(): sc = sous_chef_schema.validate(sc, fp) sc['org_id'] = org.id sc_obj = db.session.query(SousChef).filter_by(slug=sc['slug']).first() if not sc_obj: log.info('Creating sous chef: "{slug}"'.format(**sc)) sc_obj = SousChef(**sc) else: log.warning('Updating sous chef: "{slug}"'.format(**sc)) sc = sous_chef_schema.update(sc_obj.to_dict(), sc) for name, value in sc.items(): setattr(sc_obj, name, value) db.session.add(sc_obj) db.session.commit() return org
def sous_chefs(org): """ (Re)load all sous chefs. """ # load all sous-chefs for sc, fp in init.load_sous_chefs(): sc = sous_chef_schema.validate(sc, fp) sc['org_id'] = org.id sc_obj = db.session.query(SousChef).filter_by( slug=sc['slug']).first() if not sc_obj: log.info('Creating sous chef: "{slug}"'.format(**sc)) sc_obj = SousChef(**sc) else: log.warning('Updating sous chef: "{slug}"'.format(**sc)) sc = sous_chef_schema.update(sc_obj.to_dict(), sc) for name, value in sc.items(): setattr(sc_obj, name, value) db.session.add(sc_obj) db.session.commit() return org
def run(opts, **kwargs): # create the database try: with app.app_context(): echo('Creating database "{}"'.format(settings.SQLALCHEMY_DATABASE_URI), no_color=opts.no_color) db.configure_mappers() db.create_all() # create the super user u = User.query.filter_by(email=settings.SUPER_USER_EMAIL).first() if not u: echo('Creating super user "{}"'.format(settings.SUPER_USER_EMAIL), no_color=opts.no_color) u = User(name=settings.SUPER_USER, email=settings.SUPER_USER_EMAIL, password=settings.SUPER_USER_PASSWORD, admin=True, super_user=True) # optionally add super user apikey if getattr(settings, 'SUPER_USER_APIKEY', None): u.apikey = settings.SUPER_USER_APIKEY else: echo('Updating super user "{}"'.format(settings.SUPER_USER_EMAIL), no_color=opts.no_color) u.name=settings.SUPER_USER, u.email=settings.SUPER_USER_EMAIL, u.password=settings.SUPER_USER_PASSWORD, u.admin=True super_user=True db.session.add(u) echo('(Re)Loading SQL Extensions', no_color=opts.no_color) # load sql extensions + functions for sql in load_sql(): db.session.execute(sql) # load built-in sous-chefs for sc in load_sous_chefs(): sc = sous_chef_schema.validate(sc) sc_obj = db.session.query(SousChef).filter_by(slug=sc['slug']).first() if not sc_obj: echo('Importing Sous Chef "{}"'.format(sc['slug']), no_color=opts.no_color) sc_obj = SousChef(**sc) else: echo('Updating Sous Chef "{}"'.format(sc['slug']), no_color=opts.no_color) sc = sous_chef_schema.update(sc_obj.to_dict(), sc) # udpate for name, value in sc.items(): setattr(sc_obj, name, value) db.session.add(sc_obj) # commit db.session.commit() db.session.close() except Exception as e: db.session.rollback() db.session.close() raise e