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:
            pr_ids = self.env['purchase.request'].sudo().search([('origin', 'like', item.name)])
            if pr_ids:
                item.pr_mp_count = len(pr_ids)
            else:
                item.pr_mp_count = 0

    def action_view_pr_mp(self):
        """
        采购请求
        """
        self.ensure_one()
        pr_ids = self.env['purchase.request'].sudo().search([('origin', 'like', self.name),('is_subcontract', '!=', True)])
        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)],
                'view_mode': 'tree,form',
            })
        return action