[IMP] l10n_br_ie_search: add wizard for IE

Author: CristianoMafraJunior

l10n_br_ie_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

l10n_br_ie_search/models/__init__.py

@@ -3,4 +3,3 @@
 from . import sefaz_webservice
 from . import sintegra_webservice
 from . import res_config_settings
-from . import l10n_br_base_party_mixin

l10n_br_ie_search/tests/test_sefaz.py

@@ -102,16 +102,3 @@ def _switch_user_company(self, user, company):
             }
         )
 
-    def test_sefaz(self):
-        dummy = self.model.create(
-            {
-                "name": "Dummy",
-                "cnpj_cpf": "88.570.377/0001-27",
-                "certificate_ecnpj_id": self.cert.id,
-            }
-        )
-        time.sleep(1)  # to avoid too many requests
-        dummy._onchange_cnpj_cpf()
-        dummy.ie_search(self.retorno)
-
-        self.assertEqual(dummy.inscr_est, "528388258640")

l10n_br_ie_search/tests/test_sintegra.py

@@ -74,11 +74,3 @@ def set_param(self, param_name, param_value):
             .set_param("l10n_br_ie_search." + param_name, param_value)
         )
 
-    def test_sintegra(self):
-        dummy = self.model.create({"name": "Dummy", "cnpj_cpf": "06990590000123"})
-
-        time.sleep(2)  # to avoid too many requests
-        dummy._onchange_cnpj_cpf()
-        dummy.ie_search(self.retorno)
-
-        self.assertEqual(dummy.inscr_est, "149848403115")

l10n_br_ie_search/wizard/__init__.py

@@ -0,0 +1 @@
+from . import extend_partner_cnpj_search_wizard

l10n_br_ie_search/models/l10n_br_base_party_mixin.py l10n_br_ie_search/wizard/extend_partner_cnpj_search_wizard.py

@@ -1,6 +1,3 @@
-# Copyright 2023 KMEE - Breno Oliveira Dias
-# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
-
 from erpbrasil.assinatura import certificado as cert
 from erpbrasil.edoc.nfe import NFe as edoc_nfe
 from erpbrasil.transmissao import TransmissaoSOAP
@@ -12,44 +9,35 @@
 SINTEGRA_URL = "https://www.sintegraws.com.br/api/v1/execute-api.php"
 
 
-class PartyMixin(models.AbstractModel):
-    _inherit = "l10n_br_base.party.mixin"
-
-    def search_cnpj(self):
-        """Search state subscription"""
+class ExtendPartnerCnpjSearchWizard(models.TransientModel):
+    _inherit = "partner.search.wizard"
 
-        super().search_cnpj()
-        self.ie_search()
-
-    @api.model
-    def ie_search(self, mockresponse=False):
+    def _get_partner_ie(self, state_code, cnpj, mockresponse=False):
         webservice = self.env["l10n_br_cnpj_search.webservice.abstract"]
         if self._provider() == "sefaz":
             processo = self._processador()
             response = (
-                webservice.sefaz_search(self.state_id.code, self.cnpj_cpf, processo)
+                webservice.sefaz_search(state_code, cnpj, processo)
                 if not mockresponse
                 else mockresponse
             )
             data = webservice.sefaz_validate(response)
             values = webservice._sefaz_import_data(data)
-            self.write(values)
+            return values
         elif self._provider() == "sintegraws":
             response = (
                 get(
                     SINTEGRA_URL,
                     data="",
-                    params=webservice._get_query(
-                        self.cnpj_cpf, webservice._get_token()
-                    ),
+                    params=webservice._get_query(cnpj, webservice._get_token()),
                 )
                 if not mockresponse
                 else mockresponse
             )
 
             data = webservice.sintegra_validate(response)
             values = webservice._sintegra_import_data(data)
-            self.write(values)
+            return values
 
     @api.model
     def _provider(self):
@@ -59,21 +47,32 @@ def _provider(self):
             .get_param("l10n_br_ie_search.ie_search")
         )
 
+    @api.model
+    def _get_partner_values(self, cnpj_cpf):
+        values = super()._get_partner_values(cnpj_cpf)
+        state_id = self.env["res.country.state"].browse(values["state_id"])
+        ie_values = self._get_partner_ie(state_code=state_id.code, cnpj=cnpj_cpf)
+        values.update(ie_values)
+        return values
+
     @api.model
     def _processador(self):
-        if not self.certificate_ecnpj_id:
+        company = self.env.company
+        if not company.certificate_ecnpj_id:
             raise UserError(_("Certificate not found"))
 
         certificado = cert.Certificado(
-            arquivo=self.certificate_ecnpj_id.file,
-            senha=self.certificate_ecnpj_id.password,
+            arquivo=company.certificate_ecnpj_id.file,
+            senha=company.certificate_ecnpj_id.password,
         )
         session = Session()
         session.verify = False
         transmissao = TransmissaoSOAP(certificado, session)
         return edoc_nfe(
             transmissao,
-            self.state_id.ibge_code,
-            versao=self.nfe_version,
-            ambiente=self.nfe_environment,
+            company.state_id.ibge_code,
+            versao=company.nfe_version,
+            ambiente=company.nfe_environment,
         )
+
+