87 lines
3.4 KiB
Python
87 lines
3.4 KiB
Python
import re
|
|
import threading
|
|
from io import BytesIO
|
|
|
|
from django.contrib import messages
|
|
from django.http import HttpResponseRedirect
|
|
from django.shortcuts import render
|
|
from django.urls import reverse
|
|
from django.utils.translation import gettext as _
|
|
|
|
from cookbook.forms import ExportForm, ImportForm, ImportExportBase
|
|
from cookbook.helper.permission_helper import group_required
|
|
from cookbook.integration.chowdown import Chowdown
|
|
from cookbook.integration.default import Default
|
|
from cookbook.integration.mealie import Mealie
|
|
from cookbook.integration.nextcloud_cookbook import NextcloudCookbook
|
|
from cookbook.integration.paprika import Paprika
|
|
from cookbook.integration.safron import Safron
|
|
from cookbook.models import Recipe, ImportLog
|
|
|
|
|
|
def get_integration(request, export_type):
|
|
if export_type == ImportExportBase.DEFAULT:
|
|
return Default(request, export_type)
|
|
if export_type == ImportExportBase.PAPRIKA:
|
|
return Paprika(request, export_type)
|
|
if export_type == ImportExportBase.NEXTCLOUD:
|
|
return NextcloudCookbook(request, export_type)
|
|
if export_type == ImportExportBase.MEALIE:
|
|
return Mealie(request, export_type)
|
|
if export_type == ImportExportBase.CHOWDOWN:
|
|
return Chowdown(request, export_type)
|
|
if export_type == ImportExportBase.SAFRON:
|
|
return Safron(request, export_type)
|
|
|
|
|
|
@group_required('user')
|
|
def import_recipe(request):
|
|
if request.method == "POST":
|
|
form = ImportForm(request.POST, request.FILES)
|
|
if form.is_valid():
|
|
try:
|
|
integration = get_integration(request, form.cleaned_data['type'])
|
|
|
|
il = ImportLog.objects.create(type=form.cleaned_data['type'], created_by=request.user, space=request.space)
|
|
files = []
|
|
for f in request.FILES.getlist('files'):
|
|
files.append({'file': BytesIO(f.read()), 'name': f.name})
|
|
t = threading.Thread(target=integration.do_import, args=[files, il])
|
|
t.setDaemon(True)
|
|
t.start()
|
|
|
|
return HttpResponseRedirect(reverse('view_import_response', args=[il.pk]))
|
|
except NotImplementedError:
|
|
messages.add_message(request, messages.ERROR, _('Importing is not implemented for this provider'))
|
|
else:
|
|
form = ImportForm()
|
|
|
|
return render(request, 'import.html', {'form': form})
|
|
|
|
|
|
@group_required('user')
|
|
def export_recipe(request):
|
|
if request.method == "POST":
|
|
form = ExportForm(request.POST, space=request.space)
|
|
if form.is_valid():
|
|
try:
|
|
integration = get_integration(request, form.cleaned_data['type'])
|
|
return integration.do_export(form.cleaned_data['recipes'])
|
|
except NotImplementedError:
|
|
messages.add_message(request, messages.ERROR, _('Exporting is not implemented for this provider'))
|
|
|
|
else:
|
|
form = ExportForm(space=request.space)
|
|
recipe = request.GET.get('r')
|
|
if recipe:
|
|
if re.match(r'^([0-9])+$', recipe):
|
|
if recipe := Recipe.objects.filter(pk=int(recipe), space=request.space).first():
|
|
form = ExportForm(initial={'recipes': recipe}, space=request.space)
|
|
|
|
return render(request, 'export.html', {'form': form})
|
|
|
|
|
|
@group_required('user')
|
|
def import_response(request, pk):
|
|
return render(request, 'import_response.html', {'pk': pk})
|