Better type assertion

Shindora · Sep 4, 2020 · 5d5eb8a · 5d5eb8a
1 parent 8dec167
commit 5d5eb8a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/splunklib/searchcommands/search_command.py b/splunklib/searchcommands/search_command.py
@@ -862,8 +862,7 @@ def _as_binary_stream(ifile):
     @staticmethod
     def _read_chunk(istream):
         # noinspection PyBroadException
-        if not six.PY2:
-            assert issubclass(type(istream), (io.BufferedIOBase, io.RawIOBase)), 'Stream must be binary'
+        assert isinstance(istream.read(0), six.binary_type), 'Stream must be binary'
 
         try:
             header = istream.readline()