Django REST Framework(DRF):将当前用户ID设置为字段值

时间:2015-01-22 10:49:35

标签: django django-rest-framework

我为他设置了模型NewsModel和2个序列化器:

models.py

class NewsModel(models.Model):
    title = models.CharField('Заголовок', max_length=255, help_text='Максимальная длина - 255 символов')
    announce = models.TextField('Анонс', help_text='Краткий анонс новости')
    author = models.ForeignKey(settings.AUTH_USER_MODEL, help_text='Автор новости', related_name='news')
    full_text = models.TextField('Полный текст новости', help_text='Полный текст новости')
    pub_date = models.DateTimeField('Дата публикации', auto_now_add=True, default=timezone.now, help_text='Дата публикации')

    def comments_count(self):
        return NewsComment.objects.filter(news=self.id).count()

    def get_author_full_name(self):
        return self.author.get_full_name()

    class Meta:
        db_table = 'news'
        ordering = ('-pub_date',)

serilizers.py:

from rest_framework import serializers
from .models import NewsModel
from extuser.serializers import UserMiniSerializer

class NewsReadSerializer(serializers.ModelSerializer):

    author = UserMiniSerializer()

    class Meta:
        model = NewsModel
        fields = ('id', 'title', 'announce', 'comments_count', 'reviews', 'author_name')

    def get_author_full_name(self, obj):
        return obj.get_author_full_name()


class NewsWriteSerializer(serializers.ModelSerializer):

    def validate_author(self, value):
        value = self.request.user.id
        return value

    class Meta:
        model = NewsModel        

我在 api.py

中选择了序列化程序
class NewsList(ListCreateAPIView):
    queryset = NewsModel.objects.order_by('-pub_date')
    def get_serializer_class(self, *args, **kwargs):
        if self.request.method == 'GET':
            return NewsReadSerializer
        return NewsWriteSerializer

    class Meta:
        model = NewsModel

但是当我创建NewsModel项目时,我会看到错误400:错误请求[{&#39;作者&#39;:&#39;此字段是必需的&#39;}] < / p>

如何在创建新项目时将当前用户ID设置为NewsItem.author值?

4 个答案:

答案 0 :(得分:26)

我认为您没有正确使用序列化程序。设置请求相关数据的更好做法是覆盖视图中的perform_create

def perform_create(self, serializer):
    serializer.save(author=self.request.user)

def perform_update(self, serializer):
    serializer.save(author=self.request.user)

然后将您的作者序列化程序设置为只读:

author = UserMiniSerializer(read_only=True)

通过这种方式,您只需使用一个NewsSerializer进行读写操作即可。

答案 1 :(得分:7)

在DRF版本中,先前3字段必须声明为allow_null = True且default = None。如果没有这个参数,DRF就不会进行检查。结果代码:

class NewsReadSerializer(serializers.ModelSerializer):

    """
    Serializer only for reading.

    author field serialized with other custom serializer
    """

    author = UserMiniSerializer()

    class Meta:
        model = NewsModel
        fields = ('id', 'title', 'announce', 'comments_count', 'reviews', 'author', 'pub_date',)


class NewsWriteSerializer(serializers.ModelSerializer):

    """
    Serializer for creating and updating records.

    author here is the instance of PrimaryKeyRelatedField, linked to all users
    """

    author = serializers.PrimaryKeyRelatedField(
        queryset=User.objects.all(), # Or User.objects.filter(active=True)
        required=False, 
        allow_null=True, 
        default=None
    )

    # Get the current user from request context
    def validate_author(self, value):
        return self.context['request'].user

    class Meta:
        model = NewsModel      
        fields = ('title', 'announce', 'full_text', 'author',)

答案 2 :(得分:7)

在新DRF中你可以写

owner = serializers.HiddenField(
    default=serializers.CurrentUserDefault()
)

请参阅http://www.django-rest-framework.org/api-guide/validators/#currentuserdefault

答案 3 :(得分:3)

我会尝试这样的事情:

your models.py

class NewsModel(models.Model):
    title = models.CharField(
        'Заголовок', max_length=255,
        help_text='Максимальная длина - 255 символов')
    announce = models.TextField('Анонс',
        help_text='Краткий анонс новости')
    author = models.ForeignKey(
        settings.AUTH_USER_MODEL,
        help_text='Автор новости', related_name='news')
    full_text = models.TextField(
        'Полный текст новости',
        help_text='Полный текст новости')
    pub_date = models.DateTimeField(
        'Дата публикации', auto_now_add=True,
        default=timezone.now, help_text='Дата публикации')

def comments_count(self):
    return NewsComment.objects.filter(news=self.id).count()

def get_author_full_name(self):
    return self.author.get_full_name()

class Meta:
    db_table = 'news'
    ordering = ('-pub_date',)

<强> serializers.py

(参考:http://www.django-rest-framework.org/api-guide/validators/#currentuserdefault

from <yourapp>.models import NewsModel
from rest_framework import serializers


class NewsModelSerializer(serializers.ModelSerializer):
    author = serializers.HiddenField(default=serializers.CurrentUserDefault())
    class Meta:
        model = NewsModel

此外,您应将 settings.py 设置为以下内容:

REST_FRAMEWORK = {
    'DEFAULT_PERMISSION_CLASSES': (
        'rest_framework.permissions.IsAuthenticated',)
}