Ticket #7254: dumpdata.py.diff

django/core/management/commands/dumpdata.py

@@ -2 +2 @@
 from django.core import serializers
 
 from optparse import make_option
+from sys import stderr
 
 class Command(BaseCommand):
     option_list = BaseCommand.option_list + (
-        make_option('--format', default='json', dest='format',
+        make_option('--format', dest='format', default='json',
             help='Specifies the output serialization format for fixtures.'),
-        make_option('--indent', default=None, dest='indent', type='int',
+        make_option('--indent', dest='indent', type='int', default=None,
             help='Specifies the indent level to use when pretty-printing output'),
+        make_option('-e', '--exclude', dest='exclude',action='append', default=None, 
+            help='App to exclude (use multiple --exclude to exclude multiple apps).'),
+        make_option('-v', '--verbosity', action='store', dest='verbosity', default='0',
+            type='choice', choices=['0', '1', '2'],
+            help='Verbosity level; 0=minimal output, 1=normal output, 2=all output'),
     )
     help = 'Output the contents of the database as a fixture of the given format.'
     args = '[appname ...]'
@@ -16 +22 @@
     def handle(self, *app_labels, **options):
         from django.db.models import get_app, get_apps, get_models
 
-        format = options.get('format', 'json')
-        indent = options.get('indent', None)
-        show_traceback = options.get('traceback', False)
+        format = options.get('format')
+        indent = options.get('indent')
+        exclude = options.get('exclude')
+        verbosity = int(options.get('verbosity'))
+        show_traceback = options.get('traceback')
 
+        if verbosity >= 2:
+            print >>stderr, \
+                "format %s\nindent %s\nexclude %s\nshow_traceback %s\n" % \
+                (format, indent, exclude, show_traceback)
+            print >>stderr, "apps %s" % (app_labels)
+
         if len(app_labels) == 0:
             app_list = get_apps()
         else:
             app_list = [get_app(app_label) for app_label in app_labels]
 
+        if exclude:
+            excl_apps = [get_app(app_label) for app_label in exclude]
+            for excl_app in excl_apps:
+                app_list.remove(excl_app)
+                if verbosity >= 2:
+                    print >>stderr, "excluded: %s" % excl_app.__name__
+        
+        if verbosity >= 2:
+            for app in app_list:
+                print >>stderr, "included: %s" % app.__name__
+            
         # Check that the serialization format exists; this is a shortcut to
         # avoid collating all the objects and _then_ failing.
         if format not in serializers.get_public_serializer_formats():
@@ -37 +62 @@
 
         objects = []
         for app in app_list:
+            if verbosity >= 1:
+                print >>stderr, "dumping %s" % (app.__name__)
             for model in get_models(app):
+                if verbosity >= 1:
+                    # model dosn't have a good __str,
+                    # so for now "<class... model_name>" is better than nothing.
+                    print >>stderr, "dumping %s,%s" % (app.__name__, model)
                 objects.extend(model._default_manager.all())
         try:
             return serializers.serialize(format, objects, indent=indent)