saving to django model(s)

Author: chfw

README.rst

@@ -61,12 +61,7 @@ Fonts, colors and charts are not supported.
 
 Installation
 ============
-You can install it via pip::
-
-    $ pip install django-excel
-
-
-or clone it and install it::
+You can install it from github only for now:
 
     $ git clone http://github.com/chfw/django-pyexcel.git
     $ cd django-excel

django_excel/__init__.py

@@ -16,6 +16,20 @@ def load_single_sheet(self, sheet_name=None, **keywords):
     def load_book(self):
         return pe.load_book_from_memory(self._get_file_extension(), self.file.read())
 
+    def save_to_database(self, model=None,
+                         sheet_name=None, name_columns_by_row=0, name_rows_by_column=-1, **keywords):
+        sheet = self.load_single_sheet(sheet_name=sheet_name,
+                                       name_columns_by_row=name_columns_by_row,
+                                       name_rows_by_column=name_rows_by_column,
+                                       **keywords)
+        if sheet:
+            sheet.save_to_django_model(model)
+
+    def save_book_to_database(self, models=None, **keywords):
+        book = self.load_book(**keywords)
+        if book:
+            book.save_to_django_models(models)
+
 
 class ExcelInMemoryUploadedFile(ExcelMixin, InMemoryUploadedFile):
     pass
@@ -58,6 +72,12 @@ def new_file(self, file_name, *args, **kwargs):
     make_response_from_dict,
     make_response_from_records,
     make_response_from_book_dict,
-    make_response_from_a_table,
-    make_response_from_tables
 )
+
+def make_response_from_a_model(model, file_type, status=200, **keywords):
+    sheet = pe.get_sheet(model, **keywords)
+    return make_response(sheet, file_type, status, **keywords)
+
+def make_response_from_models(models, file_type, status=200, **keywords):
+    book = pe.get_book(models, **keywords)
+    return make_response(book, file_type, status, **keywords)

doc/source/index.rst

@@ -58,10 +58,7 @@ Then run the test application::
     Performing system checks...
     
     System check identified no issues (0 silenced).
-    
-    You have unapplied migrations; your app may not work properly until they are applied.
-    Run 'python manage.py migrate' to apply them.
-    January 27, 2015 - 17:57:03
+    January 29, 2015 - 18:11:06
     Django version 1.7.1, using settings 'mysite.settings'
     Starting development server at http://127.0.0.1:8000/
     Quit the server with CTRL-BREAK.

polls/models.py

@@ -4,6 +4,7 @@
 class Question(models.Model):
     question_text = models.CharField(max_length=200)
     pub_date = models.DateTimeField('date published')
+    slug = models.CharField(max_length=10, unique=True, default="question")
 
 
 class Choice(models.Model):

polls/urls.py

@@ -7,5 +7,7 @@
     url(r'^$', views.upload, name='uplink'),
     url(r'^download/(.*)', views.download, name="download"),
     url(r'^exchange/(.*)', views.exchange, name="exchange"),
-    url(r'^parse/(.*)', views.parse, name="parse")
+    url(r'^parse/(.*)', views.parse, name="parse"),
+    url(r'^import/', views.import_data, name="import"),
+    url(r'^export/(.*)', views.export_data, name="export")
 )

polls/views.py

@@ -1,8 +1,9 @@
 from django.shortcuts import render_to_response
-from django.http import HttpResponseBadRequest, JsonResponse
+from django.http import HttpResponseBadRequest, JsonResponse, HttpResponse
 from django import forms
 from django.template import RequestContext
 import django_excel as excel
+from polls.models import Question, Choice
 import pyexcel.ext.xls
 import pyexcel.ext.xlsx
 import sys
@@ -37,6 +38,50 @@ def download(request, file_type):
     sheet = excel.pe.Sheet(data)
     return excel.make_response(sheet, file_type)
 
+def export_data(request, atype):
+    if atype == "sheet":
+        return excel.make_response_from_a_model(Question, 'csv')
+    elif atype == "book":
+        return excel.make_response_from_a_model([Question, Choice], 'csv')
+    
+def import_data(request):
+    if request.method == "POST":
+        form = UploadFileForm(request.POST, request.FILES)
+        def choice_func(row):
+            print row[0]
+            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, ['question_text', 'pub_date', 'slug'], None, 0),
+                    (Choice, ['question', 'choice_text', 'votes'], choice_func, 0) 
+                 ]
+                )
+            return HttpResponse("OK")
+        else:
+            return HttpResponseBadRequest()
+    else:
+        form = UploadFileForm()
+    return render_to_response('upload_form.html', {'form': form}, context_instance=RequestContext(request))
+    
+
+def import_dataa(request):
+    if request.method == "POST":
+        form = UploadFileForm(request.POST, request.FILES)
+        if form.is_valid():
+            request.FILES['file'].save_to_database(
+                model=(Question,
+                       ['question_text', 'pub_date']))
+            return HttpResponse("OK")
+        else:
+            return HttpResponseBadRequest()
+    else:
+        form = UploadFileForm()
+    return render_to_response('upload_form.html', {'form': form}, context_instance=RequestContext(request))
+    
+
 def exchange(request, file_type):
     form = UploadFileForm(request.POST, request.FILES)
     if form.is_valid():