import re
from odoo import models, fields, api, _


class SFMessageStockPicking(models.Model):
    _name = 'stock.picking'
    _description = "库存调拨"
    _inherit = ['stock.picking', 'jikimo.message.dispatch']

    def button_validate(self):
        res = super(SFMessageStockPicking, self).button_validate()
        if self.location_id.name == '进货' and self.location_dest_id.name == '刀具房':
            self.add_queue('调拨入库')
        return res

    @api.depends('move_type', 'immediate_transfer', 'move_ids.state', 'move_ids.picking_id')
    def _compute_state(self):
        super(SFMessageStockPicking, self)._compute_state()
        if self.state == 'assigned' and self.check_in == 'PC':
            self.add_queue('坯料发料提醒')

    def _get_message(self, message_queue_ids):
        contents = []
        product_id = []
        for message_queue_id in message_queue_ids:
            if message_queue_id.message_template_id.name == '坯料发料提醒':
                content = message_queue_id.message_template_id.content
                stock_picking_line = self.env['stock.picking'].search([('id', '=', int(message_queue_id.res_id))])
                stock_picking_info = self.env['stock.picking'].search(
                    [('product_id', '=', stock_picking_line.product_id.id), ('state', '=', 'assigned'),
                     ('check_in', '=', 'PC')])
                if stock_picking_info and stock_picking_info.product_id.id not in product_id:
                    content = content.replace('{{product_id}}', stock_picking_info.product_id.name)
                    content = content.replace('{{number}}', str(len(stock_picking_info)))
                    product_id.append(stock_picking_info.product_id.id)
                    contents.append(content)
            else:
                res = super(SFMessageStockPicking, self)._get_message(message_queue_id)
                contents.append(res)
        return contents