from odoo import fields, models, api, _


class MrpProduction(models.Model):
    _inherit = 'mrp.production'

    pr_mp_count = fields.Integer('采购申请单数量', compute='_compute_pr_mp_count', store=True)

    @api.depends('state')
    def _compute_pr_mp_count(self):
        for item in self:
            if item.product_id.is_customer_provided:
                item.pr_mp_count = 0
            else:
                pr_ids = item._get_purchase_request()
                item.pr_mp_count = len(pr_ids)
            # pr_ids = self.env['purchase.request'].sudo().search([('origin', 'like', item.name), ('is_subcontract', '!=', 'True')])

    def action_view_pr_mp(self):
        """
        采购请求
        """
        self.ensure_one()

        # 由于采购申请合并了所有销售订单行的采购，所以不区分产品
        pr_ids = self._get_purchase_request()

        action = {
            'res_model': 'purchase.request',
            'type': 'ir.actions.act_window',
        }
        if len(pr_ids) == 1:
            action.update({
                'view_mode': 'form',
                'res_id': pr_ids[0].id,
            })
        else:
            action.update({
                'name': _("从 %s生成采购请求单", self.name),
                'domain': [('id', 'in', pr_ids.ids)],
                'view_mode': 'tree,form',
            })
        return action

    def _get_purchase_request(self):
        """获取跟制造订单相关的采购申请单（根据采购申请单行项目的产品匹配）"""
        mrp_names = self.env['mrp.production'].search([('origin', '=', self.origin)]).mapped('name')
        pr_ids = self.env['purchase.request'].sudo().search([('origin', 'in', mrp_names)])
        product_list = self.product_id._get_product_include_bom()
        pr_line_ids = pr_ids.line_ids.filtered(lambda l: l.product_id in product_list)
        return pr_line_ids.mapped('request_id')