improve coverage

Author: chfw

django_excel/__init__.py

@@ -18,11 +18,15 @@ def load_book(self):
 
 
 class ExcelFile(webio.ExcelInput, TemporaryUploadedFile):
+    def _get_file_extension(self):
+        extension = self.name.split(".")[1]
+        return extension
+        
     def load_single_sheet(self, sheet_name=None, **keywords):
-        return pe.load(self.file.replace(".upload", ""), sheet_name, **keywords)
+        return pe.load_from_memory(self._get_file_extension(), self.file.read(), sheet_name, **keywords)
 
     def load_book(self):
-        return pe.load_book(self.file.replace(".upload", ""))
+        return pe.load_book_from_memory(self._get_file_extension(), self.file.read())
 
 
 class ExcelMemoryFileUploadHandler(MemoryFileUploadHandler):
@@ -41,17 +45,12 @@ def file_complete(self, file_size):
         )
 
 class TemporaryExcelFileUploadHandler(TemporaryFileUploadHandler):
-    def file_complete(self, file_size):
-        self.file.seek(0)
-        return ExcelMemoryFile(
-            file=self.file,
-            field_name=self.field_name,
-            name=self.file_name,
-            content_type=self.content_type,
-            size=file_size,
-            charset=self.charset,
-            content_type_extra=self.content_type_extra
-        )
+    def new_file(self, file_name, *args, **kwargs):
+        """
+        Create the file object to append to as data is coming in.
+        """
+        super(TemporaryFileUploadHandler, self).new_file(file_name, *args, **kwargs)
+        self.file = ExcelFile(self.file_name, self.content_type, 0, self.charset, self.content_type_extra)
 
 webio.ExcelResponse = HttpResponse