Source code for shuup.front.apps.customer_information.views

from django.contrib import messages
from django.contrib.auth.forms import PasswordChangeForm
from django.contrib.auth.views import PasswordChangeView
from django.shortcuts import redirect
from django.utils.translation import gettext_lazy as _
from django.views.generic import FormView, TemplateView

from shuup.core.models import CompanyContact, SavedAddress, get_company_contact
from shuup.front.utils.companies import allow_company_registration
from shuup.front.views.dashboard import DashboardViewMixin
from shuup.utils.django_compat import reverse_lazy
from shuup.utils.importing import cached_load



[docs]
class CustomPasswordChangeView(PasswordChangeView):
    template_name = "shuup/customer_information/change_password.jinja"
    success_url = reverse_lazy("shuup:customer_edit")
    form_class = PasswordChangeForm


[docs]
    def post(self, *args, **kwargs):
        response = super().post(*args, **kwargs)
        if response.status_code == 302:
            messages.success(self.request, _("Password changed."))
        return response



[docs]
    def get_form_kwargs(self):
        kwargs = super().get_form_kwargs()
        kwargs["user"] = self.request.user
        return kwargs





[docs]
class CustomerEditView(DashboardViewMixin, FormView):
    template_name = "shuup/customer_information/edit_customer.jinja"


[docs]
    def get_form_class(self):
        return cached_load("SHUUP_CUSTOMER_INFORMATION_EDIT_FORM")



[docs]
    def get_form_kwargs(self):
        kwargs = super().get_form_kwargs()
        kwargs["request"] = self.request
        return kwargs



[docs]
    def form_valid(self, form):
        form.save()
        messages.success(self.request, _("Account information was saved."))
        return redirect("shuup:customer_edit")





[docs]
class CompanyEditView(DashboardViewMixin, FormView):
    template_name = "shuup/customer_information/edit_company.jinja"


[docs]
    def dispatch(self, request, *args, **kwargs):
        if not (bool(get_company_contact(self.request.user)) or allow_company_registration(self.request.shop)):
            return redirect("shuup:customer_edit")
        return super().dispatch(request, *args, **kwargs)



[docs]
    def get_form_class(self):
        return cached_load("SHUUP_COMPANY_INFORMATION_EDIT_FORM")



[docs]
    def get_form_kwargs(self):
        kwargs = super().get_form_kwargs()
        kwargs["request"] = self.request
        return kwargs



[docs]
    def form_valid(self, form):
        form.save()
        return redirect("shuup:company_edit")



[docs]
    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["pending_company_approval"] = CompanyContact.objects.filter(
            members__in=[self.request.customer], is_active=False
        ).exists()
        return context





[docs]
class AddressBookView(DashboardViewMixin, TemplateView):
    template_name = "shuup/customer_information/addressbook/index.jinja"


[docs]
    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["addresses"] = SavedAddress.objects.filter(owner=self.request.customer)
        context["customer"] = self.request.customer
        return context





[docs]
class AddressBookEditView(DashboardViewMixin, FormView):
    template_name = "shuup/customer_information/addressbook/edit.jinja"
    instance = None


[docs]
    def dispatch(self, request, *args, **kwargs):
        try:
            self.instance = SavedAddress.objects.get(pk=kwargs.get("pk", 0), owner=self.request.customer)
        except Exception:
            self.instance = None
        return super().dispatch(request, *args, **kwargs)



[docs]
    def get_form_class(self):
        return cached_load("SHUUP_ADDRESS_BOOK_EDIT_FORM")



[docs]
    def get_form_kwargs(self):
        kwargs = super().get_form_kwargs()
        kwargs["request"] = self.request
        kwargs["instance"] = self.instance
        return kwargs



[docs]
    def form_valid(self, form):
        saved_address = form.save()
        messages.success(self.request, _("Address information was saved."))
        return redirect("shuup:address_book_edit", pk=saved_address.pk)





[docs]
def delete_address(request, pk):
    try:
        SavedAddress.objects.get(pk=pk, owner=request.customer).delete()
    except SavedAddress.DoesNotExist:
        messages.error(request, _("Cannot remove address because it doesn't exist."))
    return redirect("shuup:address_book")