Django ModelViewSet. Как объединить два метода/функции perfrom?
Есть ли способ объединить методы/функции представления? Представление использует ModelViewSet. У меня есть две функции perform_create и perform_update, которые делают одно и то же, и мне стало интересно, могу ли я их как-то объединить?
Body
{
"title": "1 Title",
"description": "1 Description",
"author": {
"id": 1
}
}
Вид
class ArticleView(viewsets.ModelViewSet):
queryset = Article.objects.all()
serializer_class = ArticleSerializer
def perform_create(self, serializer):
author = get_object_or_404(Author, id=self.request.data['author']['id'])
return serializer.save(author=author)
def perform_update(self, serializer):
author = get_object_or_404(Author, id=self.request.data['author']['id'])
return serializer.save(author=author)
Сериализаторы
class AuthorSerializer(serializers.ModelSerializer):
class Meta:
model = Author
fields = '__all__'
class ArticleSerializer(serializers.ModelSerializer):
author = AuthorSerializer()
class Meta:
model = Article
fields = '__all__'
Назначьте функцию perform_create
также как perform_update
:
class ArticleView(viewsets.ModelViewSet):
queryset = Article.objects.all()
serializer_class = ArticleSerializer
def perform_create(self, serializer):
author = get_object_or_404(Author, id=self.request.data['author']['id'])
return serializer.save(author=author)
perform_update = perform_create