make IonPy* types picklable

amazon-ion · Apr 13, 2020 · 2423c4a · 2423c4a
1 parent 37604cf
commit 2423c4a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/amazon/ion/simple_types.py b/amazon/ion/simple_types.py
@@ -139,6 +139,7 @@ def __init__(self, *args, **kwargs):
             super(IonPyValueType, self).__init__(*args, **kwargs)
 
     IonPyValueType.__name__ = name
+    IonPyValueType.__qualname__ = name
     return IonPyValueType
 
 

diff --git a/amazon/ion/util.py b/amazon/ion/util.py
@@ -90,6 +90,9 @@ def __new__(cls, name, value):
     def __init__(self, name, value):
         self.name = name
         self.value = value
+
+    def __getnewargs__(self):
+        return (self.name, self.value)
 
     def __str__(self):
         return '<%s.%s: %s>' % (type(self).__name__, self.name, self.value)