[IMP] l10n_br_ie_search: add assistant to 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/fixtures/test_sefaz.yaml

@@ -0,0 +1,66 @@
+interactions:
+  - request:
+      body: null
+      headers:
+        Accept:
+          - "*/*"
+        Accept-Encoding:
+          - gzip, deflate
+        Authorization:
+          - Bearer 06aef429-a981-3ec5-a1f8-71d38d86481e
+        Connection:
+          - keep-alive
+        User-Agent:
+          - python-requests/2.25.1
+      method: GET
+      uri: https://www.receitaws.com.br/v1/cnpj/88570377000127
+    response:
+      body:
+        string:
+          "{\"abertura\": \"24/11/2021\", \"situacao\": \"BAIXADA\", \"tipo\":
+          \"MATRIZ\", \"nome\": \"Dummy Empresa\", \"porte\": \"MICRO EMPRESA\",
+          \"natureza_juridica\": \"213-5 - Empres\\u00e1rio (Individual)\",
+          \"logradouro\": \"RUA DUMMY\", \"numero\": \"250\", \"complemento\": \"BLOCO
+          E;APT 302\", \"municipio\": \"SAO PAULO\", \"bairro\": \"VILA FELIZ\", \"uf\":
+          \"SP\", \"cep\": \"01222001\", \"email\": \"kilian.melcher@gmail.com\",
+          \"telefone\": \"(83) 8665-0905\", \"data_situacao\": \"18/12/2023\",
+          \"motivo_situacao\": \"EXTIN\\u00c7\\u00c3O POR ENCERRAMENTO
+          LIQUIDA\\u00c7\\u00c3O VOLUNT\\u00c1RIA\", \"cnpj\": \"88570377000127\",
+          \"ultima_atualizacao\": \"2024-01-13T23:59:59.000Z\", \"status\": \"OK\",
+          \"fantasia\": \"\", \"efr\": \"\", \"situacao_especial\": \"\",
+          \"data_situacao_especial\": \"\", \"atividade_principal\": [{\"code\":
+          \"4751-2/01\", \"text\": \"********\"}], \"atividades_secundarias\":
+          [{\"code\": \"00.00-0-00\", \"text\": \"Não informada\"}], \"capital_social\":
+          \"3000.00\", \"qsa\": [], \"extra\": {}, \"billing\": {\"free\": true,
+          \"database\": true}}"
+      headers:
+        Access-Control-Allow-Headers:
+          - authorization,Access-Control-Allow-Origin,Content-Type,SOAPAction,apikey,Internal-Key,x-cpf-usuario,signature,x-signature,x-credencial,X-Request-Trace-Id
+        Access-Control-Allow-Methods:
+          - GET
+        Access-Control-Expose-Headers:
+          - ""
+        Connection:
+          - keep-alive
+        Date:
+          - Wed, 28 Jun 2023 13:45:28 GMT
+        Transfer-Encoding:
+          - chunked
+        access-control-allow-origin:
+          - "*"
+        activityid:
+          - 0c70c470-9419-4869-aa01-b964c345f5ac
+        content-type:
+          - application/json; charset=utf-8
+        etag:
+          - W/"86a-p2CeQnt0OF5zs4vPG4Lpgnsxv8k"
+        strict-transport-security:
+          - max-age=15768000
+        x-content-type-options:
+          - nosniff
+        x-powered-by:
+          - Express
+      status:
+        code: 200
+        message: OK
+version: 1

l10n_br_ie_search/tests/test_sefaz.py

@@ -1,13 +1,16 @@
 # Copyright 2023 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 time
+import os
 from datetime import timedelta
+from unittest.mock import patch
 
+import vcr
 from erpbrasil.assinatura import misc
 
 from odoo import fields
-from odoo.tests import tagged
+from odoo.tests import Form, tagged
 from odoo.tests.common import SavepointCase
 from odoo.tools.misc import format_date
 
@@ -102,16 +105,30 @@ def _switch_user_company(self, user, company):
             }
         )
 
