def delete(cls, instance): node_ids = [ _id for (_id, ) in db().query(models.Node.id).filter_by( cluster_id=instance.id).order_by(models.Node.id) ] fire_callback_on_node_collection_delete(node_ids) fire_callback_on_cluster_delete(instance) super(Cluster, cls).delete(instance)
def delete(cls, instance): node_ids = [ _id for (_id,) in db().query(models.Node.id). filter_by(cluster_id=instance.id). order_by(models.Node.id)] fire_callback_on_node_collection_delete(node_ids) fire_callback_on_cluster_delete(instance) super(Cluster, cls).delete(instance)
def test_fire_callback_on_node_collection_delete(self, get_m): node_ids = [1, 2, 3, 4] fire_callback_on_node_collection_delete(node_ids) for ext in get_m.return_value: ext.on_node_collection_delete.assert_called_once_with(node_ids)
def delete_by_ids(cls, ids): fire_callback_on_node_collection_delete(ids) db.query(cls.single.model).filter( cls.single.model.id.in_(ids)).delete(synchronize_session=False)