Merge pull request #6299 from encode/xordoquy/fix_composable_permissions

Fix composable permissions
This commit is contained in:
Xavier Ordoquy 2018-11-27 15:47:30 +01:00 committed by GitHub
commit 11edf572c5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 28 additions and 13 deletions

View File

@ -11,7 +11,21 @@ from rest_framework import exceptions
SAFE_METHODS = ('GET', 'HEAD', 'OPTIONS') SAFE_METHODS = ('GET', 'HEAD', 'OPTIONS')
class OperandHolder: class OperationHolderMixin:
def __and__(self, other):
return OperandHolder(AND, self, other)
def __or__(self, other):
return OperandHolder(OR, self, other)
def __rand__(self, other):
return OperandHolder(AND, other, self)
def __ror__(self, other):
return OperandHolder(OR, other, self)
class OperandHolder(OperationHolderMixin):
def __init__(self, operator_class, op1_class, op2_class): def __init__(self, operator_class, op1_class, op2_class):
self.operator_class = operator_class self.operator_class = operator_class
self.op1_class = op1_class self.op1_class = op1_class
@ -59,18 +73,8 @@ class OR:
) )
class BasePermissionMetaclass(type): class BasePermissionMetaclass(OperationHolderMixin, type):
def __and__(cls, other): pass
return OperandHolder(AND, cls, other)
def __or__(cls, other):
return OperandHolder(OR, cls, other)
def __rand__(cls, other):
return OperandHolder(AND, other, cls)
def __ror__(cls, other):
return OperandHolder(OR, other, cls)
@six.add_metaclass(BasePermissionMetaclass) @six.add_metaclass(BasePermissionMetaclass)

View File

@ -589,3 +589,14 @@ class PermissionsCompositionTests(TestCase):
permissions.IsAuthenticated permissions.IsAuthenticated
) )
assert composed_perm().has_permission(request, None) is True assert composed_perm().has_permission(request, None) is True
def test_several_levels_and_precedence(self):
request = factory.get('/1', format='json')
request.user = self.user
composed_perm = (
permissions.IsAuthenticated &
permissions.IsAuthenticated |
permissions.IsAuthenticated &
permissions.IsAuthenticated
)
assert composed_perm().has_permission(request, None) is True