Allow replacing the list of objects to dump

matthiask · matthiask · commit 8ebffcaf7e39 · 2023-06-12T12:25:17.000+02:00
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -13,6 +13,8 @@ Change log
 - Changed the ``assert`` statement for checking the dump version into a
   ``raise`` statement since assertions could be optimized out.
 - Renamed ``InvalidSpec`` to ``InvalidSpecError`` to make ruff happier.
+- Added an ``objects`` argument to ``dump_specs`` which allows overriding the
+  list of objects to dump.
 
 
 `0.5`_ (2023-03-15)
diff --git a/feincms3_data/data.py b/feincms3_data/data.py
@@ -89,16 +89,15 @@ def silence(*a):
     pass
 
 
-def dump_specs(specs, *, mappers=None):
+def dump_specs(specs, *, mappers=None, objects=None):
     stream = io.StringIO()
     stream.write('{"version": 1, "specs": ')
     json.dump(specs, stream)
     stream.write(', "objects": ')
     serializer = JSONSerializer(mappers=mappers or {})
-    serializer.serialize(
-        chain.from_iterable(_model_queryset(spec) for spec in specs),
-        stream=stream,
-    )
+    if objects is None:
+        objects = chain.from_iterable(_model_queryset(spec) for spec in specs)
+    serializer.serialize(objects, stream=stream)
     stream.write("}")
     return stream.getvalue()
 
