From acf521444a025c365f9decf6b97873987d2cda35 Mon Sep 17 00:00:00 2001
From: Kaynnan Lemes <kaynnanx1@gmail.com>
Date: Mon, 15 Jul 2024 11:02:32 -0300
Subject: [PATCH] [IMP] l10n_br_account_withholding: handling of partner_id
 assignment

---
 l10n_br_account_withholding/__manifest__.py   |  1 +
 .../models/__init__.py                        |  1 +
 .../models/account_move.py                    | 16 ++++++++++++++-
 .../models/res_partner.py                     | 20 +++++++++++++++++++
 .../views/res_partner.xml                     | 19 ++++++++++++++++++
 5 files changed, 56 insertions(+), 1 deletion(-)
 create mode 100644 l10n_br_account_withholding/models/res_partner.py
 create mode 100644 l10n_br_account_withholding/views/res_partner.xml

diff --git a/l10n_br_account_withholding/__manifest__.py b/l10n_br_account_withholding/__manifest__.py
index a84e12d3e499..27a858a077bd 100644
--- a/l10n_br_account_withholding/__manifest__.py
+++ b/l10n_br_account_withholding/__manifest__.py
@@ -14,6 +14,7 @@
         "l10n_br_account",
     ],
     "data": [
+        "views/res_partner.xml",
         "views/l10n_br_fiscal_tax_group.xml",
         "views/account_move.xml",
     ],
diff --git a/l10n_br_account_withholding/models/__init__.py b/l10n_br_account_withholding/models/__init__.py
index e958216983e8..d5d20ae7e74a 100644
--- a/l10n_br_account_withholding/models/__init__.py
+++ b/l10n_br_account_withholding/models/__init__.py
@@ -1,3 +1,4 @@
 from . import account_move
 from . import l10n_br_fiscal_tax_group
 from . import account_move_line
+from . import res_partner
diff --git a/l10n_br_account_withholding/models/account_move.py b/l10n_br_account_withholding/models/account_move.py
index 68918e32b4fa..918b164acfd0 100644
--- a/l10n_br_account_withholding/models/account_move.py
+++ b/l10n_br_account_withholding/models/account_move.py
@@ -78,8 +78,22 @@ def _prepare_wh_invoice(self, move_line, fiscal_group):
         """
         wh_date_invoice = move_line.move_id.date
         wh_due_invoice = wh_date_invoice.replace(day=fiscal_group.wh_due_day)
+
+        if fiscal_group.tax_scope == "city":
+            city_id = (
+                self.invoice_line_ids[0].issqn_fg_city_id
+                if self.invoice_line_ids[0].issqn_fg_city_id
+                else self.partner_id.city_id
+            )
+            partner_wh = self.env["res.partner"].search(
+                [("city_id", "=", city_id.id), ("wh_cityhall", "=", True)], limit=1
+            )
+            partner_id = partner_wh if partner_wh else fiscal_group.partner_id
+        else:
+            partner_id = fiscal_group.partner_id
+
         values = {
-            "partner_id": fiscal_group.partner_id.id,
+            "partner_id": partner_id.id,
             "date": wh_date_invoice,
             "invoice_date": wh_date_invoice,
             "invoice_date_due": wh_due_invoice + relativedelta(months=1),
diff --git a/l10n_br_account_withholding/models/res_partner.py b/l10n_br_account_withholding/models/res_partner.py
new file mode 100644
index 000000000000..8b2fc34e5bf3
--- /dev/null
+++ b/l10n_br_account_withholding/models/res_partner.py
@@ -0,0 +1,20 @@
+# Copyright 2024 - TODAY, Kaynnan Lemes <kaynnan.lemes@escodoo.com.br>
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
+
+from odoo import fields, models
+
+
+class ResPartner(models.Model):
+
+    _inherit = "res.partner"
+
+    # TODO: Add WH fields for State and Country
+    wh_cityhall = fields.Boolean(string="Is City Hall?")
+
+    _sql_constraints = [
+        (
+            "unique_wh_cityhall",
+            "UNIQUE(city_id, wh_cityhall)",
+            "Only one partner with the same City Hall can exist in the same city.",
+        ),
+    ]
diff --git a/l10n_br_account_withholding/views/res_partner.xml b/l10n_br_account_withholding/views/res_partner.xml
new file mode 100644
index 000000000000..8b4ef4f8cc1c
--- /dev/null
+++ b/l10n_br_account_withholding/views/res_partner.xml
@@ -0,0 +1,19 @@
+<?xml version="1.0" encoding="utf-8" ?>
+<!-- Copyright 2024 - TODAY, Kaynnan Lemes <kaynnan.lemes@escodoo.com.br>
+     License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl). -->
+<odoo>
+
+    <record model="ir.ui.view" id="partner_form_view">
+        <field
+            name="name"
+        >l10n_br_fiscal.partner.form (in l10n_br_account_withholding)</field>
+        <field name="model">res.partner</field>
+        <field name="inherit_id" ref="l10n_br_base.l10n_br_base_partner_form" />
+        <field name="arch" type="xml">
+            <xpath expr="//group[@name='other_infos']" position="inside">
+                <field name="wh_cityhall" />
+            </xpath>
+        </field>
+    </record>
+
+</odoo>