def update_sous_chef(user, sous_chef): sc = fetch_by_id_or_field(SousChef, 'slug', sous_chef) if not sc: raise NotFoundError( 'A SousChef does not exist with ID/slug {}' .format(sous_chef)) req_data = request_data() # validate the schema new_sous_chef = sous_chef_schema.update(sc, req_data) # udpate for name, value in new_sous_chef.items(): setattr(sc, name, value) db.session.add(sc) try: db.session.commit() except Exception as e: raise RequestError( "An error occurred while updating SousChef '{}'. " "Here's the error message: {}" .format(sc.slug, e.message)) return jsonify(sc)
def update_sous_chef(user, sous_chef): sc = fetch_by_id_or_field(SousChef, 'slug', sous_chef) if not sc: raise NotFoundError( 'A SousChef does not exist with ID/slug {}'.format(sous_chef)) req_data = request_data() # validate the schema new_sous_chef = sous_chef_schema.update(sc, req_data) # udpate for name, value in new_sous_chef.items(): setattr(sc, name, value) db.session.add(sc) try: db.session.commit() except Exception as e: raise RequestError("An error occurred while updating SousChef '{}'. " "Here's the error message: {}".format( sc.slug, e.message)) return jsonify(sc)
def test_partial_update(self): """ Simiulate a partial update process. """ sc = { "name": "Twitter List", "slug": "twitter-list", "description": "Extracts events from a twitter list.", "runs": "newslynx.sc.events.twitter.List", "creates": "events", "options": { "owner_screen_name": { "input_type": "text", "value_types": ["string"], "accepts_list": True, "required": True, "help": { "placeholder": "cspan" }, }, "min_followers": { "input_type": "number", "value_types": ["numeric"], "accepts_list": False, "required": False, "default": 0, "help": { "placeholder": 1000 } } } } sc = sous_chef_schema.validate(sc) sc = SousChef(**sc) db_session.add(sc) db_session.commit() new_sc = { 'name': 'Twitter List to Event', 'options': { 'owner_screen_name': { 'accepts_list': False } } } new_sc = sous_chef_schema.update(sc, new_sc) assert (new_sc['name'] == 'Twitter List to Event') assert (new_sc['options']['owner_screen_name']['accepts_list'] is False) db_session.delete(sc) db_session.commit()
def test_partial_update(self): """ Simiulate a partial update process. """ sc = { "name": "Twitter List", "slug": "twitter-list", "description": "Extracts events from a twitter list.", "runs": "newslynx.sc.events.twitter.List", "creates": "events", "options": { "owner_screen_name": { "input_type": "text", "value_types": ["string"], "accepts_list": True, "required": True, "help": { "placeholder": "cspan" }, }, "min_followers": { "input_type": "number", "value_types": ["numeric"], "accepts_list": False, "required": False, "default": 0, "help": { "placeholder": 1000 } } } } sc = sous_chef_schema.validate(sc) sc = SousChef(**sc) db_session.add(sc) db_session.commit() new_sc = { 'name': 'Twitter List to Event', 'options': { 'owner_screen_name': { 'accepts_list': False } } } new_sc = sous_chef_schema.update(sc, new_sc) assert(new_sc['name'] == 'Twitter List to Event') assert(new_sc['options']['owner_screen_name']['accepts_list'] is False) 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