Ticket #16330: dumpdata-primary-keys.patch

tests/regressiontests/fixtures_regress/tests.py

@@ -15 +15 @@
 from django.db import transaction
 from django.test import TestCase, TransactionTestCase, skipIfDBFeature, \
     skipUnlessDBFeature
+from django.utils import simplejson
 
 from models import Animal, Stuff
 from models import Absolute, Parent, Child
@@ -343 +344 @@
         self.assertTrue(emu_json in data)
         self.assertTrue(platypus_json in data)
 
+        stdout = StringIO()
+        management.call_command(
+            'dumpdata',
+            'fixtures_regress.animal',
+            format='json',
+            stdout=stdout,
+            primary_keys=['1 10'],
+        )
+        data = stdout.getvalue()
+        json=simplejson.loads(data)
+        self.assertEqual(len(json), 2)
+        keys=sorted([item['pk'] for item in json])
+        self.assertEqual(keys, [1, 10])
+
     def test_proxy_model_included(self):
         """
         Regression for #11428 - Proxy models aren't included when you dumpdata

django/core/management/commands/dumpdata.py

@@ +1 @@
+import sys
+import itertools
+
 from django.core.exceptions import ImproperlyConfigured
 from django.core.management.base import BaseCommand, CommandError
 from django.core import serializers
@@ -21 +24 @@
             help='Use natural keys if they are available.'),
         make_option('-a', '--all', action='store_true', dest='use_base_manager', default=False,
             help="Use Django's base manager to dump all models stored in the database, including those that would otherwise be filtered or modified by a custom manager."),
+        make_option('--pks', dest='primary_keys', action='append', default=[],
+            help="Only dump objects with given primary keys. Accepts a space seperated list of keys or '-' to read from stdin. Normally only usefull if you give a ModelName."),
     )
     help = ("Output the contents of the database as a fixture of the given "
             "format (using each model's default manager unless --all is "
@@ -35 +40 @@
         using = options.get('database', DEFAULT_DB_ALIAS)
         connection = connections[using]
         excludes = options.get('exclude',[])
+        pks=options.get('primary_keys', [])
+        assert isinstance(pks, list), pks
+        if pks==['-']:
+            # Read from Stdin
+            primary_keys=sys.stdin.read().split()
+        else:
+            primary_keys = list(itertools.chain(*[pk.split() for pk in pks]))
         show_traceback = options.get('traceback', False)
         use_natural_keys = options.get('use_natural_keys', False)
         use_base_manager = options.get('use_base_manager', False)
@@ -105 +117 @@
                 continue
             if not model._meta.proxy and router.allow_syncdb(using, model):
                 if use_base_manager:
-                    objects.extend(model._base_manager.using(using).all())
+                    queryset=model._base_manager.using(using).all()
                 else:
-                    objects.extend(model._default_manager.using(using).all())
+                    queryset=model._default_manager.using(using).all()
+                if primary_keys:
+                    queryset=queryset.filter(pk__in=primary_keys)
+                objects.extend(queryset)
 
         try:
             return serializers.serialize(format, objects, indent=indent,