from odoo import fields, models, _
from odoo.exceptions import UserError

class AccountPayment(models.Model):
    _inherit = 'account.payment'

    realization_id = fields.Many2one('hr.expense.realization', string='Originating Realization', readonly=True)

    def action_post(self):
        """
        Force synchronization before posting to ensure deductions are included.
        We use the 'skip_expense_lock' context to bypass hr_expense's strict validation.
        """
        return super(AccountPayment, self.with_context(skip_expense_lock=True)).action_post()

    def _synchronize_to_moves(self, changed_fields):
        # Trigger deduction sync if needed
        if 'deduction_line_ids' in changed_fields or 'amount_substract' in changed_fields:
            if 'amount' not in changed_fields:
                changed_fields = set(changed_fields) | {'amount'}

        if self._context.get('skip_expense_lock'):
            # Safely call super() - the hr_expense override also checks context (via our own override below)
            return super()._synchronize_to_moves(changed_fields)

        # For manual edits, standard hr_expense error will apply via the super() chain
        return super()._synchronize_to_moves(changed_fields)

    def _synchronize_from_moves(self, changed_fields):
        if self._context.get('skip_expense_lock'):
            return super()._synchronize_from_moves(changed_fields)
        return super()._synchronize_from_moves(changed_fields)

    def write(self, vals):
        if 'deduction_line_ids' in vals or 'amount_substract' in vals:
            # Force trigger sync with bypass flag
            return super(AccountPayment, self.with_context(skip_expense_lock=True)).write(vals)
        return super().write(vals)

    def write(self, vals):
        """
        Override write to manually trigger synchronization when deductions change.
        We include 'amount' in the changed_fields to trick Odoo's core sync engine
        into refreshing the move lines, which will then include the deductions.
        """
        res = super().write(vals)
        if 'deduction_line_ids' in vals or 'amount_substract' in vals:
            # Force trigger sync by pretending amount changed if deductions changed
            self._synchronize_to_moves({'amount', 'deduction_line_ids', 'amount_substract'})
        return res

    def action_cancel(self):
        res = super().action_cancel()
        for payment in self:
            if payment.expense_sheet_id:
                payment.expense_sheet_id.invalidate_recordset(['state'])
            # Also check if it's linked via realization
            if payment.realization_id and payment.realization_id.expense_sheet_id:
                payment.realization_id.expense_sheet_id.invalidate_recordset(['state'])
        return res

    def action_draft(self):
        res = super().action_draft()
        for payment in self:
            if payment.expense_sheet_id:
                payment.expense_sheet_id.invalidate_recordset(['state'])
            if payment.realization_id and payment.realization_id.expense_sheet_id:
                payment.realization_id.expense_sheet_id.invalidate_recordset(['state'])
        return res