update news_gallery model

This commit is contained in:
Kuroshini 2019-12-11 15:13:06 +03:00
parent 822b18e2f4
commit 165509bc64

View File

@ -243,6 +243,15 @@ class NewsBackOfficeGallerySerializer(serializers.ModelSerializer):
"""Get url kwargs from request."""
return self.context.get('request').parser_context.get('kwargs')
def create(self, validated_data):
news_pk = self.get_request_kwargs().get('pk')
image_id = self.get_request_kwargs().get('image_id')
news_gallery_model = models.NewsGallery.objects.filter(image_id=image_id, news_id=news_pk).first()
if news_gallery_model:
news_gallery_model.update(**validated_data)
return news_gallery_model
return super().create(validated_data)
def validate(self, attrs):
"""Override validate method."""
news_pk = self.get_request_kwargs().get('pk')