fix
[redakcja.git] / src / documents / api / serializers.py
1 from rest_framework import serializers
2 from .. import models
3
4
5 class TextField(serializers.Field):
6     def get_attribute(self, instance):
7         return instance
8
9     def to_representation(self, value):
10         return value.materialize()
11
12     def to_internal_value(self, data):
13         return data
14
15
16 class BookSerializer(serializers.ModelSerializer):
17     id = serializers.HyperlinkedIdentityField(view_name='documents_api_book')
18
19     class Meta:
20         model = models.Book
21         fields = [
22             'id',
23             'title'
24         ]
25
26
27 class ChunkSerializer(serializers.ModelSerializer):
28     id = serializers.HyperlinkedIdentityField(view_name='documents_api_chunk')
29     book = serializers.HyperlinkedRelatedField(view_name='documents_api_book', read_only=True)
30     revisions = serializers.HyperlinkedIdentityField(view_name='documents_api_chunk_revision_list')
31     head = serializers.HyperlinkedRelatedField(view_name='documents_api_revision', read_only=True)
32     ## RelatedField
33
34     class Meta:
35         model = models.Chunk
36         fields = ['id', 'book', 'revisions', 'head', 'user', 'stage']
37
38
39 class RHRF(serializers.HyperlinkedRelatedField):
40     def get_queryset(self):
41         return self.context['chunk'].change_set.all();
42
43   
44 class RevisionSerializer(serializers.ModelSerializer):
45     id = serializers.HyperlinkedIdentityField(view_name='documents_api_revision')
46     parent = RHRF(
47         view_name='documents_api_revision',
48         queryset=models.Chunk.change_model.objects.all()
49     )
50     merge_parent = RHRF(
51         view_name='documents_api_revision',
52         read_only=True
53     )
54     chunk = serializers.HyperlinkedRelatedField(view_name='documents_api_chunk', read_only=True, source='tree')
55     author = serializers.SerializerMethodField()
56
57     class Meta:
58         model = models.Chunk.change_model
59         fields = ['id', 'chunk', 'created_at', 'author', 'author_email', 'author_name', 'parent', 'merge_parent']
60         read_only_fields = ['author_email', 'author_name']
61
62     def get_author(self, obj):
63         return obj.author.username if obj.author is not None else None
64         
65
66 class BookDetailSerializer(BookSerializer):
67     chunk = ChunkSerializer(many=True, source='chunk_set')
68
69     class Meta:
70         model = models.Book
71         fields = BookSerializer.Meta.fields + ['chunk']
72
73
74
75 class ChunkDetailSerializer(ChunkSerializer):
76     pass
77
78
79 class RevisionDetailSerializer(RevisionSerializer):
80     text = TextField()
81
82     class Meta(RevisionSerializer.Meta):
83         fields = RevisionSerializer.Meta.fields + ['description', 'text']
84
85     def create(self, validated_data):
86         chunk = self.context['chunk']
87         return chunk.commit(
88             validated_data['text'],
89             author=self.context['request'].user, # what if anonymous?
90             description=validated_data['description'],
91             parent=validated_data.get('parent'),
92         )