|
| 1 | +from django.contrib import messages |
| 2 | +from django.db import models |
| 3 | +from django.http import HttpResponseRedirect |
| 4 | +from django.utils.decorators import method_decorator |
| 5 | + |
| 6 | +from honeypot.decorators import check_honeypot |
| 7 | +from wagtail.wagtailadmin.edit_handlers import FieldPanel |
| 8 | +from wagtail.wagtailcore.models import Page |
| 9 | + |
| 10 | +from .forms import ContactForm |
| 11 | + |
| 12 | + |
| 13 | +class ContactMixin(models.Model): |
| 14 | + form_class = ContactForm |
| 15 | + success_url = None |
| 16 | + success_message = 'Thank you! We will get back to you as soon as possible.' |
| 17 | + |
| 18 | + enquiry_email = models.EmailField( |
| 19 | + blank=True, |
| 20 | + help_text="Email address that will receive enquiries", |
| 21 | + null=True, |
| 22 | + ) |
| 23 | + |
| 24 | + content_panels = Page.content_panels + [ |
| 25 | + FieldPanel('enquiry_email'), |
| 26 | + ] |
| 27 | + |
| 28 | + class Meta: |
| 29 | + abstract = True |
| 30 | + |
| 31 | + def get_form(self, request): |
| 32 | + return self.form_class(request.POST or None) |
| 33 | + |
| 34 | + @method_decorator(check_honeypot) |
| 35 | + def serve(self, request, *args, **kwargs): |
| 36 | + self.form = self.get_form(request) |
| 37 | + if request.method == 'POST': |
| 38 | + if self.form.is_valid(): |
| 39 | + self.form.save(page=self) # Save triggers an email |
| 40 | + # Add a message to be displayed to the user |
| 41 | + messages.add_message( |
| 42 | + request, messages.INFO, |
| 43 | + self.get_success_message()) |
| 44 | + # Redirect to the current page, to prevent resubmissions |
| 45 | + return HttpResponseRedirect(self.get_success_url()) |
| 46 | + |
| 47 | + return super().serve(request, *args, **kwargs) |
| 48 | + |
| 49 | + def get_context(self, request): |
| 50 | + ctx = super().get_context(request) |
| 51 | + ctx['form'] = self.form |
| 52 | + return ctx |
| 53 | + |
| 54 | + def get_success_url(self): |
| 55 | + return self.success_url or self.url |
| 56 | + |
| 57 | + def get_success_message(self): |
| 58 | + return self.success_message |
0 commit comments