[REF] l10n_br_cnpj_search: adapt to use l10n_br_fiscal.legal.nature object

Author: renatonlima

l10n_br_cnpj_search/models/__init__.py

@@ -3,5 +3,4 @@
 from . import l10n_br_base_party_mixin
 from . import res_config_settings
 from . import cnpj_webservice
-from . import res_partner
 from . import res_company

l10n_br_cnpj_search/models/cnpj_webservice.py

@@ -116,16 +116,27 @@ def _validate(self, response):
         if response.status_code != 200:
             raise ValidationError(_("%s") % response.reason)
 
+    @api.model
+    def _get_legal_nature(self, raw_code):
+        code = punctuation_rm(raw_code)
+        legal_nature_id = False
+        if code:
+            legal_nature_id = (
+                self.env["l10n_br_fiscal.legal.nature"]
+                .search([("code_unmasked", "=", code)])
+                .id
+            )
+        return legal_nature_id
+
     @api.model
     def _get_cnae(self, raw_code):
         code = punctuation_rm(raw_code)
         cnae_id = False
 
         if code:
-            formatted_code = code[0:4] + "-" + code[4] + "/" + code[5:]
             cnae_id = (
                 self.env["l10n_br_fiscal.cnae"]
-                .search([("code", "=", formatted_code)])
+                .search([("code_unmasked", "=", code)])
                 .id
             )
 
@@ -168,7 +179,7 @@ def _receitaws_import_data(self, data):
             "district": self.get_data(data, "bairro", title=True),
             "street_number": self.get_data(data, "numero"),
             "zip": self.get_data(data, "cep"),
-            "legal_nature": self.get_data(data, "natureza_juridica"),
+            "legal_nature_id": self._receitaws_get_legal_nature(data),
             "phone": phone,
             "mobile": mobile,
             "state_id": state_id,
@@ -221,6 +232,16 @@ def _get_state_city(self, data):
 
         return [state_id, city_id]
 
+    @api.model
+    def _receitaws_get_legal_nature(self, data):
+        legal_nature = data.get("natureza_juridica")
+        if legal_nature:
+            legal_nature = legal_nature.split(" - ")
+            if len(legal_nature) > 1:
+                legal_nature_code = legal_nature[0]
+                return self._get_legal_nature(legal_nature_code)
+        return False
+
     @api.model
     def _receitaws_get_cnae(self, data):
         if data.get("atividade_principal"):
@@ -280,7 +301,6 @@ def _serpro_import_data(self, data):
         name = fantasy_name if fantasy_name else legal_name
         phone, mobile = self._serpro_get_phones(data)
         address = data.get("endereco")
-        nature = data.get("naturezaJuridica")
         cep = self.get_data(address, "cep")
 
         res = {
@@ -291,8 +311,7 @@ def _serpro_import_data(self, data):
             "street2": self.get_data(address, "complemento", title=True),
             "district": self.get_data(address, "bairro", title=True),
             "street_number": self.get_data(address, "numero"),
-            "legal_nature": self.get_data(nature, "codigo", title=True)
-            + self.get_data(nature, "descricao", title=True),
+            "legal_nature_id": self._serpro_get_legal_nature(data),
             "zip": cep,
             "phone": phone,
             "mobile": mobile,
@@ -388,9 +407,14 @@ def _get_city_id(self, cep):
 
         return cep_values.get("city_id")
 
+    @api.model
+    def _serpro_get_legal_nature(self, data):
+        legal_nature = data.get("naturezaJuridica")
+        legal_nature_code = self.get_data(legal_nature, "codigo")
+        return self._get_legal_nature(legal_nature_code)
+
     @api.model
     def _serpro_get_cnae(self, data):
         cnae_main = data.get("cnaePrincipal")
         cnae_code = self.get_data(cnae_main, "codigo")
-
         return self._get_cnae(cnae_code)

l10n_br_cnpj_search/models/l10n_br_base_party_mixin.py

@@ -14,7 +14,10 @@ class PartyMixin(models.AbstractModel):
 
     cnae_main_id = fields.Many2one(comodel_name="l10n_br_fiscal.cnae")
 
-    legal_nature = fields.Char()
+    legal_nature_id = fields.Many2one(
+        comodel_name="l10n_br_fiscal.legal.nature",
+        string="Legal Nature",
+    )
 
     br_currency_id = fields.Many2one(
         comodel_name="res.currency",

l10n_br_cnpj_search/models/res_company.py

@@ -12,8 +12,6 @@ class ResCompany(models.Model):
 
     mobile = fields.Char(related="partner_id.mobile")
 
-    legal_nature = fields.Char(related="partner_id.legal_nature")
-
     company_currency_id = fields.Many2one(
         "res.currency",
         related="currency_id",

l10n_br_cnpj_search/models/res_partner.py

@@ -6,12 +6,9 @@
         <field name="inherit_id" ref="l10n_br_fiscal.fiscal_res_company_form" />
         <field name="priority">99</field>
         <field name="arch" type="xml">
-            <field name="cnae_main_id" position="after">
-                <field name="cnae_secondary_ids" widget="many2many_tags" />
+            <field name="cnae_secondary_ids" position="after">
                 <field name="equity_capital" widget="monetary" />
-                <field name="legal_nature" />
             </field>
-
             <field name="cnpj_cpf" position="replace">
                 <label for="cnpj_cpf" string="CNPJ" />
                      <div class="o_row" colspan="1">

l10n_br_cnpj_search/views/res_company_view.xml

@@ -9,10 +9,8 @@
         <field name="priority">99</field>
         <field name="inherit_id" ref="l10n_br_fiscal.partner_form" />
         <field name="arch" type="xml">
-            <field name="cnae_main_id" position="after">
-                <field name="cnae_secondary_ids" widget="many2many_tags" />
+            <field name="cnae_secondary_ids" position="after">
                 <field name="equity_capital" widget="monetary" />
-                <field name="legal_nature" />
             </field>
 
             <field name="cnpj_cpf" position="replace">

l10n_br_cnpj_search/views/res_partner_view.xml

@@ -36,7 +36,7 @@ class PartnerCnpjSearchWizard(models.TransientModel):
     phone = fields.Char()
     mobile = fields.Char()
     email = fields.Char()
-    legal_nature = fields.Char()
+    legal_nature_id = fields.Many2one(comodel_name="l10n_br_fiscal.legal.nature")
     currency_id = fields.Many2one(
         comodel_name="res.currency",
         default=lambda self: self.env.ref("base.BRL"),
@@ -112,7 +112,7 @@ def action_update_partner(self):
             "phone": self.phone,
             "mobile": self.mobile,
             "email": self.email,
-            "legal_nature": self.legal_nature,
+            "legal_nature_id": self.legal_nature_id,
             "equity_capital": self.equity_capital,
             "cnae_main_id": self.cnae_main_id,
             "cnae_secondary_ids": self.cnae_secondary_ids,

l10n_br_cnpj_search/wizard/partner_cnpj_search_wizard.py

@@ -40,7 +40,7 @@
                     </group>
 
                     <group name="Other Information" string="Other Information">
-                        <field name="legal_nature" />
+                        <field name="legal_nature_id" />
                         <field name="equity_capital" widget="monetary" />
                         <field name="cnae_main_id" string="Main CNAE" />
                         <field