Added support for generic view set

drf_generators/generators.py

@@ -6,9 +6,11 @@
 from drf_generators.templates.viewset import VIEW_SET_URL, VIEW_SET_VIEW
 from drf_generators.templates.function import FUNCTION_URL, FUNCTION_VIEW
 from drf_generators.templates.modelviewset import MODEL_URL, MODEL_VIEW
+from drf_generators.templates.genericviewset import GENERIC_URL, GENERIC_VIEW
+
 
 __all__ = ['BaseGenerator', 'APIViewGenerator', 'ViewSetGenerator',
-           'FunctionViewGenerator', 'ModelViewSetGenerator']
+           'FunctionViewGenerator', 'ModelViewSetGenerator', 'GenericViewSetGenerator']
 
 
 class BaseGenerator(object):
@@ -110,3 +112,10 @@ def __init__(self, app_config, force):
         super(ModelViewSetGenerator, self).__init__(app_config, force)
         self.view_template = Template(MODEL_VIEW)
         self.url_template = Template(MODEL_URL)
+
+class GenericViewSetGenerator(BaseGenerator):
+
+    def __init__(self, app_config, force):
+        super(GenericViewSetGenerator, self).__init__(app_config, force)
+        self.view_template = Template(GENERIC_VIEW)
+        self.url_template = Template(GENERIC_URL)

drf_generators/management/commands/generate.py

@@ -60,6 +60,8 @@ def handle_app_config(self, app_config, **options):
             generator = FunctionViewGenerator(app_config, force)
         elif fmt == 'modelviewset':
             generator = ModelViewSetGenerator(app_config, force)
+        elif fmt == 'genericviewset':
+            generator = GenericViewSetGenerator(app_config, force)
         else:
             message = '\'%s\' is not a valid format. ' % options['format']
             message += '(viewset, modelviewset, apiview, function)'

drf_generators/templates/genericviewset.py

@@ -0,0 +1,42 @@
+
+__all__ = ['GENERIC_URL', 'GENERIC_VIEW']
+
+
+GENERIC_URL = """from django.conf.urls import include, url
+from {{ app }} import views
+
+
+urlpatterns = [
+{% for model in models %}
+  url(r'^{{ model|lower }}/(?P<pk>[0-9]+)/$', views.{{ model }}Detail.as_view()),
+  url(r'^{{ model|lower }}/$', views.{{ model }}List.as_view()),
+{% endfor %}
+]
+"""
+
+
+GENERIC_VIEW = """from rest_framework.permissions import IsAuthenticated
+from rest_framework import generics
+
+from {{ app }}.serializers import {{ serializers|join:', ' }}
+from {{ app }}.models import {{ models|join:', ' }}
+{% for model in models %}
+
+class {{ model }}List(generics.ListCreateAPIView):
+
+    def get_queryset(self):
+        queryset = {{ model }}.objects.all()        
+        return queryset
+
+    permission_classes = ( IsAuthenticated,)
+    serializer_class = {{ model }}Serializer
+
+class {{ model }}Detail(generics.RetrieveUpdateDestroyAPIView):
+
+    def get_queryset(self):
+        queryset = {{ model }}.objects.all()        
+        return queryset
+    serializer_class = {{ model }}Serializer
+    permission_classes = ( IsAuthenticated,)
+
+{% endfor %}"""