mirror of
				https://github.com/paperless-ngx/paperless-ngx.git
				synced 2025-11-03 19:17:13 -05:00 
			
		
		
		
	Object-level permissions + filtering
This commit is contained in:
		
							parent
							
								
									dbaa606a9f
								
							
						
					
					
						commit
						fad13c148e
					
				@ -2,6 +2,7 @@ from django.db.models import Q
 | 
				
			|||||||
from django_filters.rest_framework import BooleanFilter
 | 
					from django_filters.rest_framework import BooleanFilter
 | 
				
			||||||
from django_filters.rest_framework import Filter
 | 
					from django_filters.rest_framework import Filter
 | 
				
			||||||
from django_filters.rest_framework import FilterSet
 | 
					from django_filters.rest_framework import FilterSet
 | 
				
			||||||
 | 
					from rest_framework_guardian.filters import ObjectPermissionsFilter
 | 
				
			||||||
 | 
					
 | 
				
			||||||
from .models import Correspondent
 | 
					from .models import Correspondent
 | 
				
			||||||
from .models import Document
 | 
					from .models import Document
 | 
				
			||||||
@ -134,3 +135,17 @@ class StoragePathFilterSet(FilterSet):
 | 
				
			|||||||
            "name": CHAR_KWARGS,
 | 
					            "name": CHAR_KWARGS,
 | 
				
			||||||
            "path": CHAR_KWARGS,
 | 
					            "path": CHAR_KWARGS,
 | 
				
			||||||
        }
 | 
					        }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					class ObjectOwnedOrGrandtedPermissionsFilter(ObjectPermissionsFilter):
 | 
				
			||||||
 | 
					    """
 | 
				
			||||||
 | 
					    A filter backend that limits results to those where the requesting user
 | 
				
			||||||
 | 
					    has read object level permissions, owns the objects, or objects without
 | 
				
			||||||
 | 
					    an owner (for backwards compat)
 | 
				
			||||||
 | 
					    """
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def filter_queryset(self, request, queryset, view):
 | 
				
			||||||
 | 
					        objects_with_perms = super().filter_queryset(request, queryset, view)
 | 
				
			||||||
 | 
					        objects_owned = queryset.filter(owner=request.user)
 | 
				
			||||||
 | 
					        objects_unowned = queryset.filter(owner__isnull=True)
 | 
				
			||||||
 | 
					        return objects_with_perms | objects_owned | objects_unowned
 | 
				
			||||||
 | 
				
			|||||||
@ -1,18 +1,29 @@
 | 
				
			|||||||
from rest_framework.permissions import BasePermission
 | 
					from rest_framework.permissions import BasePermission
 | 
				
			||||||
from rest_framework.permissions import DjangoModelPermissions
 | 
					from rest_framework.permissions import DjangoObjectPermissions
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class PaperlessModelPermissions(DjangoModelPermissions):
 | 
					class PaperlessObjectPermissions(DjangoObjectPermissions):
 | 
				
			||||||
 | 
					    """
 | 
				
			||||||
 | 
					    A permissions backend that checks for object-level permissions
 | 
				
			||||||
 | 
					    or for ownership.
 | 
				
			||||||
 | 
					    """
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    perms_map = {
 | 
					    perms_map = {
 | 
				
			||||||
        "GET": ["%(app_label)s.view_%(model_name)s"],
 | 
					        "GET": ["%(app_label)s.view_%(model_name)s"],
 | 
				
			||||||
        "OPTIONS": [],
 | 
					        "OPTIONS": ["%(app_label)s.view_%(model_name)s"],
 | 
				
			||||||
        "HEAD": [],
 | 
					        "HEAD": ["%(app_label)s.view_%(model_name)s"],
 | 
				
			||||||
        "POST": ["%(app_label)s.add_%(model_name)s"],
 | 
					        "POST": ["%(app_label)s.add_%(model_name)s"],
 | 
				
			||||||
        "PUT": ["%(app_label)s.change_%(model_name)s"],
 | 
					        "PUT": ["%(app_label)s.change_%(model_name)s"],
 | 
				
			||||||
        "PATCH": ["%(app_label)s.change_%(model_name)s"],
 | 
					        "PATCH": ["%(app_label)s.change_%(model_name)s"],
 | 
				
			||||||
        "DELETE": ["%(app_label)s.delete_%(model_name)s"],
 | 
					        "DELETE": ["%(app_label)s.delete_%(model_name)s"],
 | 
				
			||||||
    }
 | 
					    }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def has_object_permission(self, request, view, obj):
 | 
				
			||||||
 | 
					        if hasattr(obj, "owner") and request.user == obj.owner:
 | 
				
			||||||
 | 
					            return True
 | 
				
			||||||
 | 
					        else:
 | 
				
			||||||
 | 
					            return super().has_object_permission(request, view, obj)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class PaperlessAdminPermissions(BasePermission):
 | 
					class PaperlessAdminPermissions(BasePermission):
 | 
				
			||||||
    def has_permission(self, request, view):
 | 
					    def has_permission(self, request, view):
 | 
				
			||||||
 | 
				
			|||||||
