fix tagcategory binding serializer

This commit is contained in:
evgeniy-st 2019-10-21 10:12:30 +03:00
parent 21e3f76f5a
commit 1cd586662a

View File

@ -83,12 +83,15 @@ class TagBindObjectSerializer(serializers.Serializer):
object_id = serializers.IntegerField()
def validate(self, attrs):
view = self.context.get('view')
request = self.context.get('request')
obj_type = attrs.get('type')
obj_id = attrs.get('object_id')
request = self.context.get('request')
view = self.context.get('view')
tag = view.get_object()
attrs['tag'] = tag
if obj_type == self.ESTABLISHMENT:
establishment = Establishment.objects.filter(pk=obj_id).first()
if not establishment:
@ -128,12 +131,14 @@ class TagCategoryBindObjectSerializer(serializers.Serializer):
object_id = serializers.IntegerField()
def validate(self, attrs):
view = self.context.get('view')
request = self.context.get('request')
obj_type = attrs.get('type')
obj_id = attrs.get('object_id')
view = self.context.get('view')
tag_category = view.get_object()
attrs['tag_category'] = tag_category
request = self.context.get('request')
if obj_type == self.ESTABLISHMENT_TYPE:
establishment_type = EstablishmentType.objects.filter(pk=obj_id).\
@ -148,7 +153,7 @@ class TagCategoryBindObjectSerializer(serializers.Serializer):
exists():
raise RemovedBindingObjectNotFound()
attrs['related_object'] = establishment_type
elif obj_type == self.NEWS:
elif obj_type == self.NEWS_TYPE:
news_type = NewsType.objects.filter(pk=obj_id).first()
if not news_type:
raise BindingObjectNotFound()