Source code for shuup.admin.modules.service_providers.wizard_forms

from django import forms
from django.utils.translation import gettext_lazy as _

from shuup.admin.forms import ShuupAdminForm
from shuup.core.models import (
    CustomCarrier,
    CustomPaymentProcessor,
    PaymentMethod,
    PaymentProcessor,
    ShippingMethod,
    TaxClass,
)




class ServiceWizardForm(ShuupAdminForm):
    service_name = forms.CharField(
        label=_("Service name"),
        help_text=_("The name shown in the shop checkout process."),
    )



    def __init__(self, **kwargs):
        self.provider = kwargs["instance"]
        super().__init__(**kwargs)




    def is_active(self):
        return bool(self.get_service()) if self.provider else None




    def get_service_choice(self, provider):
        return provider.get_service_choices()[0]




    def get_service(self):
        if issubclass(self.provider.__class__, PaymentProcessor):
            return self.get_payment_method()
        return self.get_shipping_method()




    def get_payment_method(self):
        return PaymentMethod.objects.filter(payment_processor=self.provider).first() if self.provider else None




    def get_shipping_method(self):
        return ShippingMethod.objects.filter(carrier=self.provider).first() if self.provider else None




    def save(self):
        is_new = not self.instance.pk
        provider = super().save()
        provider.enabled = True
        provider.save()
        if is_new:
            service_choice = self.get_service_choice(provider)
            shop = self.request.shop
            provider.create_service(
                service_choice,
                name=self.cleaned_data.get("service_name", service_choice.name),
                description=self.cleaned_data.get("service_description", ""),
                shop=shop,
                tax_class=TaxClass.objects.first(),
                enabled=True,
            )
        else:
            service = self.get_service()
            if service:
                service.name = self.cleaned_data.get("service_name")
                service.description = self.cleaned_data.get("service_description", "")
                service.save()
        return provider






class ManualShippingWizardForm(ServiceWizardForm):
    service_description = forms.CharField(
        label=_("Instructions"),
        required=False,
        widget=forms.Textarea,
        help_text=_("Additional instructions shown in the shop checkout process."),
    )



    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        if not self.provider:
            return
        service = self.get_shipping_method()
        if not service:
            return
        self.fields["service_name"].initial = service.name
        self.fields["service_description"].initial = service.description




    class Meta:
        model = CustomCarrier
        fields = ("name",)






class ManualPaymentWizardForm(ServiceWizardForm):
    service_description = forms.CharField(
        label=_("Instructions"),
        required=False,
        widget=forms.Textarea,
        help_text=_("Additional instructions shown in the shop checkout process."),
    )



    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        if not self.provider:
            return
        service = self.get_payment_method()
        if not service:
            return
        self.fields["service_name"].initial = service.name
        self.fields["service_description"].initial = service.description




    class Meta:
        model = CustomPaymentProcessor
        fields = ("name",)