[IMP] l10n_br_cnpj_search: add a wizard for verification of the data

CristianoMafraJunior · antoniospneto · commit bb45367ea142 · 2024-02-01T20:13:13.000-03:00
diff --git a/l10n_br_cnpj_search/__init__.py b/l10n_br_cnpj_search/__init__.py
@@ -1,3 +1,4 @@
 # License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
 
 from . import models
+from . import wizard
diff --git a/l10n_br_cnpj_search/__manifest__.py b/l10n_br_cnpj_search/__manifest__.py
@@ -16,6 +16,7 @@
     ],
     "data": [
         "security/ir.model.access.csv",
+        "wizard/partner_cnpj_search_wizard.xml",
         "views/res_partner_view.xml",
         "views/res_company_view.xml",
         "views/res_config_settings_view.xml",
diff --git a/l10n_br_cnpj_search/models/l10n_br_base_party_mixin.py b/l10n_br_cnpj_search/models/l10n_br_base_party_mixin.py
@@ -1,8 +1,7 @@
 # Copyright 2022 KMEE
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
-
-from erpbrasil.base.misc import punctuation_rm
-from requests import get
+# Copyright (C) 2024-Today - Engenere (<https://engenere.one>).
+# @author Cristiano Mafra Junior
 
 from odoo import _, api, models
 from odoo.exceptions import UserError
@@ -11,29 +10,32 @@
 class PartyMixin(models.AbstractModel):
     _inherit = "l10n_br_base.party.mixin"
 
-    def search_cnpj(self):
-        """Search CNPJ by the chosen API"""
+    def action_open_cnpj_search_wizard(self):
         if not self.cnpj_cpf:
-            raise UserError(_("Por favor insira o CNPJ"))
-
+            raise UserError(_("Please enter your CNPJ"))
         if self.cnpj_validation_disabled():
             raise UserError(
                 _(
-                    "It is necessary to activate the option to validate de CNPJ to use this "
-                    + "functionality."
+                    "It is necessary to activate the option to validate de CNPJ to use"
+                    " this functionality."
                 )
             )
-
-        cnpj_cpf = punctuation_rm(self.cnpj_cpf)
-        webservice = self.env["l10n_br_cnpj_search.webservice.abstract"]
-        response = get(
-            webservice.get_api_url(cnpj_cpf), headers=webservice.get_headers()
-        )
-
-        data = webservice.validate(response)
-        values = webservice.import_data(data)
-        values["company_type"] = "company"
-        self.write(values)
+        if self._name == "res.partner":
+            default_partner_id = self.id
+        else:
+            default_partner_id = self.partner_id.id
+
+        return {
+            "name": "Search Data by CNPJ",
+            "type": "ir.actions.act_window",
+            "res_model": "partner.search.wizard",
+            "view_type": "form",
+            "view_mode": "form",
+            "context": {
+                "default_partner_id": default_partner_id,
+            },
+            "target": "new",
+        }
 
     @api.model
     def cnpj_validation_disabled(self):
diff --git a/l10n_br_cnpj_search/models/res_partner.py b/l10n_br_cnpj_search/models/res_partner.py
@@ -1,7 +1,6 @@
 # Copyright 2022 KMEE - Luis Felipe Mileo
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 
-
 from odoo import fields, models
 
 
diff --git a/l10n_br_cnpj_search/security/ir.model.access.csv b/l10n_br_cnpj_search/security/ir.model.access.csv
@@ -2,3 +2,4 @@ id,name,model_id:id,group_id:id,perm_read,perm_write,perm_create,perm_unlink
 access_l10n_br_cnpj_search_webservice,access_l10n_br_cnpj_search_webservice,model_l10n_br_cnpj_search_webservice_abstract,base.group_user,1,1,1,1
 "l10n_br_fiscal_partner_profile_user","l10n_br_fiscal.partner_profile","l10n_br_fiscal.model_l10n_br_fiscal_partner_profile","base.group_user",1,0,0,0
 "l10n_br_fiscal_cnae","l10n_br_fiscal.cnae","l10n_br_fiscal.model_l10n_br_fiscal_cnae","base.group_user",1,0,0,0
+access_partner_search_wizard,res_partner_wizard,model_partner_search_wizard,base.group_user,1,1,1,1
diff --git a/l10n_br_cnpj_search/static/description/index.html b/l10n_br_cnpj_search/static/description/index.html
@@ -1,3 +1,4 @@
+<?xml version="1.0" encoding="utf-8"?>
 <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
 <html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en" lang="en">
 <head>
diff --git a/l10n_br_cnpj_search/tests/test_receitaws.py b/l10n_br_cnpj_search/tests/test_receitaws.py
@@ -1,8 +1,9 @@
 # Copyright 2022 KMEE
+# Copyright (C) 2024-Today - Engenere (<https://engenere.one>).
+# @author Cristiano Mafra Junior
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 
 import os
-import time  # You can't send multiple requests at the same time in trial version
 
 import vcr
 
@@ -25,11 +26,19 @@ def setUp(self):
         ignore_localhost=True,
     )
     def test_receita_ws_success(self):
-        kilian = self.model.create({"name": "Kilian", "cnpj_cpf": "44.356.113/0001-08"})
-
-        kilian._onchange_cnpj_cpf()
-        kilian.search_cnpj()
+        kilian = self.model.create(
+            {
+                "name": "Kilian",
+                "cnpj_cpf": "44.356.113/0001-08",
+            }
+        )
 
+        action_wizard = kilian.action_open_cnpj_search_wizard()
+        wizard_context = action_wizard.get("context")
+        wizard = (
+            self.env["partner.search.wizard"].with_context(wizard_context).create({})
+        )
+        wizard.action_update_partner()
         self.assertEqual(kilian.company_type, "company")
         self.assertEqual(kilian.legal_name, "Kilian Macedo Melcher 08777131460")
         self.assertEqual(kilian.name, "Kilian Macedo Melcher 08777131460")
@@ -54,9 +63,10 @@ def test_receita_ws_fail(self):
         invalido = self.model.create({"name": "invalido", "cnpj_cpf": "00000000000000"})
         invalido._onchange_cnpj_cpf()
 
-        time.sleep(2)  # Pause
         with self.assertRaises(ValidationError):
-            invalido.search_cnpj()
+            action_wizard = invalido.action_open_cnpj_search_wizard()
+            wizard_context = action_wizard.get("context")
+            self.env["partner.search.wizard"].with_context(wizard_context).create({})
 
     @vcr.use_cassette(
         os.path.dirname(__file__) + "/fixtures/test_receitaws_multiplos_telefones.yaml",
@@ -67,9 +77,12 @@ def test_receita_ws_multiple_phones(self):
         isla = self.model.create({"name": "Isla", "cnpj_cpf": "92.666.056/0001-06"})
         isla._onchange_cnpj_cpf()
 
-        time.sleep(2)  # Pause
-        isla.search_cnpj()
-
+        action_wizard = isla.action_open_cnpj_search_wizard()
+        wizard_context = action_wizard.get("context")
+        wizard = (
+            self.env["partner.search.wizard"].with_context(wizard_context).create({})
+        )
+        wizard.action_update_partner()
         self.assertEqual(isla.name.strip(), "Isla Sementes Ltda.")
         self.assertEqual(isla.phone.strip(), "(51) 9852-9561")
         self.assertEqual(isla.mobile.strip(), "(51) 2136-6600")
diff --git a/l10n_br_cnpj_search/tests/test_serpro.py b/l10n_br_cnpj_search/tests/test_serpro.py
@@ -1,9 +1,10 @@
 # Copyright 2022 KMEE
+# Copyright (C) 2024-Today - Engenere (<https://engenere.one>).
+# @author Cristiano Mafra Junior
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 
 import logging
 import os
-import time  # You can't send multiple requests at the same time in trial version
 
 import vcr
 
@@ -34,15 +35,19 @@ def test_serpro_basica(self):
         dummy_basica = self.model.create(
             {"name": "Dummy Basica", "cnpj_cpf": "34.238.864/0001-68"}
         )
-        time.sleep(3)
         dummy_basica._onchange_cnpj_cpf()
-        dummy_basica.search_cnpj()
 
-        self.assertEqual(dummy_basica.company_type, "company")
+        action_wizard = dummy_basica.action_open_cnpj_search_wizard()
+        wizard_context = action_wizard.get("context")
+        wizard = (
+            self.env["partner.search.wizard"].with_context(wizard_context).create({})
+        )
+        wizard.action_update_partner()
         self.assertEqual(
             dummy_basica.legal_name,
             "Uhieqkx Whnhiwd Nh Fixkhuuwphmvx Nh Nwnxu (Uhifix)",
         )
+        self.assertEqual(dummy_basica.company_type, "company")
         self.assertEqual(dummy_basica.name, "Uhifix Uhnh")
         self.assertEqual(dummy_basica.email, "EMPRESA@XXXXXX.BR")
         self.assertEqual(dummy_basica.street_name, "Nh Biwmnh Wihw Mxivh")
@@ -62,15 +67,16 @@ def test_serpro_basica(self):
         ignore_localhost=True,
     )
     def test_serpro_not_found(self):
