serializser working for tags but not single tags
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
from rest_framework import serializers
|
||||
|
||||
from taggit_serializer.serializers import TagListSerializerField, TaggitSerializer
|
||||
|
||||
|
||||
class SingleTag(str):
|
||||
@@ -37,3 +38,40 @@ class SingleTagSerializerField(serializers.Field):
|
||||
value = value.name
|
||||
value = SingleTag(value)
|
||||
return value
|
||||
|
||||
|
||||
class CustomTagSerializer(serializers.Serializer):
|
||||
|
||||
def create(self, validated_data):
|
||||
to_be_tagged, validated_data = self._pop_tags(validated_data)
|
||||
|
||||
tag_object = super(CustomTagSerializer, self).create(validated_data)
|
||||
|
||||
return self._save_tags(tag_object, to_be_tagged)
|
||||
|
||||
def update(self, instance, validated_data):
|
||||
to_be_tagged, validated_data = self._pop_tags(validated_data)
|
||||
|
||||
tag_object = super(CustomTagSerializer, self).update(
|
||||
instance, validated_data)
|
||||
|
||||
return self._save_tags(tag_object, to_be_tagged)
|
||||
|
||||
def _save_tags(self, tag_object, tags):
|
||||
for key in tags.keys():
|
||||
tag_values = tags.get(key)
|
||||
getattr(tag_object, key).set(*tag_values)
|
||||
|
||||
return tag_object
|
||||
|
||||
def _pop_tags(self, validated_data):
|
||||
to_be_tagged = {}
|
||||
import ipdb; ipdb.set_trace()
|
||||
for key in self.fields.keys():
|
||||
field = self.fields[key]
|
||||
if isinstance(field, TagListSerializerField):
|
||||
if key in validated_data:
|
||||
to_be_tagged[key] = validated_data.pop(key)
|
||||
|
||||
return (to_be_tagged, validated_data)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user