@ -28,8 +28,9 @@ from django.utils.translation import get_language
 | 
				
			|||||||
from django.views.decorators.cache import cache_control
 | 
					from django.views.decorators.cache import cache_control
 | 
				
			||||||
from django.views.generic import TemplateView
 | 
					from django.views.generic import TemplateView
 | 
				
			||||||
from django_filters.rest_framework import DjangoFilterBackend
 | 
					from django_filters.rest_framework import DjangoFilterBackend
 | 
				
			||||||
 | 
					from documents.filters import ObjectOwnedOrGrandtedPermissionsFilter
 | 
				
			||||||
from documents.permissions import PaperlessAdminPermissions
 | 
					from documents.permissions import PaperlessAdminPermissions
 | 
				
			||||||
from documents.permissions import PaperlessModelPermissions
 | 
					from documents.permissions import PaperlessObjectPermissions
 | 
				
			||||||
from documents.tasks import consume_file
 | 
					from documents.tasks import consume_file
 | 
				
			||||||
from packaging import version as packaging_version
 | 
					from packaging import version as packaging_version
 | 
				
			||||||
from paperless import version
 | 
					from paperless import version
 | 
				
			||||||
@ -146,8 +147,12 @@ class CorrespondentViewSet(ModelViewSet):
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
    serializer_class = CorrespondentSerializer
 | 
					    serializer_class = CorrespondentSerializer
 | 
				
			||||||
    pagination_class = StandardPagination
 | 
					    pagination_class = StandardPagination
 | 
				
			||||||
    permission_classes = (IsAuthenticated, PaperlessModelPermissions)
 | 
					    permission_classes = (IsAuthenticated, PaperlessObjectPermissions)
 | 
				
			||||||
    filter_backends = (DjangoFilterBackend, OrderingFilter)
 | 
					    filter_backends = (
 | 
				
			||||||
 | 
					        DjangoFilterBackend,
 | 
				
			||||||
 | 
					        OrderingFilter,
 | 
				
			||||||
 | 
					        ObjectOwnedOrGrandtedPermissionsFilter,
 | 
				
			||||||
 | 
					    )
 | 
				
			||||||
    filterset_class = CorrespondentFilterSet
 | 
					    filterset_class = CorrespondentFilterSet
 | 
				
			||||||
    ordering_fields = (
 | 
					    ordering_fields = (
 | 
				
			||||||
        "name",
 | 
					        "name",
 | 
				
			||||||
@ -172,7 +177,7 @@ class TagViewSet(ModelViewSet):
 | 
				
			|||||||
            return TagSerializer
 | 
					            return TagSerializer
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    pagination_class = StandardPagination
 | 
					    pagination_class = StandardPagination
 | 
				
			||||||
    permission_classes = (IsAuthenticated, PaperlessModelPermissions)
 | 
					    permission_classes = (IsAuthenticated, PaperlessObjectPermissions)
 | 
				
			||||||
    filter_backends = (DjangoFilterBackend, OrderingFilter)
 | 
					    filter_backends = (DjangoFilterBackend, OrderingFilter)
 | 
				
			||||||
    filterset_class = TagFilterSet
 | 
					    filterset_class = TagFilterSet
 | 
				
			||||||
    ordering_fields = ("name", "matching_algorithm", "match", "document_count")
 | 
					    ordering_fields = ("name", "matching_algorithm", "match", "document_count")
 | 
				
			||||||
@ -187,7 +192,7 @@ class DocumentTypeViewSet(ModelViewSet):
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
    serializer_class = DocumentTypeSerializer
 | 
					    serializer_class = DocumentTypeSerializer
 | 
				
			||||||
    pagination_class = StandardPagination
 | 
					    pagination_class = StandardPagination
 | 
				
			||||||
    permission_classes = (IsAuthenticated, PaperlessModelPermissions)
 | 
					    permission_classes = (IsAuthenticated, PaperlessObjectPermissions)
 | 
				
			||||||
    filter_backends = (DjangoFilterBackend, OrderingFilter)
 | 
					    filter_backends = (DjangoFilterBackend, OrderingFilter)
 | 
				
			||||||
    filterset_class = DocumentTypeFilterSet
 | 
					    filterset_class = DocumentTypeFilterSet
 | 
				
			||||||
    ordering_fields = ("name", "matching_algorithm", "match", "document_count")
 | 
					    ordering_fields = ("name", "matching_algorithm", "match", "document_count")
 | 
				
			||||||
@ -204,7 +209,7 @@ class DocumentViewSet(
 | 
				
			|||||||
    queryset = Document.objects.all()
 | 
					    queryset = Document.objects.all()
 | 
				
			||||||
    serializer_class = DocumentSerializer
 | 
					    serializer_class = DocumentSerializer
 | 
				
			||||||
    pagination_class = StandardPagination
 | 
					    pagination_class = StandardPagination
 | 
				
			||||||
    permission_classes = (IsAuthenticated, PaperlessModelPermissions)
 | 
					    permission_classes = (IsAuthenticated, PaperlessObjectPermissions)
 | 
				
			||||||
    filter_backends = (DjangoFilterBackend, SearchFilter, OrderingFilter)
 | 
					    filter_backends = (DjangoFilterBackend, SearchFilter, OrderingFilter)
 | 
				
			||||||
    filterset_class = DocumentFilterSet
 | 
					    filterset_class = DocumentFilterSet
 | 
				
			||||||
    search_fields = ("title", "correspondent__name", "content")
 | 
					    search_fields = ("title", "correspondent__name", "content")
 | 
				
			||||||
@ -552,7 +557,7 @@ class SavedViewViewSet(ModelViewSet):
 | 
				
			|||||||
    queryset = SavedView.objects.all()
 | 
					    queryset = SavedView.objects.all()
 | 
				
			||||||
    serializer_class = SavedViewSerializer
 | 
					    serializer_class = SavedViewSerializer
 | 
				
			||||||
    pagination_class = StandardPagination
 | 
					    pagination_class = StandardPagination
 | 
				
			||||||
    permission_classes = (IsAuthenticated, PaperlessModelPermissions)
 | 
					    permission_classes = (IsAuthenticated, PaperlessObjectPermissions)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def get_queryset(self):
 | 
					    def get_queryset(self):
 | 
				
			||||||
        user = self.request.user
 | 
					        user = self.request.user
 | 
				
			||||||
@ -828,7 +833,7 @@ class StoragePathViewSet(ModelViewSet):
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
    serializer_class = StoragePathSerializer
 | 
					    serializer_class = StoragePathSerializer
 | 
				
			||||||
    pagination_class = StandardPagination
 | 
					    pagination_class = StandardPagination
 | 
				
			||||||
    permission_classes = (IsAuthenticated, PaperlessModelPermissions)
 | 
					    permission_classes = (IsAuthenticated, PaperlessObjectPermissions)
 | 
				
			||||||
    filter_backends = (DjangoFilterBackend, OrderingFilter)
 | 
					    filter_backends = (DjangoFilterBackend, OrderingFilter)
 | 
				
			||||||
    filterset_class = StoragePathFilterSet
 | 
					    filterset_class = StoragePathFilterSet
 | 
				
			||||||
    ordering_fields = ("name", "path", "matching_algorithm", "match", "document_count")
 | 
					    ordering_fields = ("name", "path", "matching_algorithm", "match", "document_count")
 | 
				
			||||||
 | 
				
			|||||||
@ -258,6 +258,11 @@ CHANNEL_LAYERS = {
 | 
				
			|||||||
# Security                                                                    #
 | 
					# Security                                                                    #
 | 
				
			||||||
###############################################################################
 | 
					###############################################################################
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					AUTHENTICATION_BACKENDS = [
 | 
				
			||||||
 | 
					    "guardian.backends.ObjectPermissionBackend",
 | 
				
			||||||
 | 
					    "django.contrib.auth.backends.ModelBackend",
 | 
				
			||||||
 | 
					]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
AUTO_LOGIN_USERNAME = os.getenv("PAPERLESS_AUTO_LOGIN_USERNAME")
 | 
					AUTO_LOGIN_USERNAME = os.getenv("PAPERLESS_AUTO_LOGIN_USERNAME")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
if AUTO_LOGIN_USERNAME:
 | 
					if AUTO_LOGIN_USERNAME:
 | 
				
			||||||
@ -274,11 +279,7 @@ HTTP_REMOTE_USER_HEADER_NAME = os.getenv(
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
if ENABLE_HTTP_REMOTE_USER:
 | 
					if ENABLE_HTTP_REMOTE_USER:
 | 
				
			||||||
    MIDDLEWARE.append("paperless.auth.HttpRemoteUserMiddleware")
 | 
					    MIDDLEWARE.append("paperless.auth.HttpRemoteUserMiddleware")
 | 
				
			||||||
    AUTHENTICATION_BACKENDS = [
 | 
					    AUTHENTICATION_BACKENDS.insert(0, "django.contrib.auth.backends.RemoteUserBackend")
 | 
				
			||||||
        "django.contrib.auth.backends.RemoteUserBackend",
 | 
					 | 
				
			||||||
        "django.contrib.auth.backends.ModelBackend",
 | 
					 | 
				
			||||||
        "guardian.backends.ObjectPermissionBackend",
 | 
					 | 
				
			||||||
    ]
 | 
					 | 
				
			||||||
    REST_FRAMEWORK["DEFAULT_AUTHENTICATION_CLASSES"].append(
 | 
					    REST_FRAMEWORK["DEFAULT_AUTHENTICATION_CLASSES"].append(
 | 
				
			||||||
        "rest_framework.authentication.RemoteUserAuthentication",
 | 
					        "rest_framework.authentication.RemoteUserAuthentication",
 | 
				
			||||||
    )
 | 
					    )
 | 
				
			||||||
 | 
				
			|||||||
@ -6,7 +6,7 @@ from django.db.models.functions import Lower
 | 
				
			|||||||
from django.http import HttpResponse
 | 
					from django.http import HttpResponse
 | 
				
			||||||
from django.views.generic import View
 | 
					from django.views.generic import View
 | 
				
			||||||
from django_filters.rest_framework import DjangoFilterBackend
 | 
					from django_filters.rest_framework import DjangoFilterBackend
 | 
				
			||||||
from documents.permissions import PaperlessModelPermissions
 | 
					from documents.permissions import PaperlessObjectPermissions
 | 
				
			||||||
from paperless.filters import GroupFilterSet
 | 
					from paperless.filters import GroupFilterSet
 | 
				
			||||||
from paperless.filters import UserFilterSet
 | 
					from paperless.filters import UserFilterSet
 | 
				
			||||||
from paperless.serialisers import GroupSerializer
 | 
					from paperless.serialisers import GroupSerializer
 | 
				
			||||||
@ -43,7 +43,7 @@ class UserViewSet(ModelViewSet):
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
    serializer_class = UserSerializer
 | 
					    serializer_class = UserSerializer
 | 
				
			||||||
    pagination_class = StandardPagination
 | 
					    pagination_class = StandardPagination
 | 
				
			||||||
    permission_classes = (IsAuthenticated, PaperlessModelPermissions)
 | 
					    permission_classes = (IsAuthenticated, PaperlessObjectPermissions)
 | 
				
			||||||
    filter_backends = (DjangoFilterBackend, OrderingFilter)
 | 
					    filter_backends = (DjangoFilterBackend, OrderingFilter)
 | 
				
			||||||
    filterset_class = UserFilterSet
 | 
					    filterset_class = UserFilterSet
 | 
				
			||||||
    ordering_fields = ("username",)
 | 
					    ordering_fields = ("username",)
 | 
				
			||||||
@ -56,7 +56,7 @@ class GroupViewSet(ModelViewSet):
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
    serializer_class = GroupSerializer
 | 
					    serializer_class = GroupSerializer
 | 
				
			||||||
    pagination_class = StandardPagination
 | 
					    pagination_class = StandardPagination
 | 
				
			||||||
    permission_classes = (IsAuthenticated, PaperlessModelPermissions)
 | 
					    permission_classes = (IsAuthenticated, PaperlessObjectPermissions)
 | 
				
			||||||
    filter_backends = (DjangoFilterBackend, OrderingFilter)
 | 
					    filter_backends = (DjangoFilterBackend, OrderingFilter)
 | 
				
			||||||
    filterset_class = GroupFilterSet
 | 
					    filterset_class = GroupFilterSet
 | 
				
			||||||
    ordering_fields = ("name",)
 | 
					    ordering_fields = ("name",)
 | 
				
			||||||
 | 
				
			|||||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user