💚 fix broken test

pyexcel · Aug 23, 2017 · ec325aa · ec325aa
1 parent 789d410
commit ec325aa
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/pyexcel_xls/xlsr.py b/pyexcel_xls/xlsr.py
@@ -10,10 +10,11 @@
 import math
 import datetime
 import xlrd
+from io import UnsupportedOperation
 
 from pyexcel_io.book import BookReader
 from pyexcel_io.sheet import SheetReader
-from pyexcel_io._compact import OrderedDict
+from pyexcel_io._compact import OrderedDict, PY2
 
 
 XLS_KEYWORDS = [
@@ -125,8 +126,14 @@ def _get_book(self, on_demand=False):
         if self._file_name:
             xlrd_params['filename'] = self._file_name
         elif self._file_stream:
-            if hasattr(self._file_stream, 'seek'):
-                self._file_stream.seek(0)
+            if PY2:
+                if hasattr(self._file_stream, 'seek'):
+                    self._file_stream.seek(0)
+            else:
+                try:
+                    self._file_stream.seek(0)
+                except UnsupportedOperation:
+                    pass
             file_content = self._file_stream.read()
             xlrd_params['file_contents'] = file_content
         elif self._file_content is not None: