enable isave_as and isave_book_as to save SheetStream and BookStream to django model(s)

Author: chfw

django_excel/__init__.py

@@ -61,6 +61,31 @@ def save_book_to_database(self, models=None, initializers=None,
         params['dest_batch_size'] = batch_size
         pe.save_book_as(**params)
 
+    def isave_to_database(self, model=None, initializer=None, mapdict=None,
+                          **keywords):
+        """
+        Save data from a sheet to a nominated django model
+        """
+        params = self.get_params(**keywords)
+        params['dest_model'] = model
+        params['dest_initializer'] = initializer
+        params['dest_mapdict'] = mapdict
+        pe.isave_as(**params)
+        self.free_resources()
+
+    def isave_book_to_database(self, models=None, initializers=None,
+                               mapdicts=None, batch_size=None,
+                               **keywords):
+        """
+        Save data from a book to a nominated django models
+        """
+        params = self.get_params(**keywords)
+        params['dest_models'] = models
+        params['dest_initializers'] = initializers
+        params['dest_mapdicts'] = mapdicts
+        params['dest_batch_size'] = batch_size
+        pe.isave_book_as(**params)
+
 
 class ExcelInMemoryUploadedFile(ExcelMixin, InMemoryUploadedFile):
     """
@@ -135,7 +160,7 @@ def _make_response(content, content_type, status, file_name=None):
     return response
 
 
-webio.ExcelResponse = _make_response
+webio.init_webio(_make_response)
 
 
 from pyexcel_webio import (  # noqa

polls/urls.py

@@ -23,4 +23,10 @@
     # survey_result
     url('^survey_result/',
         views.survey_result, name='survey_result'),
+
+    # testing purpose
+    url(r'^import_using_isave/',
+        views.import_data_using_isave_book_as),
+    url(r'^import_sheet_using_isave/',
+        views.import_sheet_using_isave_to_database)
 ]

polls/views.py

@@ -101,6 +101,38 @@ def choice_func(row):
         })
 
 
+def import_data_using_isave_book_as(request):
+    if request.method == "POST":
+        form = UploadFileForm(request.POST,
+                              request.FILES)
+
+        def choice_func(row):
+            q = Question.objects.filter(slug=row[0])[0]
+            row[0] = q
+            return row
+        if form.is_valid():
+            request.FILES['file'].save_book_to_database(
+                models=[Question, Choice],
+                initializers=[None, choice_func],
+                mapdicts=[
+                    ['question_text', 'pub_date', 'slug'],
+                    ['question', 'choice_text', 'votes']]
+            )
+            return redirect('handson_view')
+        else:
+            return HttpResponseBadRequest()
+    else:
+        form = UploadFileForm()
+    return render(
+        request,
+        'upload_form.html',
+        {
+            'form': form,
+            'title': 'Import excel data into database example',
+            'header': 'Please upload sample-data.xls:'
+        })
+
+
 def import_sheet(request):
     if request.method == "POST":
         form = UploadFileForm(request.POST,
@@ -121,6 +153,25 @@ def import_sheet(request):
         {'form': form})
 
 
+def import_sheet_using_isave_to_database(request):
+    if request.method == "POST":
+        form = UploadFileForm(request.POST,
+                              request.FILES)
+        if form.is_valid():
+            request.FILES['file'].isave_to_database(
+                model=Question,
+                mapdict=['question_text', 'pub_date', 'slug'])
+            return HttpResponse("OK")
+        else:
+            return HttpResponseBadRequest()
+    else:
+        form = UploadFileForm()
+    return render(
+        request,
+        'upload_form.html',
+        {'form': form})
+
+
 def exchange(request, file_type):
     form = UploadFileForm(request.POST, request.FILES)
     if form.is_valid():

sample-sheet-for-isave.xls

@@ -162,7 +162,6 @@ def test_exchange(self):
             tmp_filename = "test.%s" % in_file_type
             sheet.save_as(tmp_filename)
             for file_type in FILE_TYPE_MIME_TABLE.keys():
-                print("Post %s -> Respond %s" % (in_file_type, file_type))
                 with open(tmp_filename, "rb") as fp:
                     response = self.client.post(
                         '/polls/exchange/'+file_type,
@@ -219,6 +218,43 @@ def testBook(self):
         +---------------+----+-------------+-------+""").strip('\n')  # noqa
         self.assertEqual(str(book), content)
 
