encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-07-27 16:40:03 +03:00
Code Issues Packages Projects Releases Wiki Activity

Added support for queryset in DjangoModelPermission for functional views

Browse Source
This commit is contained in:
Gabor 2020-12-12 23:47:15 +02:00
parent bb133522ef
commit d4c82b56b4
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
rest_framework/decorators.py
View File

@ -69,6 +69,7 @@ def api_view(http_method_names=None):
WrappedAPIView.permission_classes = getattr(func, 'permission_classes',
APIView.permission_classes)
WrappedAPIView.queryset = getattr(func, 'queryset', APIView.permission_classes)
WrappedAPIView.schema = getattr(func, 'schema',
APIView.schema)
@ -106,9 +107,11 @@ def throttle_classes(throttle_classes):
return decorator
def permission_classes(permission_classes):
def permission_classes(permission_classes, queryset=None):
def decorator(func):
func.permission_classes = permission_classes
if queryset is not None:
func.queryset = queryset
return func
return decorator