From 5e97d70a5e6d6403304e0599ca8917fa6bad8750 Mon Sep 17 00:00:00 2001
From: PCoder <purple.coder@yahoo.co.uk>
Date: Tue, 17 Dec 2019 22:52:22 +0530
Subject: [PATCH] Save VAT number

---
 datacenterlight/views.py | 5 +++++
 hosting/views.py         | 3 +++
 2 files changed, 8 insertions(+)

diff --git a/datacenterlight/views.py b/datacenterlight/views.py
index 8ed0b794..d93fcc2d 100644
--- a/datacenterlight/views.py
+++ b/datacenterlight/views.py
@@ -942,6 +942,11 @@ class OrderConfirmationView(DetailView, FormView):
             'user': custom_user.id
         })
 
+        # Customer is created, we save his VAT Number
+        custom_user.vat_number = request.session.get(
+            'billing_address_data').get("vat_number")
+        custom_user.save()
+
         if 'generic_payment_type' in request.session:
             stripe_cus = StripeCustomer.objects.filter(
                 stripe_id=stripe_api_cus_id
diff --git a/hosting/views.py b/hosting/views.py
index 21ede03e..b81c5ea3 100644
--- a/hosting/views.py
+++ b/hosting/views.py
@@ -616,6 +616,9 @@ class SettingsView(LoginRequiredMixin, FormView):
                     instance=self.request.user.billing_addresses.first(),
                     data=billing_address_data)
                 billing_address_user_form.save()
+                self.request.user.vat_number = billing_address_data.get(
+                    "vat_number")
+                self.request.user.save()
                 msg = _("Billing address updated successfully")
                 messages.add_message(request, messages.SUCCESS, msg)
             else: