from odoo import fields, models, api

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

    bom_id = fields.Many2one('mrp.bom', required=False)
    bom_ids = fields.Many2many('mrp.bom', string='Applies to BOMs', help='BOMs that use this operation.')
    company_id = fields.Many2one('res.company', 'Company', related=False, store=True, readonly=False, required=True, default=lambda self: self.env.company)

    @api.onchange('bom_id')
    def _onchange_bom_id(self):
        if self.bom_id and self.bom_id.company_id:
            self.company_id = self.bom_id.company_id

    @api.onchange('bom_ids')
    def _onchange_bom_ids(self):
        """
        If bom_ids contains a single BOM and bom_id is empty, set bom_id.
        This helps maintain compatibility and ease of use.
        """
        if len(self.bom_ids) == 1 and not self.bom_id:
            self.bom_id = self.bom_ids[0]