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 %}
38 {{ ref.plural }} = serializers.HyperlinkedRelatedField(view_name='{{ ref }}-detail')
42 fields = ({% for prop in object.props %}'{{ prop }}',{% endfor %})
46 {% for object in generator.all %}
47 {{ object.camel }}: {{ object.camel }}Serializer,
52 # Based on core/views/*.py
53 {% for object in generator.all %}
55 class {{ object.camel }}List(generics.ListCreateAPIView):
56 #queryset = {{ object.camel }}.objects.all()
57 serializer_class = {{ object.camel }}Serializer
59 def get_queryset(self):
60 return {{ object.camel }}.select_by_user(self.request.user)
62 def create(self, request, *args, **kwargs):
63 #obj = {{ object.camel }}().update(request.DATA)
64 obj = self.get_object()
65 obj.caller = request.user
66 if obj.can_update(request.user):
67 return super({{ object.camel }}List, self).create(request, *args, **kwargs)
69 return Response(status=status.HTTP_400_BAD_REQUEST)
71 class {{ object.camel }}Detail(generics.RetrieveUpdateDestroyAPIView):
72 #queryset = {{ object.camel }}.objects.all()
73 serializer_class = {{ object.camel }}Serializer
75 def get_queryset(self):
76 return {{ object.camel }}.select_by_user(self.request.user)
78 def update(self, request, *args, **kwargs):
79 obj = self.get_object()
80 if obj.can_update(request.user):
81 return super({{ object.camel }}Detail, self).update(request, *args, **kwargs)
83 return Response(status=status.HTTP_400_BAD_REQUEST)
85 def destroy(self, request, *args, **kwargs):
86 obj = self.get_object()
87 if obj.can_update(request.user):
88 return super({{ object.camel }}Detail, self).destroy(request, *args, **kwargs)
90 return Response(status=status.HTTP_400_BAD_REQUEST)