initial commit 2

1 files changed, 45 insertions, 0 deletions

diff --git a/addons/hr_expense/wizard/hr_expense_refuse_reason.py b/addons/hr_expense/wizard/hr_expense_refuse_reason.py
new file mode 100644
index 00000000..f457c039
--- /dev/null
+++ b/addons/hr_expense/wizard/hr_expense_refuse_reason.py
@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import api, fields, models
+
+
+class HrExpenseRefuseWizard(models.TransientModel):
+    """This wizard can be launched from an he.expense (an expense line)
+    or from an hr.expense.sheet (En expense report)
+    'hr_expense_refuse_model' must be passed in the context to differentiate
+    the right model to use.
+    """
+
+    _name = "hr.expense.refuse.wizard"
+    _description = "Expense Refuse Reason Wizard"
+
+    reason = fields.Char(string='Reason', required=True)
+    hr_expense_ids = fields.Many2many('hr.expense')
+    hr_expense_sheet_id = fields.Many2one('hr.expense.sheet')
+
+    @api.model
+    def default_get(self, fields):
+        res = super(HrExpenseRefuseWizard, self).default_get(fields)
+        active_ids = self.env.context.get('active_ids', [])
+        refuse_model = self.env.context.get('hr_expense_refuse_model')
+        if refuse_model == 'hr.expense':
+            res.update({
+                'hr_expense_ids': active_ids,
+                'hr_expense_sheet_id': False,
+            })
+        elif refuse_model == 'hr.expense.sheet':
+            res.update({
+                'hr_expense_sheet_id': active_ids[0] if active_ids else False,
+                'hr_expense_ids': [],
+            })
+        return res
+
+    def expense_refuse_reason(self):
+        self.ensure_one()
+        if self.hr_expense_ids:
+            self.hr_expense_ids.refuse_expense(self.reason)
+        if self.hr_expense_sheet_id:
+            self.hr_expense_sheet_id.refuse_sheet(self.reason)
+
+        return {'type': 'ir.actions.act_window_close'}