-        # Na versão Trial só há alguns registros de CNPJ cadastrados
+        # In the Trial version there are only a few registered CNPJ records
         invalid = self.model.create(
             {"name": "invalid", "cnpj_cpf": "44.356.113/0001-08"}
         )
         invalid._onchange_cnpj_cpf()
 
-        time.sleep(3)  # Pause
         with self.assertRaises(ValidationError):
-            invalid.search_cnpj()
+            action_wizard = invalid.action_open_cnpj_search_wizard()
+            wizard_context = action_wizard.get("context")
+            self.env["partner.search.wizard"].with_context(wizard_context).create({})
 
     def assert_socios(self, partner, expected_cnpjs):
         socios = self.model.search_read(
@@ -126,9 +132,13 @@ def test_serpro_empresa(self):
             {"name": "Dummy Empresa", "cnpj_cpf": "34.238.864/0001-68"}
         )
 
-        time.sleep(3)  # Pause
         dummy_empresa._onchange_cnpj_cpf()
-        dummy_empresa.search_cnpj()
+        action_wizard = dummy_empresa.action_open_cnpj_search_wizard()
+        wizard_context = action_wizard.get("context")
+        wizard = (
+            self.env["partner.search.wizard"].with_context(wizard_context).create({})
+        )
+        wizard.action_update_partner()
 
         expected_cnpjs = {
             "Joana": "23982012600",
@@ -155,9 +165,13 @@ def test_serpro_qsa(self):
             {"name": "Dummy QSA", "cnpj_cpf": "34.238.864/0001-68"}
         )
 
-        time.sleep(3)  # Pause
         dummy_qsa._onchange_cnpj_cpf()
