Add interfaces to set loading encoding

thombashi · Aug 19, 2017 · ecc0a6e · ecc0a6e
1 parent c44cd0f
commit ecc0a6e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/pytablereader/factory/_base.py b/pytablereader/factory/_base.py
@@ -27,9 +27,9 @@ def source(self):
 
         return self._source
 
-    def __init__(self, source):
+    def __init__(self, source, encoding=Default.ENCODING):
         self._source = source
-        self._encoding = Default.ENCODING
+        self._encoding = encoding
 
     @abc.abstractmethod
     def create_from_path(self):  # pragma: no cover

diff --git a/pytablereader/loadermanager/_file.py b/pytablereader/loadermanager/_file.py
@@ -7,7 +7,7 @@
 from __future__ import absolute_import
 
 import typepy
-
+from .._constant import Default
 from ..factory import TableFileLoaderFactory
 from ._base import TableLoaderManager
 
@@ -41,8 +41,8 @@ class TableFileLoader(TableLoaderManager):
             * :py:meth:`pytablereader.factory.TableFileLoaderFactory.create_from_path`
     """
 
-    def __init__(self, file_path, format_name=None):
-        loader_factory = TableFileLoaderFactory(file_path)
+    def __init__(self, file_path, format_name=None, encoding=Default.ENCODING):
+        loader_factory = TableFileLoaderFactory(file_path, encoding=encoding)
 
         if typepy.is_not_null_string(format_name):
             loader = loader_factory.create_from_format_name(format_name)