python-common/.../service_spec: implement ServiceSpec.__getnewargs__ to allow unpickl...

Signed-off-by: Samuel Just <sjust@redhat.com>
(cherry picked from commit d4a1f2b117a7eeb966ba8072f38d291c00e4a5e5)

diff --git a/src/python-common/ceph/deployment/service_spec.py b/src/python-common/ceph/deployment/service_spec.py
index 0408024121599707a508353b32fed6a852181c73..d81734fbf507e76bb682aa2d7889085aa93ce3ab 100644 (file)
--- a/src/python-common/ceph/deployment/service_spec.py
+++ b/src/python-common/ceph/deployment/service_spec.py
@@ -866,6 +866,14 @@ class ServiceSpec(object):
         sub_cls: Any = cls._cls(service_type)
         return object.__new__(sub_cls)
 
+    def __getnewargs__(self):
+        """
+        Pickle will pass the return of this function to __new__ upon
+        unpickle.  We need to ensure it gets service_type in order
+        to get the right subtype.
+        """
+        return (self.service_type,)
+
     def __init__(self,
                  service_type: str,
                  service_id: Optional[str] = None,