def create_detail(self, object_list, bundle): if isinstance(bundle.obj, Business): business = bundle.obj elif hasattr(bundle.obj, 'business'): business = getattr(bundle.obj, 'business') else: raise exceptions.BadResourceClass("Model %s.%s need foreign key 'business'" % ( bundle.obj.__class__._meta.app_label, bundle.obj.__class__.__name__)) return self._get_business_for_user(bundle.request.user, perms=['manage_business'] ).filter(pk=business.pk).exists()
def _get_objects_for_user(self, object_list, bundle, perms): user = bundle.request.user if isinstance(bundle.obj, Business): return object_list.filter( pk__in=self._get_business_for_user(user, perms)) elif hasattr(bundle.obj, 'business_id'): return object_list.filter( business_id__in=self._get_business_for_user(user, perms)) else: raise exceptions.BadResourceClass( "Model %s.%s need foreign key 'business'" % (bundle.obj.__class__._meta.app_label, bundle.obj.__class__.__name__))