Add pickle support to Butler.

lsst · May 22, 2018 · b3ac2e1 · b3ac2e1
1 parent 3bd5259
commit b3ac2e1
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/python/lsst/daf/butler/butler.py b/python/lsst/daf/butler/butler.py
@@ -167,6 +167,11 @@ def __init__(self, config=None, collection=None, run=None):
             if self.run is None:
                 self.run = self.registry.makeRun(runCollection)
 
+    def __reduce__(self):
+        """Support pickling.
+        """
+        return (Butler, (self.config, ))
+
     def put(self, obj, datasetType, dataId, producer=None):
         """Store and register a dataset.
 

diff --git a/tests/test_butler.py b/tests/test_butler.py
@@ -24,7 +24,11 @@
 
 import os
 import unittest
+<<<<<<< HEAD
 from tempfile import TemporaryDirectory
+=======
+import pickle
+>>>>>>> Add pickle support to Butler.
 
 import lsst.utils.tests
 
@@ -159,6 +163,13 @@ def testMakeRepo(self):
         self.assertIn("datastore.formatters", full)
         self.assertNotIn("datastore.formatters", limited)
 
+    def testPickle(self):
+        """Test pickle support.
+        """
+        butler = Butler(self.configFile)
+        butlerOut = pickle.loads(pickle.dumps(butler))
+        self.assertIsInstance(butlerOut, Butler)
+
 
 class MemoryTester(lsst.utils.tests.MemoryTestCase):
     pass