adding a custom rule for memoryview

Author: crazyeng

wolframclient/deserializers/wxf/wxfparser.py

@@ -10,6 +10,28 @@
 )
 from wolframclient.serializers.wxfencoder.streaming import ExactSizeReader, ZipCompressedReader
 from wolframclient.utils import six
+from wolframclient.utils.dispatch import Dispatch
+
+wxf_input_to_buffer = Dispatch()
+
+@wxf_input_to_buffer.dispatch((six.binary_type, six.buffer_types))
+def encode_buffer(wxf_input):
+    return six.BytesIO(wxf_input)
+
+if six.PY2:
+    @wxf_input_to_buffer.dispatch(memoryview, replace_existing = True)
+    def encode_buffer(wxf_input):
+        return six.BytesIO(wxf_input.tobytes())    
+
+@wxf_input_to_buffer.dispatch(object)
+def encode_default(wxf_input):
+    if hasattr(wxf_input, 'read'):
+        return wxf_input
+    raise TypeError(
+        "Class %s neither implements a read method nor is a binary type."
+        % wxf_input.__class__.__name__
+    )
+
 
 
 class WXFParser(object):
@@ -58,15 +80,8 @@ def __init__(self, wxf_input):
         """WXF parser returning Python object from a WXF encoded byte sequence.
         """
         self.context = SerializationContext()
-        if isinstance(wxf_input, (six.binary_type, six.buffer_types)):
-            self.reader = six.BytesIO(wxf_input)
-        elif hasattr(wxf_input, "read"):
-            self.reader = wxf_input
-        else:
-            raise TypeError(
-                "Class %s neither implements a read method nor is a binary type."
-                % wxf_input.__class__.__name__
-            )
+        self.reader  = wxf_input_to_buffer(wxf_input)
+
         version, compress = self.parse_header()
         if compress == True:
             self.reader = ZipCompressedReader(self.reader)