# -*- coding: utf-8 -*-
from odoo import models, api


class IrModelPatch(models.Model):
    _inherit = 'ir.model'

    def _reflect_model_params(self, model):
        vals = super(IrModelPatch, self)._reflect_model_params(model)
        # 让 stock.picking 在工作流配置界面可被搜索到
        if model._name == 'stock.picking':
            vals['is_workflow'] = True
        return vals


class WorkflowRecordPatch(models.Model):
    _inherit = 'soong.workflow.record'

    @api.model
    def fix_borrow_workflow_id(self):
        """修复借用领用单 workflow record 缺少 workflow_id 的旧数据"""
        records = self.search([
            ('res_model', '=', 'stock.picking'),
            ('workflow_id', '=', False),
        ])
        if records:
            records.write({'workflow_id': 239})