0.1.0 uses pyexcel-io 0.1.0

Author: chfw

MANIFEST.in

@@ -1 +1,2 @@
 include README.rst
+include VERSION

VERSION

@@ -0,0 +1 @@
+0.1.0-dev

pyexcel_text/__init__.py

@@ -160,8 +160,14 @@ def __init__(self, filename, **keywords):
         TextWriter.__init__(self, filename, **keywords)
 
     def write(self, sheet_dicts):
-        import json
-        self.f.write(json.dumps(sheet_dicts))
+        if self.keywords.get('single_sheet_in_book', False):
+            keys = list(sheet_dicts.keys())
+            sheet = self.create_sheet(keys[0])
+            sheet.write_array(sheet_dicts[keys[0]])
+            sheet.close()
+        else:
+            import json
+            self.f.write(json.dumps(sheet_dicts))
 
     def create_sheet(self, name):
         return JsonSheetWriter(self.f, name)

setup.py

@@ -15,10 +15,13 @@
 with open("README.rst", 'r') as readme:
     README_txt = readme.read()
 
+with open("VERSION", "r") as version:
+    version_txt = version.read().rstrip()
+
 setup(
     name='pyexcel-text',
     author="C. W.",
-    version='0.0.3',
+    version=version_txt,
     author_email="[email protected]",
     url="https://github.com/chfw/pyexcel-text",
     packages=find_packages(exclude=['ez_setup', 'examples', 'tests']),

tests/test_io.py

@@ -1,5 +1,7 @@
 import os
 import sys
+import json
+
 from textwrap import dedent
 import pyexcel as pe
 from pyexcel.ext import text
@@ -98,6 +100,7 @@ def test_dict(self):
         5  6
         7  8
         =  =""").strip('\n')
+        print written_content
         assert written_content.strip('\n') == content
 
     def tearDown(self):
@@ -115,25 +118,20 @@ def test_new_normal_usage(self):
             [7, 8, 999]
         ]
         pe.save_as(array=content, dest_file_name=self.testfile)
-        f = open(self.testfile, "r")
-        written_content = f.read()
-        f.close()
-        content = dedent("""
-            [[1, 2, 3], [4, 588, 6], [7, 8, 999]]""").strip('\n')
-        assert written_content == content
+        with open(self.testfile, "r") as f:
+            written_content = json.load(f)
+            print written_content
+            assert written_content == content
 
     def test_dict(self):
         adict = {
             'sheet 1': [[1,2],[3,4]],
             'sheet 2': [[5,6],[7,8]]
         }
         pe.save_book_as(bookdict=adict, dest_file_name=self.testfile)
-        f = open(self.testfile, "r")
-        written_content = f.read()
-        f.close()
-        content = dedent("""
-            {"sheet 1": [[1, 2], [3, 4]], "sheet 2": [[5, 6], [7, 8]]}""").strip('\n')
-        assert written_content == content
+        with open(self.testfile, "r") as f:
+            written_content = json.load(f)
+            assert written_content == adict
 
     def tearDown(self):
         if os.path.exists(self.testfile):