3 Created on Sep 2, 2013 |
3 Created on Sep 2, 2013 |
4 |
4 |
5 @author: ymh |
5 @author: ymh |
6 ''' |
6 ''' |
7 from p4l.api.serializers import RecordSerializer |
7 from p4l.api.serializers import RecordSerializer |
8 from p4l.models import Record |
8 from p4l.models import Record, signals |
9 from rest_framework import viewsets |
9 from rest_framework import viewsets |
10 |
10 |
11 class RecordViewSet(viewsets.ModelViewSet): |
11 class RecordViewSet(viewsets.ModelViewSet): |
12 ''' |
12 ''' |
13 API endpoint that allow Record to be viewed or edited |
13 API endpoint that allow Record to be viewed or edited |
14 ''' |
14 ''' |
15 def pre_save(self, obj): |
15 def pre_save(self, obj): |
16 viewsets.ModelViewSet.pre_save(self, obj) |
16 super(RecordViewSet, self).pre_save(obj) |
17 if self.request and hasattr(self.request, 'user') and self.request.user: |
17 if self.request and hasattr(self.request, 'user') and self.request.user: |
18 obj.modified_by = self.request.user |
18 obj.modified_by = self.request.user |
|
19 |
|
20 def post_save(self, obj, created=False): |
|
21 super(RecordViewSet, self).post_save(obj, created=created) |
|
22 signals.record_saved.send(Record, instance=obj, created=created) |
|
23 |
19 |
24 |
20 queryset = Record.objects.all() |
25 queryset = Record.objects.all() |
21 serializer_class = RecordSerializer |
26 serializer_class = RecordSerializer |
22 lookup_field = 'identifier' |
27 lookup_field = 'identifier' |