Merge PR #1078 into 16.0

Signed-off-by dhongu
OCA · Sep 29, 2024 · 5d0b98a · 5d0b98a
2 parents ed949e5 + cc92b7d
commit 5d0b98a
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 2 deletions.
diff --git a/l10n_ro_message_spv/models/account_move.py b/l10n_ro_message_spv/models/account_move.py
@@ -14,3 +14,14 @@ class AccountMove(models.Model):
         string="Romania - E-invoice messages",
         help="E-invoice messages related to this invoice.",
     )
+
+    def unlink(self):
+        domain = [("invoice_id", "in", self.ids)]
+        message_spv_ids = self.env["l10n.ro.message.spv"].search(domain)
+        attachments = self.env["ir.attachment"]
+        attachments += message_spv_ids.mapped("attachment_id")
+        attachments += message_spv_ids.mapped("attachment_xml_id")
+        attachments += message_spv_ids.mapped("attachment_anaf_pdf_id")
+        attachments += message_spv_ids.mapped("attachment_embedded_pdf_id")
+        attachments.sudo().write({"res_id": False, "res_model": False})
+        return super().unlink()
diff --git a/l10n_ro_message_spv/models/message_spv.py b/l10n_ro_message_spv/models/message_spv.py
@@ -237,6 +237,7 @@ def check_anaf_message_xml(self, zip_content):
         return info_msg
 
     def get_invoice_from_move(self):
+        self.get_partner()
         messages_without_invoice = self.filtered(lambda m: not m.invoice_id)
         message_ids = messages_without_invoice.mapped("name")
         request_ids = messages_without_invoice.mapped("request_id")
@@ -310,6 +311,7 @@ def get_invoice_from_move(self):
         self.get_data_from_invoice()
 
     def get_data_from_invoice(self):
+        self.get_partner()
         for message in self:
             if not message.invoice_id:
                 continue
@@ -324,7 +326,8 @@ def get_data_from_invoice(self):
 
             message.write(
                 {
-                    "partner_id": message.invoice_id.commercial_partner_id.id,
+                    "partner_id": message.invoice_id.commercial_partner_id.id
+                    or message.partner_id.id,
                     "invoice_amount": invoice_amount,
                     "state": state,
                 }
@@ -336,11 +339,12 @@ def get_data_from_invoice(self):
                 attachments += message.attachment_xml_id
                 attachments += message.attachment_anaf_pdf_id
                 attachments += message.attachment_embedded_pdf_id
-                attachments.write(
+                attachments.sudo().write(
                     {"res_id": message.invoice_id.id, "res_model": "account.move"}
                 )
 
     def create_invoice(self):
+        self.get_partner()
         for message in self.filtered(lambda m: not m.invoice_id):
             if not message.message_type == "in_invoice":
                 continue
@@ -473,3 +477,19 @@ def _action_download(self, attachment_field_id):
             "url": f"/web/content/{attachment_field_id}?download=true",
             "target": "self",
         }
+
+    def get_partner(self):
+
+        for message in self.filtered(lambda m: not m.partner_id):
+            if message.cif:
+                domain = [("vat", "like", message.cif), ("is_company", "=", True)]
+                partner = self.env["res.partner"].search(domain, limit=1)
+                if not partner:
+                    partner = self.env["res.partner"].create(
+                        {
+                            "name": message.cif,
+                            "vat": message.cif,
+                            "is_company": True,
+                        }
+                    )
+                message.write({"partner_id": partner.id})