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__))