WIP: pickle: Context: no _message_handler

1 files changed, 13 insertions, 0 deletions

diff --git a/src/buildstream/_context.py b/src/buildstream/_context.py
index fc758d459..9aa6295a7 100644
--- a/src/buildstream/_context.py
+++ b/src/buildstream/_context.py
@@ -169,6 +169,19 @@ class Context():
         self._casquota = None
         self._directory = directory
 
+    def __getstate__(self):
+        state = self.__dict__.copy()
+
+        # Looks we can construct these if they're missing.
+        # Note that we'll need figure out what to do with 'required elements',
+        # hopefully nothing.
+        # state['_artifactcache'] = None
+        # state['_sourcecache'] = None
+
+        del state['_message_handler']
+
+        return state
+
     # load()
     #
     # Loads the configuration files