+    @vcr.use_cassette(
+        os.path.dirname(__file__) + "/fixtures/test_sefaz.yaml",
+        match_on=["method", "scheme", "host", "port", "path", "query", "body"],
+        ignore_localhost=True,
+    )
     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")
+        with patch(
+            "odoo.addons.l10n_br_ie_search.models.sefaz_webservice.SefazWebservice.sefaz_search"
+        ) as mock_get_partner_ie:
+            mock_get_partner_ie.return_value = self.retorno
+            dummy = self.model.create(
+                {
+                    "name": "Dummy",
+                    "cnpj_cpf": "88.570.377/0001-27",
+                    "certificate_ecnpj_id": self.cert.id,
+                }
+            )
+            dummy._onchange_cnpj_cpf()
+            action_wizard = dummy.action_open_cnpj_search_wizard()
+            wizard_context = action_wizard.get("context")
+            wizard = Form(
+                self.env["partner.search.wizard"]
+                .with_context(wizard_context)
+                .create({})
+            ).save()
+            wizard.action_update_partner()
+            self.assertEqual(dummy.inscr_est, "528388258640")

l10n_br_ie_search/tests/test_sintegra.py

@@ -1,10 +1,12 @@
 # Copyright 2023 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 time  # You can't send multiple requests at the same time in trial version
+from unittest.mock import patch
 
-from odoo.tests import tagged
+from odoo.tests import Form, tagged
 from odoo.tests.common import SavepointCase
 
 _logger = logging.getLogger(__name__)
@@ -74,11 +76,15 @@ def set_param(self, param_name, param_value):
             .set_param("l10n_br_ie_search." + param_name, param_value)
         )
 
-    def test_sintegra(self):
+    @patch("requests.get")
+    def test_sintegra(self, mock_get_partner_ie):
+        mock_get_partner_ie.return_value = self.retorno
         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)
-
+        action_wizard = dummy.action_open_cnpj_search_wizard()
+        wizard_context = action_wizard.get("context")
+        wizard = Form(
+            self.env["partner.search.wizard"].with_context(wizard_context).create({})
+        ).save()
+        wizard.action_update_partner()
         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,55 +1,40 @@
 # Copyright 2023 KMEE - Breno Oliveira Dias
+# 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 requests
 from erpbrasil.assinatura import certificado as cert
 from erpbrasil.edoc.nfe import NFe as edoc_nfe
 from erpbrasil.transmissao import TransmissaoSOAP
-from requests import Session, get
+from requests import Session
 
 from odoo import _, api, models
 from odoo.exceptions import UserError
 
 SINTEGRA_URL = "https://www.sintegraws.com.br/api/v1/execute-api.php"
 
 
-class PartyMixin(models.AbstractModel):
-    _inherit = "l10n_br_base.party.mixin"
+class ExtendPartnerCnpjSearchWizard(models.TransientModel):
+    _inherit = "partner.search.wizard"
 
-    def search_cnpj(self):
-        """Search state subscription"""
-
-        super().search_cnpj()
-        self.ie_search()
-
-    @api.model
-    def ie_search(self, mockresponse=False):
+    def _get_partner_ie(self, state_code, cnpj):
         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)
-                if not mockresponse
-                else mockresponse
-            )
+            response = webservice.sefaz_search(state_code, cnpj, processo)
             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()
-                    ),
-                )
-                if not mockresponse
-                else mockresponse
+            response = requests.get(
+                SINTEGRA_URL,
+                data="",
+                params=webservice._get_query(cnpj, webservice._get_token()),
             )
-
             data = webservice.sintegra_validate(response)
             values = webservice._sintegra_import_data(data)
-            self.write(values)
+            return values
 
     @api.model
     def _provider(self):
@@ -59,6 +44,15 @@ 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)
+        if ie_values:
+            values.update(ie_values)
+        return values
+
     @api.model
     def _processador(self):
         company = self.env.company