wagtail-birdsong/birdsong/views/editor.py

98 lines
3.1 KiB
Python

from django.http.response import JsonResponse
from django.shortcuts import render
from django.template.loader import render_to_string
from wagtail.contrib.modeladmin.views import CreateView, EditView, InspectView
def preview(request, campaign, test_contact):
return render(
request,
campaign.get_template(request),
campaign.get_context(request, test_contact)
)
def confirm_send(request, campaign, form, send_url, index_url):
context = {
'self': campaign,
'form': form,
'request': request,
'send_url': send_url,
'index_url': index_url
}
return render(request, "birdsong/editor/send_confirm.html", context)
def confirm_test(request, campaign, form, send_url, index_url):
context = {
'self': campaign,
'form': form,
'request': request,
'send_url': send_url,
'index_url': index_url
}
return render(request, "birdsong/editor/test_confirm.html", context)
class InspectCampaign(InspectView):
def __init__(self, model_admin, instance_pk):
super().__init__(model_admin, instance_pk)
self.contact_class = model_admin.contact_class
def get_context_data(self, **kwargs):
first_receipt = self.instance.receipts.first()
if first_receipt:
preview_contact = self.contact_class.objects.filter(
pk=first_receipt.pk).first()
else:
preview_contact = None
# Should this be frozen? Changes to templates will change old campaigns
preview = render_to_string(
self.instance.get_template(self.request),
self.instance.get_context(self.request, preview_contact),
)
context = {
'receipts': self.instance.receipt_set.all(),
'preview': preview
}
context.update(kwargs)
return super().get_context_data(**context)
def ajax_preview(request, view):
FormClass = view.get_form_class()
form = FormClass(request.POST)
if form.is_valid():
campaign = form.save(commit=False)
contact_class = view.model_admin.contact_class
# FIXME won't work with no contacts
test_contact = contact_class.objects.first()
content = render_to_string(
campaign.get_template(request),
campaign.get_context(request, test_contact)
)
return JsonResponse({
'success': True,
'preview': content,
})
else:
return JsonResponse({
'success': False,
'errors': form.errors,
})
class EditCampaignView(EditView):
def post(self, request, *args, **kwargs):
if request.META.get('HTTP_X_REQUESTED_WITH') == 'XMLHttpRequest':
return ajax_preview(request, self)
return super().post(request, *args, **kwargs)
class CreateCampaignView(CreateView):
def post(self, request, *args, **kwargs):
if request.META.get('HTTP_X_REQUESTED_WITH') == 'XMLHttpRequest':
return ajax_preview(request, self)
return super().post(request, *args, **kwargs)