further abstraction for pyramid and Flask decendants

Author: chfw

pyexcel_webio/__init__.py

@@ -15,7 +15,7 @@
 else:
     from io import BytesIO
 
-    
+
 FILE_TYPE_MIME_TABLE = {
     "csv": "text/csv",
     "tsv": "text/tab-separated-values",
@@ -35,7 +35,7 @@ class ExcelInput(object):
     """
     def load_single_sheet(self, sheet_name=None, **keywords):
         """Abstract method
-        
+
         :param form_field_name: the file field name in the html form for file upload
         :param sheet_name: For an excel book, there could be multiple sheets. If it is left
                          unspecified, the sheet at index 0 is loaded. For 'csv', 'tsv' file,
@@ -158,6 +158,33 @@ def save_book_to_database(self, session=None, tables=None, initializers=None, ma
             book.save_to_database(session, tables, initializers=initializers, mapdicts=mapdicts)
 
 
+
+class ExcelInputInMultiDict(ExcelInput):
+    """ A generic interface for an upload excel file appearing in a dictionary
+    """
+    def get_file_tuple(self, field_name):
+        raise NotImplementedError("Please implement this function")
+
+    def load_single_sheet(self, field_name=None, sheet_name=None, **keywords):
+        file_type, file_handle = self.get_file_tuple(field_name)
+        if file_type is not None and file_handle is not None:
+            return pe.get_sheet(file_type=file_type,
+                                file_content=file_handle.read(),
+                                sheet_name=sheet_name,
+                                **keywords)
+        else:
+            return None
+
+    def load_book(self, field_name=None, **keywords):
+        file_type, file_handle = self.get_file_tuple(field_name)
+        if file_type is not None and file_handle is not None:        
+            return pe.get_book(file_type=file_type,
+                               file_content=file_handle.read(),
+                               **keywords)
+        else:
+            return None
+
+
 def dummy_func(content, content_type=None, status=200):
     return None
 
@@ -216,4 +243,4 @@ def make_response_from_a_table(session, table, file_type, status=200, **keywords
 
 def make_response_from_tables(session, tables, file_type, status=200, **keywords):
     book = pe.get_book(session=session, tables=tables, **keywords)
-    return make_response(book, file_type, status, **keywords)
+    return make_response(book, file_type, status, **keywords)

setup.py

@@ -15,7 +15,7 @@
 setup(
     name='pyexcel-webio',
     author="C. W.",
-    version='0.0.2',
+    version='0.0.3',
     author_email="wangc_2011@hotmail.com",
     url="https://github.com/chfw/pyexcel-webio",
     description='A generic request and response interface for pyexcel web extensions.',