+    def testBookUsingIsave(self):
+        fp = open(self.testfile, "rb")
+        response = self.client.post('/polls/import_using_isave/',
+                                    data={"file": fp})
+        eq_(response.status_code, 302)
+        response2 = self.client.get('/polls/export/book')
+        assert response2.status_code == 200
+        book = pe.get_book(file_type='xls', file_content=response2.content)
+        content = dedent("""
+        question:
+        +----+---------------------------+----------------------------------------------+----------+
+        | id | pub_date                  | question_text                                | slug     |
+        +----+---------------------------+----------------------------------------------+----------+
+        | 1  | 2015-01-28T00:00:00+00:00 | What is your favourite programming language? | language |
+        +----+---------------------------+----------------------------------------------+----------+
+        | 2  | 2015-01-29T00:00:00+00:00 | What is your favourite IDE?                  | ide      |
+        +----+---------------------------+----------------------------------------------+----------+
+        choice:
+        +---------------+----+-------------+-------+
+        | choice_text   | id | question_id | votes |
+        +---------------+----+-------------+-------+
+        | Java          | 1  | 1           | 0     |
+        +---------------+----+-------------+-------+
+        | C++           | 2  | 1           | 0     |
+        +---------------+----+-------------+-------+
+        | C             | 3  | 1           | 0     |
+        +---------------+----+-------------+-------+
+        | Eclipse       | 4  | 2           | 0     |
+        +---------------+----+-------------+-------+
+        | Visual Studio | 5  | 2           | 0     |
+        +---------------+----+-------------+-------+
+        | PyCharm       | 6  | 2           | 0     |
+        +---------------+----+-------------+-------+
+        | IntelliJ      | 7  | 2           | 0     |
+        +---------------+----+-------------+-------+""").strip('\n')  # noqa
+        self.assertEqual(str(book), content)
+
     def testSheet(self):
         fp = open(self.testfile, "rb")
         response = self.client.post('/polls/import/', data={"file": fp})
@@ -255,6 +291,25 @@ def testImportSheet(self):
         +----+---------------------------+----------------------------------------------+----------+""").strip('\n')  # noqa
         self.assertEqual(str(sheet), content)
 
+    def testImportSheetUsingISave(self):
+        fp = open("sample-sheet-for-isave.xls", "rb")
+        response = self.client.post('/polls/import_sheet_using_isave/',
+                                    data={"file": fp})
+        eq_(response.status_code, 200)
+        response2 = self.client.get('/polls/export/sheet')
+        eq_(response2.status_code, 200)
+        sheet = pe.get_sheet(file_type='xls', file_content=response2.content)
+        content = dedent("""
+        question:
+        +----+---------------------------+----------------------------------------------+----------+
+        | id | pub_date                  | question_text                                | slug     |
+        +----+---------------------------+----------------------------------------------+----------+
+        | 1  | 2015-01-28T00:00:00+00:00 | What is your favourite programming language? | language |
+        +----+---------------------------+----------------------------------------------+----------+
+        | 2  | 2015-01-29T00:00:00+00:00 | What is your favourite IDE?                  | ide      |
+        +----+---------------------------+----------------------------------------------+----------+""").strip('\n')  # noqa
+        self.assertEqual(str(sheet), content)
+
     def testCustomExport(self):
         fp = open(self.testfile, "rb")
         response = self.client.post('/polls/import/', data={"file": fp})