# -*- coding: utf-8 -*-
# Part of YiZuo. See LICENSE file for full copyright and licensing details.
from odoo.exceptions import UserError, ValidationError
from datetime import datetime
from odoo import models, api, fields


class WorkpieceDeliveryWizard(models.TransientModel):
    _name = 'sf.workpiece.delivery.wizard'
    _description = '工件配送'

    delivery_ids = fields.Many2many('sf.workpiece.delivery', string='配送')
    workorder_id = fields.Many2one('mrp.workorder', string='工单')

    def confirm(self):
        if self.workorder_id:
            self.workorder_id.workpiece_delivery_ids[0]._delivery_avg()
        else:
            self.delivery_ids._delivery_avg()