from odoo import models


class StockWarehouseOrderpoint(models.Model):
    _inherit = "stock.warehouse.orderpoint"

    def _get_orderpoint_action(self):
        return super(StockWarehouseOrderpoint, self.with_context(ignore_rental_returns=True))._get_orderpoint_action()

    def _get_product_context(self, visibility_days=0):
        context = super()._get_product_context(visibility_days)
        context['ignore_rental_returns'] = True
        return context

    def _quantity_in_progress(self):
        res = super()._quantity_in_progress()
        rental_loc_ids = self.env.companies.rental_loc_id.ids
        for orderpoint in self:
            if not orderpoint.product_id.rent_ok:
                continue
            domain = [
            ('product_id', '=', orderpoint.product_id.id),
            ('state', 'in', ['confirmed', 'assigned', 'waiting', 'partially_available']),
            ('date', '<=', orderpoint.lead_days_date),
            '|',
                ('location_id', 'in', rental_loc_ids),
                ('location_dest_id', 'in', rental_loc_ids),
            '|',
                ('location_id.warehouse_id', '=', orderpoint.warehouse_id.id),
                ('location_dest_id.warehouse_id', '=', orderpoint.warehouse_id.id)
            ]
            rental_moves = self.env['stock.move'].search(domain, order="date, id")
            outs_qty, current_qty, max_missing_qty = 0, 0, 0
            for move in rental_moves:
                if move.location_id.id in rental_loc_ids:
                    current_qty += move.product_qty
                else:
                    outs_qty += move.product_qty
                    current_qty -= move.product_qty
                    if (current_qty < max_missing_qty):
                        max_missing_qty = current_qty
            res[orderpoint.id] += outs_qty + max_missing_qty
        return res