import re
from odoo import models, fields, api, _
from urllib.parse import urlencode


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

    @api.model_create_multi
    def create(self, vals):
        result = super(SFMessageStockPicking, self).create(vals)
        for obj in result:
            if obj.location_id.name == '进货' and obj.location_dest_id.name == '刀具房':
                obj.add_queue('调拨入库')
        return result

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

    def _get_message(self, message_queue_ids):
        contents = []
        product_id = []
        for message_queue_id in message_queue_ids:
            i = 0
            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))])
                mrp_production_info = self.env['mrp.production'].search(
                    [('name', '=', stock_picking_line.origin)])
                mrp_production_list = self.env['mrp.production'].search(
                    [('product_id', '=', mrp_production_info.product_id.id)])
                for mrp_production_line in mrp_production_list:
                    picking_ids = mrp_production_line.picking_ids
                    for picking_id in picking_ids:
                        if picking_id.state == 'assigned' and picking_id.check_in == 'PC':
                            i += 1
                if i > 0 and mrp_production_info.product_id.id not in product_id:
                    url = self.request_url()
                    content = content.replace('{{product_id}}', mrp_production_info.product_id.name).replace(
                        '{{number}}', str(i)).replace('{{request_url}}', url)
                    product_id.append(mrp_production_info.product_id.id)
                    contents.append(content)
                return contents
            else:
                res = super(SFMessageStockPicking, self)._get_message(message_queue_id)
                return res

    def request_url(self):
        url = self.env['ir.config_parameter'].get_param('web.base.url')
        action_id = self.env.ref('stock.stock_picking_type_action').id
        menu_id = self.env['ir.model.data'].search([('name', '=', 'module_theme_treehouse')]).id
        # 查询参数
        params = {'menu_id': menu_id, 'action': action_id, 'model': 'stock.picking',
                  'view_type': 'kanban'}
        # 拼接查询参数
        query_string = urlencode(params)
        # 拼接URL
        full_url = url + "/web#" + query_string
        return full_url