Files
manufacture/mrp_workcenter_hierarchical/models/mrp_workcenter.py

78 lines
2.5 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.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()
exclude_ids = [workcenter.id, workcenter.parent_id.id]
l_id = [False, False, False]
l_id[0] = get_next_level(parent_ids, workcenter)
workcenter.parent_level_1_id = (
l_id[0] if l_id[0] not in exclude_ids else False
)
exclude_ids.append(l_id[0])
l_id[1] = get_next_level(parent_ids, workcenter)
workcenter.parent_level_2_id = (
l_id[1] if l_id[1] not in exclude_ids else False
)
exclude_ids.append(l_id[1])
l_id[2] = get_next_level(parent_ids, workcenter)
workcenter.parent_level_3_id = (
l_id[2] if l_id[2] not in exclude_ids else False
)