Adds import view

bookwyrm/views/__init__.py

@@ -5,4 +5,4 @@ from .invite import ManageInvites, Invite
 from .landing import About, Home, Feed, Discover
 from .notifications import Notifications
 from .direct_message import DirectMessage
-from .import_datra import Import
+from .import_data import Import, ImportStatus

bookwyrm/views/import_data.py

@@ -0,0 +1,83 @@
+''' import books from another app '''
+from io import TextIOWrapper
+
+from django.contrib.auth.decorators import login_required
+from django.core.exceptions import PermissionDenied
+from django.http import HttpResponseBadRequest
+from django.shortcuts import get_object_or_404, redirect
+from django.template.response import TemplateResponse
+from django.utils.decorators import method_decorator
+from django.views import View
+
+from bookwyrm import forms, goodreads_import, models
+from bookwyrm.tasks import app
+
+# pylint: disable= no-self-use
+@method_decorator(login_required, name='dispatch')
+class Import(View):
+    ''' import view '''
+    def get(self, request):
+        ''' load import page '''
+        return TemplateResponse(request, 'import.html', {
+            'title': 'Import Books',
+            'import_form': forms.ImportForm(),
+            'jobs': models.ImportJob.
+                    objects.filter(user=request.user).order_by('-created_date'),
+        })
+
+    def post(self, request):
+        ''' ingest a goodreads csv '''
+        form = forms.ImportForm(request.POST, request.FILES)
+        if form.is_valid():
+            include_reviews = request.POST.get('include_reviews') == 'on'
+            privacy = request.POST.get('privacy')
+            try:
+                job = goodreads_import.create_job(
+                    request.user,
+                    TextIOWrapper(
+                        request.FILES['csv_file'],
+                        encoding=request.encoding),
+                    include_reviews,
+                    privacy,
+                )
+            except (UnicodeDecodeError, ValueError):
+                return HttpResponseBadRequest('Not a valid csv file')
+            goodreads_import.start_import(job)
+            return redirect('/import-status/%d' % job.id)
+        return HttpResponseBadRequest()
+
+
+@method_decorator(login_required, name='dispatch')
+class ImportStatus(View):
+    ''' status of an existing import '''
+    def get(self, request, job_id):
+        ''' status of an import job '''
+        job = models.ImportJob.objects.get(id=job_id)
+        if job.user != request.user:
+            raise PermissionDenied
+        task = app.AsyncResult(job.task_id)
+        items = job.items.order_by('index').all()
+        failed_items = [i for i in items if i.fail_reason]
+        items = [i for i in items if not i.fail_reason]
+        return TemplateResponse(request, 'import_status.html', {
+            'title': 'Import Status',
+            'job': job,
+            'items': items,
+            'failed_items': failed_items,
+            'task': task
+        })
+
+    def post(self, request, job_id):
+        ''' retry lines from an import '''
+        job = get_object_or_404(models.ImportJob, id=job_id)
+        items = []
+        for item in request.POST.getlist('import_item'):
+            items.append(get_object_or_404(models.ImportItem, id=item))
+
+        job = goodreads_import.create_retry_job(
+            request.user,
+            job,
+            items,
+        )
+        goodreads_import.start_import(job)
+        return redirect('/import-status/%d' % job.id)