1 from rest_framework.decorators import api_view
2 from rest_framework.response import Response
3 from rest_framework.reverse import reverse
4 from rest_framework import serializers
5 from core.models import *
6 from django.forms import widgets
9 Schema of the generator object:
10 all: Set of all Model objects
11 all_if(regex): Set of Model objects that match regex
14 plural: English plural of object name
15 camel: CamelCase version of object name
16 refs: list of references to other Model objects
17 props: list of properties minus refs
19 TODO: Deal with subnets
22 # Based on api_root.py
25 def api_root(request, format=None):
27 {% for object in generator.all %}'{{ object.plural }}': reverse('{{ object }}-list', request=request, format=format),
31 # Based on serializers.py
33 {% for object in generator.all %}
35 class {{ object.camel }}Serializer(serializers.HyperlinkedModelSerializer):
36 id = serializers.Field()
37 {% for ref in object.refs %}
39 {{ ref.plural }} = serializers.HyperlinkedRelatedField(many=True, read_only=True, view_name='{{ ref }}-detail')
41 {{ ref }} = serializers.HyperlinkedRelatedField(many=True, read_only=True, view_name='{{ ref }}-detail')
46 fields = ({% for prop in object.props %}'{{ prop }}',{% endfor %})
50 {% for object in generator.all %}
51 {{ object.camel }}: {{ object.camel }}Serializer,
56 # Based on core/views/*.py
57 {% for object in generator.all %}
59 class {{ object.camel }}List(generics.ListCreateAPIView):
60 queryset = {{ object.camel }}.objects.select_related.all()
61 serializer_class = {{ object.camel }}Serializer
63 def get_queryset(self):
64 return {{ object.camel }}.select_by_user(self.request.user)
66 def create(self, request, *args, **kwargs):
67 #obj = {{ object.camel }}().update(request.DATA)
68 obj = self.get_object()
69 obj.caller = request.user
70 if obj.can_update(request.user):
71 return super({{ object.camel }}List, self).create(request, *args, **kwargs)
73 return Response(status=status.HTTP_400_BAD_REQUEST)
75 class {{ object.camel }}Detail(generics.RetrieveUpdateDestroyAPIView):
76 queryset = {{ object.camel }}.objects.select_related.all()
77 serializer_class = {{ object.camel }}Serializer
79 def get_queryset(self):
80 return {{ object.camel }}.select_by_user(self.request.user)
82 def update(self, request, *args, **kwargs):
83 obj = self.get_object()
84 if obj.can_update(request.user):
85 return super({{ object.camel }}Detail, self).update(request, *args, **kwargs)
87 return Response(status=status.HTTP_400_BAD_REQUEST)
89 def destroy(self, request, *args, **kwargs):
90 obj = self.get_object()
91 if obj.can_update(request.user):
92 return super({{ object.camel }}Detail, self).destroy(request, *args, **kwargs)
94 return Response(status=status.HTTP_400_BAD_REQUEST)