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

_logger = logging.getLogger(__name__)

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

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

    def action_post(self):
        """
        Confirmation bypass. Calls standard post with skip flag and FORCES a sync
        to ensure deductions are applied to the Journal Entry.
        """
        # 1. Force a synchronization of the moves right before posting.
        # We pass 'amount' and 'deduction_line_ids' to trigger vendor_payment_diff_amount.
        self_bypass = self.with_context(skip_expense_lock=True)
        for payment in self_bypass:
            if payment.state == 'draft':
                payment._synchronize_to_moves({'amount', 'deduction_line_ids', 'amount_substract'})
        
        # 2. Standard confirmation logic with bypass flag for internal writes
        return super(AccountPayment, self.with_context(skip_expense_lock=True)).action_post()

    def _synchronize_to_moves(self, changed_fields):
        # Force the refresh by ensuring 'amount' is in fields if deductions are involved
        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'}

        # SURGICAL JUMPER: If we have the bypass flag, we jump over the hr_expense method.
        if self._context.get('skip_expense_lock'):
            mro = type(self).mro()
            hr_expense_class = next((c for c in mro if c.__module__.startswith('odoo.addons.hr_expense')), None)
            if hr_expense_class:
                return super(hr_expense_class, self)._synchronize_to_moves(changed_fields)
        
        return super()._synchronize_to_moves(changed_fields)

    def _synchronize_from_moves(self, changed_fields):
        # Same jumper for synchronization back from moves
        if self._context.get('skip_expense_lock'):
            mro = type(self).mro()
            hr_expense_class = next((c for c in mro if c.__module__.startswith('odoo.addons.hr_expense')), None)
            if hr_expense_class:
                return super(hr_expense_class, self)._synchronize_from_moves(changed_fields)
        
        return super()._synchronize_from_moves(changed_fields)

    def write(self, vals):
        # Propagate bypass flag during writes to avoid locked checks
        if self.expense_sheet_id and self.state == 'draft':
            return super(AccountPayment, self.with_context(skip_expense_lock=True)).write(vals)
        return super().write(vals)

    def action_cancel(self):
        res = super().action_cancel()
        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

    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