From 792b902488a55a416c51b1a4403b4d458a155f9e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dar=C3=ADo=20Lodeiros?= <dario@commitsun.com>
Date: Tue, 5 Nov 2024 10:27:46 +0100
Subject: [PATCH] [IMP]pms: compute_document_id

---
 pms/models/pms_checkin_partner.py         | 17 +++++++++++------
 pms_l10n_es/models/pms_checkin_partner.py |  5 +++++
 2 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/pms/models/pms_checkin_partner.py b/pms/models/pms_checkin_partner.py
index ecb824cd4..325d4fe17 100644
--- a/pms/models/pms_checkin_partner.py
+++ b/pms/models/pms_checkin_partner.py
@@ -509,19 +509,24 @@ class PmsCheckinPartner(models.Model):
                         )
                     )
                     if not id_number_id:
+                        document_vals = record.get_document_vals()
                         id_number_id = self.env["res.partner.id_number"].create(
-                            {
-                                "partner_id": record.partner_id.id,
-                                "name": record.document_number,
-                                "category_id": record.document_type.id,
-                                "valid_from": record.document_expedition_date,
-                            }
+                            document_vals
                         )
 
                     record.document_id = id_number_id
             else:
                 record.document_id = False
 
+    def get_document_vals(self):
+        return {
+            "name": self.document_number,
+            "partner_id": self.partner_id.id,
+            "category_id": self.document_type.id,
+            "valid_from": self.document_expedition_date,
+            "country_id": self.document_country_id.id,
+        }
+
     @api.depends(
         "document_number",
         "document_type",
diff --git a/pms_l10n_es/models/pms_checkin_partner.py b/pms_l10n_es/models/pms_checkin_partner.py
index 02122cc61..d3d9620e8 100644
--- a/pms_l10n_es/models/pms_checkin_partner.py
+++ b/pms_l10n_es/models/pms_checkin_partner.py
@@ -110,6 +110,11 @@ class PmsCheckinPartner(models.Model):
         manual_fields.extend(["support_number"])
         return manual_fields
 
+    def get_document_vals(self):
+        vals = super(PmsCheckinPartner, self).get_document_vals()
+        vals["support_number"] = self.support_number
+        return vals
+
     def write(self, vals):
         result = super(PmsCheckinPartner, self).write(vals)
         for record in self: