feat: Implement authorize and charge speratly (configurable) @wip

Author: sveneberth

src/viur/shop/payment_providers/unzer_paylater_invoice.py

@@ -1,11 +1,13 @@
+import enum
 import functools
 import typing as t  # noqa
 
 import unzer
-from viur.core import current, db, errors, exposed
-from viur.core.skeleton import SkeletonInstance
+from unzer import PaymentResponse
 
 from viur import toolkit
+from viur.core import current, db, errors, exposed
+from viur.core.skeleton import SkeletonInstance
 from viur.shop.skeletons import OrderSkel
 from viur.shop.types import *
 from .unzer_abstract import UnzerAbstract, log_unzer_error
@@ -14,6 +16,11 @@
 logger = SHOP_LOGGER.getChild(__name__)
 
 
+class ChargeMode(enum.IntEnum):
+    CHARGE_DIRECTLY = enum.auto()
+    CHARGE_MANUAL = enum.auto()
+
+
 class UnzerPaylaterInvoice(UnzerAbstract):
     """
     Unzer Paylater Invoice payment method integration for the ViUR Shop.
@@ -23,6 +30,15 @@ class UnzerPaylaterInvoice(UnzerAbstract):
 
     name: t.Final[str] = "unzer-paylater_invoice"
 
+    def __init__(
+        self,
+        *args: t.Any,
+        charge_mode: ChargeMode = ChargeMode.CHARGE_DIRECTLY,
+        **kwargs: t.Any,
+    ) -> None:
+        super().__init__(*args, **kwargs)
+        self.charge_mode = charge_mode
+
     def can_order(
         self,
         order_skel: SkeletonInstance_T[OrderSkel],
@@ -56,16 +72,11 @@ def checkout(
         logger.debug(f"{payment=} [authorize response]")
         unzer_session["paymentId"] = payment.paymentId
         unzer_session["redirectUrl"] = payment.redirectUrl
-        processing_data = payment.processing.asDict()
-
-        payment = payment.charge(
-            amount=order_skel["total"]
-        )
-        logger.debug(f"{payment=} [charge response]")
-
         logger.debug(f"{unzer_session=}")
         current.session.get().markChanged()
 
+        processing_data = payment.processing.asDict()
+
         def set_payment(skel: SkeletonInstance):
             skel["payment"]["payments"][-1]["payment_id"] = payment.paymentId
             skel["payment"]["payments"][-1]["processing_data"] = processing_data
@@ -76,8 +87,25 @@ def set_payment(skel: SkeletonInstance):
             skel=order_skel,
         )
 
+        if self.charge_mode == ChargeMode.CHARGE_DIRECTLY:
+            order_skel, payment = self.charge(order_skel=order_skel, payment=payment)
+
         return unzer_session
 
+    def charge(
+        self,
+        order_skel: SkeletonInstance_T[OrderSkel],
+        payment: PaymentResponse | None = None,
+    ) -> tuple[SkeletonInstance_T[OrderSkel], PaymentResponse]:
+        if payment is None:
+            payment = self.client.getPayment(order_skel["payments"][-1]["payment_id"])
+
+        payment = payment.charge(
+            amount=order_skel["total"]
+        )
+        logger.debug(f"{payment=} [charge response]")
+        return order_skel, payment
+
     def get_customer(self, order_skel: SkeletonInstance) -> unzer.Customer:
         customer = self.customer_from_order_skel(order_skel)
         logger.debug(f"{customer=}")