Ticket #101: createview.diff

django/bin/django-admin.py

@@ -395 +395 @@
         sys.exit(1)
 runserver.args = '[optional port number]'
 
+DJANGO_VIEW_TEMPLATE = """\
+# from {{ model_qname }} import {{ model_name }}
+from django.utils.httpwrappers import HttpResponse
+
+# Place here your methods
+"""
+
+DJANGO_VIEW_METHOD_TEMPLATE = """
+def {{ method_name }}(request):
+    return HttpResponse("{{ method_name }} method")
+"""
+
+def create_view(mod, filename, *methods):
+    "Generates a skeleton view module"
+    model_qname = mod.__name__
+    model_name = model_qname.split('.')[-1]
+
+    app_dir = os.path.normpath(os.path.join(mod.__file__, '../../'))
+    app_name = os.path.basename(app_dir)
+    url_file = os.path.join(app_dir, 'urls', app_name + '.py')
+    view_file = os.path.join(app_dir, 'views', filename + '.py')
+
+    if os.path.exists(view_file):
+        sys.stderr.write("Error: %s exists\n" % view_file)
+
+    try:
+        fp = open(view_file, 'w')
+        vt = DJANGO_VIEW_TEMPLATE
+        vt = vt.replace('{{ model_qname }}', model_qname)
+        vt = vt.replace('{{ model_name }}', model_name)
+        fp.write(vt)
+
+        for method_name in methods:
+            fp.write(DJANGO_VIEW_METHOD_TEMPLATE.replace('{{ method_name }}',
+                                                         method_name))
+    except IOError:
+        sys.stderr.write("Error: I/O error while trying to create '%s' view\n" % filename)
+
+create_view.args = '[app viewFileName optionalMethodName ...]'
+
 def usage():
     sys.stderr.write("Usage: %s [action]\n" % sys.argv[0])
 
@@ -421 +461 @@
     'startproject': startproject,
     'init': init,
     'install': install,
+    'createview': create_view,
 }
 
 if __name__ == "__main__":
@@ -449 +490 @@
     elif action == 'dbcheck':
         from django.core import meta
         mod_list = meta.get_all_installed_modules()
+    elif action == 'createview':
+        from django.core import meta
+        try:
+            mod = meta.get_app(sys.argv[2])
+            view = sys.argv[3]
+        except ImportError, e:
+            sys.stderr.write("Error: %s. Are you sure your INSTALLED_APPS setting is correct?\n" % e)
+            sys.exit(1)
+        except IndexError:
+            usage()
+        create_view(mod, view, *sys.argv[4:])
+        sys.exit(0)
     else:
         from django.core import meta
         try: