lib.itmens/sync/views.py
2021-09-01 11:44:00 +02:00

71 lines
2.3 KiB
Python

from django.shortcuts import render
from django.contrib.auth.decorators import login_required
from django.http import HttpResponseBadRequest, JsonResponse, HttpResponse
from .models import SyncTask
from .forms import SyncTaskForm
from .jobs import sync_task_manager
import tempfile
import os
from threading import Thread
import openpyxl
from django.utils.datastructures import MultiValueDictKeyError
from openpyxl.utils.exceptions import InvalidFileException
from zipfile import BadZipFile
@login_required
def sync_douban(request):
"""
Sync douban data from .xlsx file generated by doufen
"""
if request.method == 'POST':
# validate sunmitted data
try:
uploaded_file = request.FILES['file']
wb = openpyxl.open(uploaded_file, read_only=True,
data_only=True, keep_links=False)
wb.close()
except (MultiValueDictKeyError, InvalidFileException, BadZipFile) as e :
# raise e
return HttpResponseBadRequest(content="invalid excel file")
# file_data = {'file': request.FILES['xlsx']}
form = SyncTaskForm(request.POST, request.FILES)
if form.is_valid():
# stop all preivous task
SyncTask.objects.filter(user=request.user, is_finished=False).update(is_finished=True)
form.save()
sync_task_manager.add_task(form.instance)
return HttpResponse(status=204)
else:
return HttpResponseBadRequest()
else:
return HttpResponseBadRequest()
@login_required
def query_progress(request):
task = request.user.user_synctasks.order_by('-id').first()
if task is not None:
return JsonResponse({
'progress': task.get_progress()
})
else:
return JsonResponse()
def query_last_task(request):
task = request.user.user_synctasks.order_by('-id').first()
if task is not None:
return JsonResponse({
'total_items': task.total_items,
'success_items': task.success_items,
'finished_items': task.finished_items,
'status': task.get_status_emoji(),
'is_finished': task.is_finished,
'failed_urls': task.failed_urls,
'ended_time': task.ended_time if task.is_finished else None,
})
else:
return JsonResponse()