17 lines
615 B
Python
17 lines
615 B
Python
from django.contrib.auth.models import User
|
|
|
|
from rest_framework import serializers
|
|
from .models import Snippet, LANGUAGE_CHOICES, STYLE_CHOICES
|
|
|
|
class SnippetSerializer(serializers.ModelSerializer):
|
|
owner = serializers.ReadOnlyField(source='owner.username')
|
|
class Meta:
|
|
model = Snippet
|
|
fields = ['id', 'title', 'owner', 'code', 'linenos', 'language', 'style']
|
|
|
|
class UserSerializer(serializers.ModelSerializer):
|
|
snippets = serializers.PrimaryKeyRelatedField(many=True, queryset=Snippet.objects.all())
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = ['id', 'username', 'snippets'] |