From 56e6cfffe2f40ad381be121c6e67a2598f8ca5ad Mon Sep 17 00:00:00 2001
From: Gabriel Detraz <detraz@crans.org>
Date: Sun, 12 Aug 2018 01:09:48 +0200
Subject: [PATCH] Fix comnpay url

---
 .../0031_comnpaypayment_production.py         | 20 +++++++++++++++++++
 cotisations/payment_methods/comnpay/models.py | 13 +++++++++++-
 2 files changed, 32 insertions(+), 1 deletion(-)
 create mode 100644 cotisations/migrations/0031_comnpaypayment_production.py

diff --git a/cotisations/migrations/0031_comnpaypayment_production.py b/cotisations/migrations/0031_comnpaypayment_production.py
new file mode 100644
index 00000000..25ec7fb8
--- /dev/null
+++ b/cotisations/migrations/0031_comnpaypayment_production.py
@@ -0,0 +1,20 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.10.7 on 2018-08-11 23:03
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('cotisations', '0030_custom_payment'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='comnpaypayment',
+            name='production',
+            field=models.BooleanField(default=True, verbose_name='Production mode enabled (production url, instead of homologation)'),
+        ),
+    ]
diff --git a/cotisations/payment_methods/comnpay/models.py b/cotisations/payment_methods/comnpay/models.py
index ff6fed0d..dbc2f4ba 100644
--- a/cotisations/payment_methods/comnpay/models.py
+++ b/cotisations/payment_methods/comnpay/models.py
@@ -65,6 +65,16 @@ class ComnpayPayment(PaymentMethodMixin, models.Model):
         decimal_places=2,
         default=1,
     )
+    production = models.BooleanField(
+        default=True,
+        verbose_name=_l("Production mode enabled (production url, instead of homologation)"),
+    )
+
+    def return_url_comnpay(self):
+        if self.production:
+            return 'https://secure.comnpay.com'
+        else:
+            return 'https://secure.homologation.comnpay.com'
 
     def end_payment(self, invoice, request):
         """
@@ -87,8 +97,9 @@ class ComnpayPayment(PaymentMethodMixin, models.Model):
             "",
             "D"
         )
+
         r = {
-            'action': 'https://secure.homologation.comnpay.com',
+            'action': self.return_url_comnpay(),
             'method': 'POST',
             'content': p.buildSecretHTML(
                 _("Pay invoice no : ")+str(invoice.id),