svthalia/concrexit

"""Widgets provided by the payments package."""
from django.forms import Widget

from payments.models import Payment, PaymentUser
from payments.payables import payables


class PaymentWidget(Widget):
    """Custom widget for the Payment object, used in registrations."""

    template_name = "payments/widgets/payment.html"

    def __init__(self, attrs=None, obj=None):
        super().__init__(attrs)
        self.obj = obj

    def get_context(self, name, value, attrs) -> dict:
        context = super().get_context(name, value, attrs)
        if self.obj:
            # Make sure to ALWAYS use committed data from the database, and never invalid data that is not saved, for example from an invalid form
            self.obj.refresh_from_db()
            payable = payables.get_payable(self.obj)
            context["obj"] = payable
            context["app_label"] = self.obj._meta.app_label
            context["model_name"] = self.obj._meta.model_name
            if payable.payment:
                value = payable.payment.pk
        if self.obj and not value:
            context["payable_payer"] = (
                PaymentUser.objects.get(pk=payable.payment_payer.pk)
                if getattr(payable, "payment_payer", None) is not None
                else None
            )
        elif value:
            payment = Payment.objects.get(pk=value)
            context["url"] = payment.get_admin_url()
            context["payment"] = payment
        return context

    def value_from_datadict(self, data, files, name):
        if self.obj:
            payable = payables.get_payable(self.obj)
            if payable.payment:
                return payable.payment.pk
        return None

    class Media:
        js = ("admin/payments/js/payments.js",)


class SignatureWidget(Widget):
    """Widget for signature image."""

    template_name = "payments/widgets/signature.html"

    class Media:
        js = (
            "payments/js/signature_pad.min.js",
            "payments/js/main.js",
        )
        css = {"all": ("admin/payments/css/signature.css",)}