From a02065d4d872f7f37ea34c5fad5231daa2b098f6 Mon Sep 17 00:00:00 2001
From: Andrii Skrypka <andrijskrypa@ukr.net>
Date: Fri, 19 Mar 2021 18:13:18 +0200
Subject: [PATCH] [IMP] base_location_nuts: add context on partner form and
 reload page after import

---
 base_location_nuts/models/res_partner.py      |  8 ++--
 base_location_nuts/views/res_partner_view.xml | 40 +++++++++++++++----
 base_location_nuts/wizard/nuts_import.py      |  5 ++-
 3 files changed, 40 insertions(+), 13 deletions(-)

diff --git a/base_location_nuts/models/res_partner.py b/base_location_nuts/models/res_partner.py
index 5941ea0aa..2d0916d9b 100644
--- a/base_location_nuts/models/res_partner.py
+++ b/base_location_nuts/models/res_partner.py
@@ -100,21 +100,21 @@ class ResPartner(models.Model):
     def onchange_state_id_base_location_nuts(self):
         if self.state_id:
             self.country_id = self.state_id.country_id
-            if self.state_id.country_id.state_level:
+            if self.country_id.state_level:
                 nuts_state = self.env["res.partner.nuts"].search(
                     [
-                        ("level", "=", self.state_id.country_id.state_level),
+                        ("level", "=", self.country_id.state_level),
                         ("state_id", "=", self.state_id.id),
                     ],
                     limit=1,
                 )
                 if nuts_state:
-                    field = "nuts%d_id" % self.state_id.country_id.state_level
+                    field = "nuts%d_id" % self.country_id.state_level
                     self[field] = nuts_state
 
     @api.model
     def _address_fields(self):
-        fields = super(ResPartner, self)._address_fields()
+        fields = super()._address_fields()
         if fields:
             fields += ["nuts1_id", "nuts2_id", "nuts3_id", "nuts4_id"]
         return fields
diff --git a/base_location_nuts/views/res_partner_view.xml b/base_location_nuts/views/res_partner_view.xml
index 1f450d6cb..c37ae6ad2 100644
--- a/base_location_nuts/views/res_partner_view.xml
+++ b/base_location_nuts/views/res_partner_view.xml
@@ -6,19 +6,43 @@
         <field name="inherit_id" ref="base.view_partner_form" />
         <field name="arch" type="xml">
             <xpath expr="//sheet/group//field[@name='country_id']/.." position="after">
-                <field name="nuts1_id" />
-                <field name="nuts2_id" />
-                <field name="nuts3_id" />
-                <field name="nuts4_id" />
+                <field
+                    name="nuts1_id"
+                    context="{'default_country_id': country_id, 'default_level': 1}"
+                />
+                <field
+                    name="nuts2_id"
+                    context="{'default_country_id': country_id, 'default_level': 2, 'default_parent_id': nuts1_id}"
+                />
+                <field
+                    name="nuts3_id"
+                    context="{'default_country_id': country_id, 'default_level': 3, 'default_parent_id': nuts2_id}"
+                />
+                <field
+                    name="nuts4_id"
+                    context="{'default_country_id': country_id, 'default_level': 4, 'default_parent_id': nuts3_id}"
+                />
             </xpath>
             <xpath
                 expr="//field[@name='child_ids']/form//field[@name='country_id']/.."
                 position="after"
             >
-                <field name="nuts1_id" />
-                <field name="nuts2_id" />
-                <field name="nuts3_id" />
-                <field name="nuts4_id" />
+                <field
+                    name="nuts1_id"
+                    context="{'default_country_id': country_id, 'default_level': 1}"
+                />
+                <field
+                    name="nuts2_id"
+                    context="{'default_country_id': country_id, 'default_level': 2, 'default_parent_id': nuts1_id}"
+                />
+                <field
+                    name="nuts3_id"
+                    context="{'default_country_id': country_id, 'default_level': 3, 'default_parent_id': nuts2_id}"
+                />
+                <field
+                    name="nuts4_id"
+                    context="{'default_country_id': country_id, 'default_level': 4, 'default_parent_id': nuts3_id}"
+                />
             </xpath>
         </field>
     </record>
diff --git a/base_location_nuts/wizard/nuts_import.py b/base_location_nuts/wizard/nuts_import.py
index 4c5e484fb..12c74d3f6 100644
--- a/base_location_nuts/wizard/nuts_import.py
+++ b/base_location_nuts/wizard/nuts_import.py
@@ -229,4 +229,7 @@ class NutsImport(models.TransientModel):
             "The wizard to create NUTS entries from RAMON "
             "has been successfully completed."
         )
-        return True
+        return {
+            "type": "ir.actions.client",
+            "tag": "reload",
+        }