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


class PurchaseRequest(models.Model):
    _inherit = 'purchase.request'
    _description = '采购申请'

     # 为state添加取消状态
    state = fields.Selection(
        selection_add=[('cancel', '已取消')],
        ondelete={'cancel': 'set default'}  # 添加 ondelete 策略
    )

    rule_new_add = fields.Boolean('采购请求为规则创建', default=False, compute='_compute_state', store=True)

    @api.depends('state')
    def _compute_state(self):
        for pr in self:
            if pr.state != 'draft' and pr.rule_new_add:
                pr.rule_new_add = False

    def action_view_purchase_order(self):
        action = super(PurchaseRequest, self).action_view_purchase_order()
        origin_context = ast.literal_eval(action['context'])
        if 'search_default_draft' in origin_context:
            origin_context.pop('search_default_draft')
        action['context'] = origin_context
        return action

class PurchaseRequestLine(models.Model):
    _inherit = 'purchase.request.line'
    _description = '采购申请明细'

    origin = fields.Char(string="Source Document")

    part_number = fields.Char('零件图号', store=True, compute='_compute_part_number')
    part_name = fields.Char('零件名称', store=True, compute='_compute_part_number')
    related_product = fields.Many2one('product.product', string='关联产品',
                                      help='经此产品工艺加工成的成品')

    supply_method = fields.Selection([
        ('automation', "自动化产线加工"),
        ('manual', "人工线下加工"),
        ('purchase', "外购"),
        ('outsourcing', "委外加工"),
    ], string='供货方式', compute='_compute_supply_method', store=True)

    @api.depends('origin')
    def _compute_supply_method(self):
        for prl in self:
            order_ids = []
            if not prl.origin:
                continue
            origin = [origin.replace(' ', '') for origin in prl.origin.split(',')]
            if 'S' in prl.origin:
                # 原单据是销售订单
                order_ids = self.env['sale.order'].sudo().search([('name', 'in', origin)]).ids
            elif 'MO' in prl.origin:
                # 原单据是制造订单
                mp_ids = self.env['mrp.production'].sudo().search([('name', 'in', origin)])
                order_ids = [mp_id.sale_order_id.id for mp_id in mp_ids] if mp_ids else []
            elif 'WH' in prl.origin:
                # 原单据是调拨单
                sp_ids = self.env['stock.picking'].sudo().search([('name', 'in', origin)])
                order_ids = [sp_id.sale_order_id.id for sp_id in sp_ids] if sp_ids else []
            order_line = self.env['sale.order.line'].sudo().search(
                [('product_id', '=', prl.product_id.id), ('order_id', 'in', order_ids)])
            if order_line:
                prl.supply_method = order_line[0].supply_method
            else:
                prl.supply_method = None

    @api.depends('product_id')
    def _compute_part_number(self):
        for record in self:
            if record.part_number and record.part_name:
                continue
            if record.product_id.categ_id.name == '坯料':
                product_name = ''
                match = re.search(r'(S\d{5}-\d)', record.product_id.name)
                # 如果匹配成功，提取结果
                if match:
                    product_name = match.group(0)
                sale_order_name = ''
                match_sale = re.search(r'S(\d+)', record.product_id.name)
                if match_sale:
                    sale_order_name = match_sale.group(0)
                sale_order = self.env['sale.order'].sudo().search(
                    [('name', '=', sale_order_name)])
                if sale_order:
                    filtered_order_line = sale_order.order_line.filtered(
                        lambda order_line: re.search(f'{product_name}$', order_line.product_id.name)
                    )
                    record.part_number = filtered_order_line.product_id.part_number
                    record.part_name = filtered_order_line.product_id.part_name
            else:
                record.part_number = record.product_id.part_number
                record.part_name = record.product_id.part_name