Как редактировать разрешение пользователя в Django Rest Framework - PullRequest
Новая
       60

Как редактировать разрешение пользователя в Django Rest Framework

0 голосов
/ 24 марта 2019

Я следую учебному пособию django Rest Framework . Я хочу добавить пользовательское разрешение, чтобы только аутентифицированный пользователь мог просматривать подробную информацию о каждом пользователе. Цель: любой пользователь может просматривать UserList, но только владелец может просматривать его UserDetail.

models.py 

class Meeting(models.Model):
        created = models.DateTimeField(auto_now_add=True)
        sinceWhen = models.DateTimeField(null=True)
        tilWhen = models.DateTimeField(null=True)
        owner = models.ForeignKey('auth.User', related_name='meetings', on_delete=models.CASCADE)
        #highlighted = models.TextField()

        def save(self, *args, **kwargs):
                super(Meeting, self).save(*args, **kwargs)


        class Meta:
                ordering = ('created',)

views.py 

from django.contrib.auth.models import User
# User is not created inside models.py

class UserList(generics.ListAPIView):
    queryset = User.objects.all()
    serializer_class = UserListSerializer

class UserDetail(generics.RetrieveAPIView):
        queryset = User.objects.all()
        serializer_class = UserSerializer
        permission_classes = (permissions.IsAuthenticatedOrReadOnly, IsOwnerOrReadOnly,)
# I added IsOwnerOrReadOnly to make it work, but this is the part where it causes error!

serializers.py 

class UserSerializer(serializers.ModelSerializer):
        meetings = serializers.PrimaryKeyRelatedField(many=True, queryset=Meeting.objects.all())
        #owner = serializers.ReadOnlyField(source='owner.username')

        class Meta:
                model = User
                fields = ('id', 'username', 'meetings',)

class UserListSerializer(serializers.ModelSerializer):
        #meetings = serializers.PrimaryKeyRelatedField(many=True, queryset=Meeting.objects.all())

        class Meta:
                model = User
                fields = ('username',)

permissions.py 

from rest_framework import permissions

class IsOwnerOrReadOnly(permissions.BasePermission):  
        def has_object_permission(self, request, view, obj):

                # Any permissions are only allowed to the owner of the meeting
                return obj.owner == request.user

Я переопределил IsOwnerOrReadOnly, чтобы только пользователь мог просматривать детали его / ее сведений о пользователе. И добавьте это вmission_class в views.py.

Тогда я получил эту ошибку: 

File "/home/tony/env/lib/python3.6/site-packages/rest_framework/views.py" in check_object_permissions
  345.             if not permission.has_object_permission(request, self, obj):

File "/home/tony/swpp_hw1/meetings/permissions.py" in has_object_permission
  15.       return obj.owner == request.user

Exception Type: AttributeError at /users/1/
Exception Value: 'User' object has no attribute 'owner'

Я пытался добавить класс User в models.py, но снова это вызывает ошибку ... Как можно решить эту проблему?

1 Ответ

1 голос
/ 24 марта 2019

Попробуйте изменить его следующим образом: 

return obj == request.user

, поскольку object - это пользователь, к которому вы пытаетесь получить доступ, а request.user - текущий аутентифицированный пользователь.

...