From 43e266d779f53fe72d81749f6fde4fb99aa93f72 Mon Sep 17 00:00:00 2001
From: Jared Kipe <jared@hibou.io>
Date: Mon, 12 Dec 2022 22:02:01 +0000
Subject: [PATCH] [FIX] l10n_pe_hr_payroll: add manual override for "IR 5th
 Cat." calculation

---
 l10n_pe_hr_payroll/data/ir_5ta_cat_rules.xml  | 9 +++++++++
 l10n_pe_hr_payroll/models/rules/ir_5ta_cat.py | 4 ++++
 2 files changed, 13 insertions(+)

diff --git a/l10n_pe_hr_payroll/data/ir_5ta_cat_rules.xml b/l10n_pe_hr_payroll/data/ir_5ta_cat_rules.xml
index c296bbec..7d7d73d5 100644
--- a/l10n_pe_hr_payroll/data/ir_5ta_cat_rules.xml
+++ b/l10n_pe_hr_payroll/data/ir_5ta_cat_rules.xml
@@ -44,6 +44,15 @@
 
 
     <!-- EE Rules -->
+    <record id="input_type_ee_ir_5ta_cat" model="hr.payslip.input.type">
+        <field name="name">Manual IR 5th Cat.</field>
+        <field name="code">EE_PE_IR_5TA_CAT</field>
+        <field name="country_id" ref="base.pe" />
+        <field name="struct_ids" eval="[
+            (4, ref('l10n_pe_hr_payroll.hr_payroll_structure')),
+        ]" />
+    </record>
+
     <record id="hr_payroll_rule_ee_ir_5ta_cat" model="hr.salary.rule">
         <field name="sequence" eval="196"/>
         <field name="struct_id" ref="hr_payroll_structure"/>
diff --git a/l10n_pe_hr_payroll/models/rules/ir_5ta_cat.py b/l10n_pe_hr_payroll/models/rules/ir_5ta_cat.py
index 6945274a..8164f6b0 100644
--- a/l10n_pe_hr_payroll/models/rules/ir_5ta_cat.py
+++ b/l10n_pe_hr_payroll/models/rules/ir_5ta_cat.py
@@ -8,6 +8,10 @@ _logger = logging.getLogger(__name__)
 
 
 def ir_5ta_cat(payslip, categories, worked_days, inputs):
+    if inputs.EE_PE_IR_5TA_CAT:
+        # cannot look for amount because it could be forced to zero
+        return inputs.EE_PE_IR_5TA_CAT.amount, 100.0
+    
     basic_wage = categories.BASIC
     if payslip.dict.contract_id.pe_payroll_config_value('ee_5ta_cat_exempt'):
         return 0.0, 0.0