-        dummy_qsa.search_cnpj()
+        action_wizard = dummy_qsa.action_open_cnpj_search_wizard()
+        wizard_context = action_wizard.get("context")
+        wizard = (
+            self.env["partner.search.wizard"].with_context(wizard_context).create({})
+        )
+        wizard.action_update_partner()
 
         expected_cnpjs = {
             "Joana": False,
diff --git a/l10n_br_cnpj_search/views/res_company_view.xml b/l10n_br_cnpj_search/views/res_company_view.xml
@@ -17,7 +17,7 @@
                      <div class="o_row" colspan="1">
                         <field name="cnpj_cpf" />
                         <button
-                        name="search_cnpj"
+                        name="action_open_cnpj_search_wizard"
                         type="object"
                         class="btn-sm btn-link mb4 fa fa-search oe_edit_only oe_inline"
                         aria-label="Pesquisar CNPJ"
diff --git a/l10n_br_cnpj_search/views/res_partner_view.xml b/l10n_br_cnpj_search/views/res_partner_view.xml
@@ -19,7 +19,7 @@
                      <div class="o_row" colspan="4">
                         <field name="cnpj_cpf" nolabel="1" />
                         <button
-                        name="search_cnpj"
+                        name="action_open_cnpj_search_wizard"
                         type="object"
                         attrs="{'invisible':
                                 [
diff --git a/l10n_br_cnpj_search/wizard/__init__.py b/l10n_br_cnpj_search/wizard/__init__.py
@@ -0,0 +1 @@
+from . import partner_cnpj_search_wizard
diff --git a/l10n_br_cnpj_search/wizard/partner_cnpj_search_wizard.py b/l10n_br_cnpj_search/wizard/partner_cnpj_search_wizard.py
@@ -0,0 +1,108 @@
+# Copyright (C) 2024-Today - Engenere (<https://engenere.one>).
+# @author Cristiano Mafra Junior
+from erpbrasil.base.fiscal import cnpj_cpf
+from erpbrasil.base.misc import punctuation_rm
+from requests import get
+
+from odoo import api, fields, models
+
+
+class PartnerCnpjSearchWizard(models.TransientModel):
+    _name = "partner.search.wizard"
+
+    partner_id = fields.Many2one(comodel_name="res.partner")
+    provider_name = fields.Char()
+    cnpj_cpf = fields.Char()
+    legal_name = fields.Char()
+    name = fields.Char()
+    inscr_est = fields.Char()
+    zip = fields.Char()
+    street_name = fields.Char()
+    street_number = fields.Char()
+    street2 = fields.Char()
+    district = fields.Char()
+    state_id = fields.Many2one(comodel_name="res.country.state")
+    city_id = fields.Many2one(
+        comodel_name="res.city",
+        domain="[('state_id', '=', state_id)]",
+    )
+    country_id = fields.Many2one(comodel_name="res.country")
+    phone = fields.Char()
+    mobile = fields.Char()
+    email = fields.Char()
+    legal_nature = fields.Char()
+    currency_id = fields.Many2one(
+        comodel_name="res.currency",
+        default=lambda self: self.env.ref("base.BRL"),
+    )
+    equity_capital = fields.Monetary(currency_field="currency_id")
+    cnae_main_id = fields.Many2one(comodel_name="l10n_br_fiscal.cnae")
+    cnae_secondary_ids = fields.Many2many(
+        comodel_name="l10n_br_fiscal.cnae",
+        relation="wizard_fiscal_cnae_rel",
+        column1="company_id",
+        column2="cnae_id",
+    )
+    child_ids = fields.Many2many(
+        comodel_name="res.partner",
+        relation="parent_id_wizard_id",
+        column1="parent_id",
+        column2="wizard_id",
+    )
+
+    @api.onchange("cnpj_cpf")
+    def _onchange_cnpj_cpf(self):
+        self.cnpj_cpf = cnpj_cpf.formata(str(self.cnpj_cpf))
+
+    def _get_partner_values(self, cnpj_cpf):
+        webservice = self.env["l10n_br_cnpj_search.webservice.abstract"]
+        provider_name = webservice.get_provider()
+        response = get(
+            webservice.get_api_url(cnpj_cpf), headers=webservice.get_headers()
+        )
+        data = webservice.validate(response)
+        values = webservice.import_data(data)
+        values["provider_name"] = provider_name
+        values["cnpj_cpf"] = cnpj_cpf
+        return values
+
+    def default_get(self, fields):
+        res = super().default_get(fields)
+        partner_id = self._context.get("default_partner_id")
+        partner_model = self.env["res.partner"]
+        partner = partner_model.browse(partner_id)
+        cnpj_cpf = punctuation_rm(partner.cnpj_cpf)
+        values = self._get_partner_values(cnpj_cpf)
+        res.update(values)
+        return res
+
+    def action_update_partner(self):
+        values_to_update = {
+            "legal_name": self.legal_name,
+            "name": self.name,
+            "inscr_est": self.inscr_est,
+            "zip": self.zip,
+            "street_name": self.street_name,
+            "street_number": self.street_number,
+            "street2": self.street2,
+            "district": self.district,
+            "state_id": self.state_id.id,
+            "city_id": self.city_id.id,
+            "country_id": self.country_id.id,
+            "phone": self.phone,
+            "mobile": self.mobile,
+            "email": self.email,
+            "legal_nature": self.legal_nature,
+            "equity_capital": self.equity_capital,
+            "cnae_main_id": self.cnae_main_id,
+            "cnae_secondary_ids": self.cnae_secondary_ids,
+            "child_ids": [(6, 0, self.child_ids.ids)],
+            "company_type": "company",
+        }
+        non_empty_values = {
+            key: value for key, value in values_to_update.items() if value
+        }
+        if non_empty_values:
+            # Update partner only if there are non-empty values
+            self.partner_id.write(non_empty_values)
+        return {"type": "ir.actions.act_window_close"}
diff --git a/l10n_br_cnpj_search/wizard/partner_cnpj_search_wizard.xml b/l10n_br_cnpj_search/wizard/partner_cnpj_search_wizard.xml

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,4 @@`
`1`	`1`	`# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).`
`2`	`2`
`3`	`3`	`from . import models`
	`4`	`+from . import wizard`
Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,4 @@`
	`1`	`+<?xml version="1.0" encoding="utf-8"?>`
`1`	`2`	`<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">`
`2`	`3`	`<html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en" lang="en">`
`3`	`4`	`<head>`
Original file line number	Diff line number	Diff line change
`@@ -19,7 +19,7 @@`
`19`	`19`	`<div class="o_row" colspan="4">`
`20`	`20`	`<field name="cnpj_cpf" nolabel="1" />`
`21`	`21`	`<button`
`22`		`- name="search_cnpj"`
	`22`	`+ name="action_open_cnpj_search_wizard"`
`23`	`23`	`type="object"`
`24`	`24`	`attrs="{'invisible':`
`25`	`25`	`[`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+from . import partner_cnpj_search_wizard`