Merge pull request #4 from jurismarches/qn-extra-info

Add extra info
jurismarches · Mar 4, 2021 · 8b054cd · 8b054cd
2 parents 9b33dfe + 075acf6
commit 8b054cd
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/payzen/__init__.py b/payzen/__init__.py
@@ -1,2 +1,2 @@
-__version__ = '0.0.10'
+__version__ = '0.0.11'
 __version_info__ = tuple(__version__.split('.'))
diff --git a/payzen/utils.py b/payzen/utils.py
@@ -15,7 +15,7 @@ class SEPAMandateFormData:
 
     def __init__(self, user, payzen_id, comeback_url, payzen_certificate,
                  payzen_shop_id, payzen_context, payzen_version, update=False,
-                 redirect_timeout=None, allow_fragments=True):
+                 redirect_timeout=None, allow_fragments=True, extra={}):
 
         self._certificate = payzen_certificate
 
@@ -34,6 +34,9 @@ def __init__(self, user, payzen_id, comeback_url, payzen_certificate,
         self.vads_version = payzen_version
         self.vads_payment_cards = ''
 
+        for key, value in extra.items():
+            setattr(self, f"vads_ext_info_{key}", str(value))
+
         query_param_name = 'payment'
         if urlparse(comeback_url, allow_fragments=allow_fragments)[4]:
             # already have query params
@@ -74,11 +77,11 @@ 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, redirect_timeout=None,
-                 allow_fragments=True):
+                 allow_fragments=True, extra={}):
         super().__init__(
             user, payzen_id, comeback_url, payzen_certificate,
             payzen_shop_id, payzen_context, payzen_version, update, redirect_timeout,
-            allow_fragments)
+            allow_fragments, extra)
         self.vads_page_action = 'REGISTER_PAY'
         a = amount.quantize(Decimal('1.00'), rounding=ROUND_HALF_UP)
         self.vads_amount = str(a).replace('.', '')