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


class ResWorkcenter(models.Model):
    _inherit = 'mrp.workcenter'

    machine_tool_id = fields.Many2one('mrs.machine_tool', '机床')


class ResWorkcenterProductivity(models.Model):
    _inherit = 'mrp.workcenter.productivity'

    workcenter_id = fields.Many2one('mrp.workcenter',  required=False)


class ResMrpWorkOrder(models.Model):
    _inherit = 'mrp.workorder'

    workcenter_id = fields.Many2one('mrp.workcenter', required=False)
    processing_panel = fields.Char('加工面')
    routing_type = fields.Selection([
        ('获取CNC加工程序', '获取CNC加工程序'),
        ('装夹', '装夹'),
        ('前置三元定位检测', '前置三元定位检测'),
        ('CNC加工', 'CNC加工'),
        ('后置三元质量检测', '后置三元质量检测'),
        ('解除装夹', '解除装夹'),
    ], string="工序类型")

    def json_workorder_str(self, k, production, route):
        workorders_values_str = [0, '', {
            'product_uom_id': production.product_uom_id.id,
            'qty_producing': 0,
            'operation_id': False,
            'name': route.route_workcenter_id.name,
            'processing_panel': k,
            'routing_type': route.routing_type,
            'workcenter_id': self.env['mrp.routing.workcenter'].get_workcenter(route.workcenter_ids.ids),
            'date_planned_start': False,
            'date_planned_finished': False,
            'duration_expected': 60,
            'duration': 0
        }]
        return workorders_values_str