def create(session, func_key): func_key_row = db_converter.create_func_key_row(func_key) with commit_or_abort(session, ElementCreationError, 'FuncKey'): session.add(func_key_row) func_key.id = func_key_row.id destination_row = db_converter.create_destination_row(func_key) with commit_or_abort(session, ElementCreationError, 'FuncKey'): session.add(destination_row) return func_key
def create(session, func_key): func_key_row = db_converter.create_func_key_row(func_key) with commit_or_abort(session, DataError.on_create, 'FuncKey'): session.add(func_key_row) func_key.id = func_key_row.id destination_row = db_converter.create_destination_row(func_key) with commit_or_abort(session, DataError.on_create, 'FuncKey'): session.add(destination_row) return func_key
def create(session, incall): extension = extension_dao.get(incall.extension_id) incall_row = incall_db_converter.to_incall(incall, extension) with commit_or_abort(session, ElementCreationError, 'Incall'): session.add(incall_row) incall.id = incall_row.id dialaction_row = incall_db_converter.to_dialaction(incall) with commit_or_abort(session, ElementCreationError, 'Incall'): session.add(dialaction_row) return incall
def associate_destination(session, extension_id, destination, destination_id): updated_row = {'type': destination, 'typeval': str(destination_id)} with commit_or_abort(session, ElementEditionError, 'Extension'): (session.query(ExtensionSchema) .filter(ExtensionSchema.id == extension_id) .update(updated_row))
def create_private_template(session): template = FuncKeyTemplateSchema(private=True) with commit_or_abort(session, ElementCreationError, 'FuncKeyTemplate'): session.add(template) return template.id
def create_private_template(session): template = FuncKeyTemplateSchema(private=True) with commit_or_abort(session, DataError.on_create, 'FuncKeyTemplate'): session.add(template) return template.id
def delete_private_template(session, template_id): with commit_or_abort(session, ElementDeletionError, 'FuncKeyTemplate'): (session.query(FuncKeyMappingSchema).filter( FuncKeyMappingSchema.template_id == template_id).delete()) (session.query(FuncKeyTemplateSchema).filter( FuncKeyTemplateSchema.id == template_id).delete())
def edit(session, user): user_row = _fetch_commented_user_row(session, user.id) db_converter.update_source(user_row, user) with commit_or_abort(session, DataError.on_edit, 'User'): session.add(user_row)
def delete(session, func_key): helper = QueryHelper(session) func_key_query = helper.delete_func_key(func_key.id) destination_query = helper.delete_destination(func_key.destination, func_key.destination_id) with commit_or_abort(session, DataError.on_delete, 'FuncKey'): destination_query.delete() func_key_query.delete()
def delete_private_template(session, template_id): with commit_or_abort(session, ElementDeletionError, 'FuncKeyTemplate'): (session.query(FuncKeyMappingSchema) .filter(FuncKeyMappingSchema.template_id == template_id) .delete()) (session.query(FuncKeyTemplateSchema) .filter(FuncKeyTemplateSchema.id == template_id) .delete())
def delete(session, func_key): helper = QueryHelper(session) func_key_query = helper.delete_func_key(func_key.id) destination_query = helper.delete_destination(func_key.destination, func_key.destination_id) with commit_or_abort(session, ElementDeletionError, 'FuncKey'): destination_query.delete() func_key_query.delete()
def create(session, user): user_row = db_converter.to_source(user) user_row.entityid = entity_dao.default_entity_id() with commit_or_abort(session, DataError.on_create, 'User'): session.add(user_row) user.id = user_row.id return user
def delete(session, incall): incall_query = (session.query(Incall).filter(Incall.id == incall.id)) dialaction_query = (session.query(Dialaction).filter( Dialaction.category == 'incall').filter( Dialaction.categoryval == str(incall.id))) with commit_or_abort(session, ElementDeletionError, 'Incall'): incall_query.delete() dialaction_query.delete()
def test_when_adding_a_row_then_row_is_committed(self): row = Mock() session = Mock() error = Mock() element = Mock() with db_utils.commit_or_abort(session, error, element): session.add(row) session.begin.assert_called_once_with() session.commit.assert_called_once_with()
def delete(session, incall): incall_query = (session.query(Incall) .filter(Incall.id == incall.id)) dialaction_query = (session.query(Dialaction) .filter(Dialaction.category == 'incall') .filter(Dialaction.categoryval == str(incall.id))) with commit_or_abort(session, ElementDeletionError, 'Incall'): incall_query.delete() dialaction_query.delete()
def test_given_row_added_when_error_then_rollback_and_error_raised(self): row = Mock() session = Mock() session.commit.side_effect = SQLAlchemyError() element = Mock() with self.assertRaises(CustomException): with db_utils.commit_or_abort(session, CustomException, element): session.add(row) session.begin.assert_called_once_with() session.rollback.assert_called_once_with()
def create_default_dial_actions_for_user(session, user): with commit_or_abort(session, ElementCreationError, 'Dialaction'): for event in USER_EVENTS: dialaction = _create_user_dial_action(user.id, event) session.add(dialaction)
def remove_func_key_from_templates(session, func_key): with commit_or_abort(session, ElementDeletionError, 'FuncKeyTemplate'): (session.query(FuncKeyMappingSchema) .filter(FuncKeyMappingSchema.func_key_id == func_key.id) .delete())
def create_default_dial_actions_for_user(session, user): with commit_or_abort(session, DataError.on_create, 'Dialaction'): for event in USER_EVENTS: dialaction = _create_user_dial_action(user.id, event) session.add(dialaction)
def associate_destination(session, extension_id, destination, destination_id): updated_row = {'type': destination, 'typeval': str(destination_id)} with commit_or_abort(session, ElementEditionError, 'Extension'): (session.query(ExtensionSchema).filter( ExtensionSchema.id == extension_id).update(updated_row))
def remove_func_key_from_templates(session, func_key): with commit_or_abort(session, ElementDeletionError, 'FuncKeyTemplate'): (session.query(FuncKeyMappingSchema).filter( FuncKeyMappingSchema.func_key_id == func_key.id).delete())
def delete(session, user): with commit_or_abort(session, DataError.on_delete, 'User'): _delete_user(session, user.id)