Also handle ch_vat

datacenterlight/utils.py

@@ -131,7 +131,7 @@ def check_otp(name, realm, token):
     return response.status_code
 
 
-def validate_vat_number(stripe_customer_id, vat_number):
+def validate_vat_number(stripe_customer_id, vat_number, country):
     try:
         billing_address = BillingAddress.objects.get(vat_number=vat_number)
     except BillingAddress.DoesNotExist as dne:
@@ -163,9 +163,13 @@ def validate_vat_number(stripe_customer_id, vat_number):
                         "validated_on": ""
                     }
             else:
-                tax_id_obj = create_tax_id(stripe_customer_id, vat_number)
+                tax_id_obj = create_tax_id(
+                    stripe_customer_id, vat_number,
+                    "ch_vat" if country.lower() == "ch" else "eu_vat")
     else:
-        tax_id_obj = create_tax_id(stripe_customer_id, vat_number)
+        tax_id_obj = create_tax_id(
+            stripe_customer_id, vat_number,
+            "ch_vat" if country.lower() == "ch" else "eu_vat")
 
     if 'response_object' in tax_id_obj:
         return tax_id_obj
@@ -176,11 +180,12 @@ def validate_vat_number(stripe_customer_id, vat_number):
     }
 
 
-def create_tax_id(stripe_customer_id, vat_number):
+def create_tax_id(stripe_customer_id, vat_number, type):
     stripe_utils = StripeUtils()
     tax_id_response = stripe_utils.create_tax_id_for_user(
         stripe_customer_id,
         vat_number=vat_number,
+        type=type
     )
 
     tax_id_obj = tax_id_response.get('response_object')

datacenterlight/views.py

@@ -546,7 +546,8 @@ class PaymentOrderView(FormView):
             if vat_number:
                 validate_result = validate_vat_number(
                     stripe_customer_id=request.session['customer'],
-                    vat_number=address_form.cleaned_data.get('vat_number')
+                    vat_number=address_form.cleaned_data.get('vat_number'),
+                    country=address_form.cleaned_data.get("country").strip()
                 )
 
                 if 'response_object' in validate_result:

utils/stripe_utils.py

@@ -436,10 +436,10 @@ class StripeUtils(object):
         subscription.save()
 
     @handleStripeError
-    def create_tax_id_for_user(self, stripe_customer_id, vat_number):
+    def create_tax_id_for_user(self, stripe_customer_id, vat_number, type="eu_vat"):
         tax_id_obj = stripe.Customer.create_tax_id(
             stripe_customer_id,
-            type="eu_vat",
+            type=type,
             value=vat_number,
         )
         return tax_id_obj