serializers.py 517 B

123456789101112131415161718
  1. from rest_framework import serializers
  2. from extras.models import Graph
  3. class GraphSerializer(serializers.ModelSerializer):
  4. embed_url = serializers.SerializerMethodField()
  5. embed_link = serializers.SerializerMethodField()
  6. class Meta:
  7. model = Graph
  8. fields = ['name', 'embed_url', 'embed_link']
  9. def get_embed_url(self, obj):
  10. return obj.embed_url(self.context['graphed_object'])
  11. def get_embed_link(self, obj):
  12. return obj.embed_link(self.context['graphed_object'])