Changed the s3file.load API design

Previous s3file.laod returns BytesIO or StringIO but now it returns
raw content of the file (instance of either `str` or `bytes`.

Author: yhori991

s3file/__S3File.py

@@ -1,3 +1,4 @@
+import sys
 import os
 import boto3
 from io import BytesIO, StringIO
@@ -8,9 +9,14 @@
 __DEFAULT_TEMP_DIR = '/tmp/s3file'
 
 
+thismodule = sys.modules[__name__]
 s3 = boto3.resource('s3')
 
 
+def s3_set_profile(profile_name):
+    thismodule.s3 = boto3.session.Session(profile_name=profile_name).resource('s3')
+
+
 def _split_into_bucket_and_key(path):
     bucket, key = '', ''
     path_split = path.split('/')
@@ -136,13 +142,7 @@ def s3_load(path, mode='rb', force_download=False, cache_dir=__DEFAULT_CACHE_DIR
     with open(local, mode) as fp:
         content = fp.read()
 
-    # return StringIO or BytesIO
-    if mode == 'r':
-        return StringIO(content)
-    elif mode == 'rb':
-        return BytesIO(content)
-    else:
-        return None
+    return content
 
 
 def s3_save(path, content, cache_dir=__DEFAULT_CACHE_DIR):

s3file/__init__.py

@@ -1,3 +1,4 @@
+# from .__S3File import s3_set_profile as set_profile
 from .__S3File import s3_xlist as xlist
 from .__S3File import s3_download as download
 from .__S3File import s3_upload as upload

s3file/__version__.py

@@ -2,7 +2,7 @@
 __title__ = 'py-s3file'
 __description__ = 'A simple utility tool to access s3 object through basic File I/O operations.'
 __copywrite__ = 'Copyright (C) yhori991'
-__version__ = '0.0.3'
+__version__ = '0.0.4'
 __license__ = 'MIT'
 __author__ = 'yhori991'
 __author_email__ = '[email protected]'