Add a form for REGISTER_PAY action

jurismarches · Apr 12, 2018 · a7135fa · a7135fa
1 parent 7088335
commit a7135fa
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/payzen/utils.py b/payzen/utils.py
@@ -1,4 +1,5 @@
 import hashlib
+from decimal import Decimal, ROUND_HALF_UP
 import operator
 from datetime import datetime
 
@@ -54,6 +55,22 @@ def signature(self):
         return get_payzen_signature(data_to_sign, self._certificate)
 
 
+class SEPAMandateAndPayFormData(SEPAMandateFormData):
+
+    def __init__(self, user, payzen_id, comeback_url, payzen_certificate,
+                 payzen_shop_id, payzen_context, payzen_version, amount,
+                 trans_id, payment_config, update=False):
+        super().__init__(
+            user, payzen_id, comeback_url, payzen_certificate,
+            payzen_shop_id, payzen_context, payzen_version, update)
+        self.vads_page_action = 'REGISTER_PAY'
+        a = amount.quantize(Decimal('1.00'), rounding=ROUND_HALF_UP)
+        self.vads_amount = str(a).replace('.', '')
+        self.vads_currency = '978'
+        self.vads_trans_id = str(trans_id)
+        self.vads_payment_config = payment_config
+
+
 def get_payzen_signature(data, certificate):
     """
     The signature is build following those steps :