def get_queryset(self): """Return the right base queryset depending on the situation. Note that permissions checks still apply on top of that, against the add-on as per check_object_permissions() above.""" requested = self.request.GET.get('filter') # By default we restrict to valid versions. However: # # When accessing a single version or if requesting it explicitly when # listing, admins can access all versions, including deleted ones. should_access_all_versions_included_deleted = ( (requested == 'all_with_deleted' or self.action != 'list') and self.request.user.is_authenticated() and self.request.user.is_staff) # When accessing a single version or if requesting it explicitly when # listing, reviewers and add-on authors can access all non-deleted # versions. should_access_all_versions = ( (requested == 'all' or self.action != 'list') and (AllowReviewer().has_permission(self.request, self) or AllowAddonAuthor().has_object_permission( self.request, self, self.get_addon_object()))) # Everyone can see (non deleted) beta version when they request it # explicitly. should_access_only_beta_versions = (requested == 'beta_only') if should_access_all_versions_included_deleted: self.queryset = Version.unfiltered.all() elif should_access_all_versions: self.queryset = Version.objects.all() elif should_access_only_beta_versions: self.queryset = Version.objects.filter( files__status=amo.STATUS_BETA).distinct() # Now that the base queryset has been altered, call super() to use it. qs = super(AddonVersionViewSet, self).get_queryset() # Filter with the add-on. return qs.filter(addon=self.get_addon_object())
def setUp(self): self.permission = AllowReviewer() self.request_factory = RequestFactory() self.unsafe_methods = ('patch', 'post', 'put', 'delete') self.safe_methods = ('get', 'options', 'head')