phzhik/gault-millau
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix tagcategory binding serializer

Browse Source
This commit is contained in:
evgeniy-st 2019-10-21 10:12:30 +03:00
parent 21e3f76f5a
commit 1cd586662a
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
apps/tag/serializers.py
View File

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