Files
manufacture/mrp_workcenter_hierarchical/models/mrp_workcenter.py
2021-01-19 13:04:54 +01:00

62 lines
2.0 KiB
Python

# 2016 Akretion (http://www.akretion.com)
# David BEAL <david.beal@akretion.com>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
from odoo import api, fields, models
class MrpWorkcenter(models.Model):
_inherit = "mrp.workcenter"
_parent_name = "parent_id"
_parent_store = True
parent_id = fields.Many2one(
comodel_name="mrp.workcenter", string="Parent", index=True
)
parent_path = fields.Char(index=True)
child_ids = fields.One2many(
comodel_name="mrp.workcenter", inverse_name="parent_id", string="Children"
)
parent_level_1_id = fields.Many2one(
comodel_name="mrp.workcenter",
compute="_compute_parent_level",
string="Parent Level 1",
store=True,
)
parent_level_2_id = fields.Many2one(
comodel_name="mrp.workcenter",
compute="_compute_parent_level",
string="Parent Level 2",
store=True,
)
parent_level_3_id = fields.Many2one(
comodel_name="mrp.workcenter",
compute="_compute_parent_level",
string="Parent Level 3",
store=True,
)
def _get_parent_ids(self):
self.ensure_one()
if self.parent_id:
ids = self.parent_id._get_parent_ids()
ids.append(self.parent_id.id)
else:
ids = []
return ids
@api.depends("parent_id.parent_id.parent_id", "child_ids")
def _compute_parent_level(self):
def get_next_level(parent_ids, workcenter):
return (
parent_ids
and parent_ids.pop(0)
or (workcenter.child_ids and workcenter.id or workcenter.parent_id.id)
)
for workcenter in self:
parent_ids = workcenter._get_parent_ids()
workcenter.parent_level_1_id = get_next_level(parent_ids, workcenter)
workcenter.parent_level_2_id = get_next_level(parent_ids, workcenter)
workcenter.parent_level_3_id = get_next_level(parent_ids, workcenter)