[#2931] Update company name + address from KVK API on login

Author: Paul Schilling

src/open_inwoner/accounts/signals.py

@@ -26,6 +26,25 @@
 }
 
 
+from open_inwoner.kvk.views import company_branch_selected
+
+
+@receiver(company_branch_selected)
+def update_company_from_kvk_api(sender, *args, **kwargs):
+    request = kwargs["request"]
+    user = request.user
+
+    company_data = kwargs["company_data"]
+    user.company_name = company_data["name"]
+
+    if city := company_data.get("city", None):
+        user.city = city
+    if street := company_data.get("street", None):
+        user.street = street
+
+    user.save()
+
+
 @receiver(user_logged_in)
 def update_user_from_klant_on_login(sender, user, request, *args, **kwargs):
     # This additional guard is mainly to facilitate easier testing, where not

src/open_inwoner/kvk/signals.py

@@ -1,6 +1,7 @@
 import logging
 
 from django.contrib.auth.signals import user_logged_in
+from django.db.models.signals import Signal
 from django.dispatch import receiver
 from django.utils.translation import gettext as _
 
@@ -34,3 +35,6 @@ def on_kvk_change(sender, user, request, *args, **kwargs):
             user.rsin = rsin
             user.is_prepopulated = True
             user.save()
+
+
+company_branch_selected = Signal()

src/open_inwoner/kvk/views.py

@@ -11,6 +11,7 @@
 from ..utils.url import get_next_url_from
 from .client import KvKClient
 from .forms import CompanyBranchChoiceForm
+from .signals import company_branch_selected
 
 
 class CompanyBranchChoiceView(FormView):
@@ -90,6 +91,27 @@ def post(self, request):
             # Directly calling `super().form_invalid(form)` would override the error
             return self.render_to_response(context)
 
-        request.session[KVK_BRANCH_SESSION_VARIABLE] = request.POST["branch_number"]
+        branch_number = request.POST["branch_number"]
+        request.session[KVK_BRANCH_SESSION_VARIABLE] = branch_number
+
+        # data for update company signal
+        company_branch = (
+            next(
+                branch
+                for branch in form.company_branches
+                if branch["vestigingsnummer"] == branch_number
+            )
+            or form.company_branches[0]
+        )
+        company_data = {"name": company_branch["naam"]}
+        if company_address := company_branch.get("adres", None):
+            company_data["city"] = company_address["binnenlandsAdres"]["plaats"]
+            company_data["street"] = company_address["binnenlandsAdres"]["straatnaam"]
+
+        company_branch_selected.send(
+            sender=self,
+            request=request,
+            company_data=company_data,
+        )
 
         return HttpResponseRedirect(redirect)