File: permissions.py

package info (click to toggle)
drf-extensions 0.8.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 1,308 kB
  • sloc: python: 7,421; makefile: 11
file content (21 lines) | stat: -rw-r--r-- 754 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
from rest_framework.permissions import DjangoObjectPermissions


class ExtendedDjangoObjectPermissions(DjangoObjectPermissions):
    hide_forbidden_for_read_objects = True

    def has_object_permission(self, request, view, obj):
        if self.hide_forbidden_for_read_objects:
            return super().has_object_permission(request, view, obj)
        else:
            model_cls = getattr(view, 'model', None)
            queryset = getattr(view, 'queryset', None)

            if model_cls is None and queryset is not None:
                model_cls = queryset.model

            perms = self.get_required_object_permissions(
                request.method, model_cls)
            user = request.user

            return user.has_perms(perms, obj)