Ticket #5610: dumpdata-additions-trunk-5610.diff

django/core/management/commands/dumpdata.py

@@ -14 +14 @@
     args = '[appname ...]'
 
     def handle(self, *app_labels, **options):
-        from django.db.models import get_app, get_apps, get_models
+        from django.db.models import get_app, get_apps, get_models, get_model
 
         format = options.get('format', 'json')
         indent = options.get('indent', None)
@@ -23 +23 @@
         if len(app_labels) == 0:
             app_list = get_apps()
         else:
-            app_list = [get_app(app_label) for app_label in app_labels]
+            app_list = {}
+            for app_label in app_labels:
+                app_model_label = app_label.split('.')
+                if len(app_model_label) <= 1:
+                    # This is just an app
+                    app_list[app_model_label[0]] = None
+                else:
+                    # This is app.Model
+                    if app_model_label[0] in app_list.keys() and app_list[app_model_label[0]]:
+                        app_list[app_model_label[0]].append(app_model_label[1])
+                    else:
+                        app_list[app_model_label[0]] = [app_model_label[1]]
 
         # Check that the serialization format exists; this is a shortcut to
         # avoid collating all the objects and _then_ failing.
@@ -36 +47 @@
             raise CommandError("Unknown serialization format: %s" % format)
 
         objects = []
-        for app in app_list:
-            for model in get_models(app):
-                objects.extend(model._default_manager.all())
+        for app_label in app_list:
+            if app_list[app_label]:
+                for model_label in app_list[app_label]:
+                    model = get_model(app_label, model_label)
+                    objects.extend(model.objects.all())
+            else:
+                app = get_app(app_label)
+                for model in get_models(app):
+                    objects.extend(model.objects.all())
         try:
             return serializers.serialize(format, objects, indent=indent)
         except Exception, e:

tests/modeltests/fixtures/fixtures/fixture4.json

@@ +1 @@
+[
+    {
+        "pk": "2", 
+        "model": "fixtures.article", 
+        "fields": {
+            "headline": "Poker has no place on ESPN", 
+            "pub_date": "2006-06-16 12:00:00"
+        }
+    }, 
+    {
+        "pk": "3", 
+        "model": "fixtures.article", 
+        "fields": {
+            "headline": "Time to reform copyright", 
+            "pub_date": "2006-06-16 13:00:00"
+        }
+    },
+    {
+        "pk" : "1",
+        "model": "fixtures.category",
+        "fields": {
+            "title" : "News Stories",
+            "description" : "Latest news stories"
+        }
+    }
+]
+ No newline at end of file

tests/modeltests/fixtures/models.py

@@ -21 +21 @@
     class Meta:
         ordering = ('-pub_date', 'headline')
 
+class Category(models.Model):
+    title = models.CharField(max_length=100)
+    description = models.TextField()
+          
+    def __unicode__(self):
+        return self.title
+      
+    class Meta:
+        ordering = ('title',)
+
 __test__ = {'API_TESTS': """
 >>> from django.core import management
 >>> from django.db.models import get_app
@@ -82 +92 @@
 # Dump the current contents of the database as a JSON fixture
 >>> management.call_command('dumpdata', 'fixtures', format='json')
 [{"pk": 3, "model": "fixtures.article", "fields": {"headline": "Time to reform copyright", "pub_date": "2006-06-16 13:00:00"}}, {"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place on ESPN", "pub_date": "2006-06-16 12:00:00"}}, {"pk": 1, "model": "fixtures.article", "fields": {"headline": "Python program becomes self aware", "pub_date": "2006-06-16 11:00:00"}}]
+
+# Reset Database representation again.
+>>> management.call_command('flush', verbosity=0, interactive=False)
+>>> management.call_command('loaddata', 'fixture4', verbosity=0)
+>>> Article.objects.all()
+[<Article: Time to reform copyright>, <Article: Poker has no place on ESPN>, <Article: Python program becomes self aware>]
+>>> Category.objects.all()
+[<Category: News Stories>]
+
+# Dump the contents of the database as a JSON fixture
+>>> management.call_command('dumpdata', 'fixtures', format='json')
+[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": "News Stories"}}, {"pk": 3, "model": "fixtures.article", "fields": {"headline": "Time to reform copyright", "pub_date": "2006-06-16 13:00:00"}}, {"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place on ESPN", "pub_date": "2006-06-16 12:00:00"}}, {"pk": 1, "model": "fixtures.article", "fields": {"headline": "Python program becomes self aware", "pub_date": "2006-06-16 11:00:00"}}]
+
+# Try just dumping the contents of fixtures.Category
+>>> management.call_command('dumpdata', 'fixtures.Category', format='json')
+[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": "News Stories"}}]
+
+# ..and just fixtures.Article
+>>> management.call_command('dumpdata', 'fixtures.Article', format='json')
+[{"pk": 3, "model": "fixtures.article", "fields": {"headline": "Time to reform copyright", "pub_date": "2006-06-16 13:00:00"}}, {"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place on ESPN", "pub_date": "2006-06-16 12:00:00"}}, {"pk": 1, "model": "fixtures.article", "fields": {"headline": "Python program becomes self aware", "pub_date": "2006-06-16 11:00:00"}}]
 """
 
 from django.test import TestCase

docs/django-admin.txt

@@ -122 +122 @@
 Note that Django's default settings live in ``django/conf/global_settings.py``,
 if you're ever curious to see the full list of defaults.
 
-dumpdata <appname appname ...>
+dumpdata <appname appname appname.Model ...>
 ------------------------------
 
 Outputs to standard output all data in the database associated with the named
@@ -130 +130 @@
 
 If no application name is provided, all installed applications will be dumped.
 
+Alternatively, you can provide a list of models in the form of ``appname.Model``
+if  you want to limit what models you dump. You can also provide a mixture of 
+both.                                                                          
+
 The output of ``dumpdata`` can be used as input for ``loaddata``.
 
 Note that ``dumpdata`` uses the default manager on the model for selecting the