45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
from odoo import models
|
|
|
|
|
|
class ProjectTask(models.Model):
|
|
_inherit = 'project.task'
|
|
|
|
def get_hierarchy(self, project_id=None):
|
|
"""返回项目下任务的树形结构（嵌套 children），供前端 D3 渲染。
|
|
|
|
:param project_id: 项目ID，为空则返回所有任务
|
|
:return: 含虚拟根的嵌套字典
|
|
"""
|
|
domain = []
|
|
if project_id:
|
|
domain.append(('project_id', '=', project_id))
|
|
tasks = self.search(domain)
|
|
|
|
by_id = {}
|
|
for t in tasks:
|
|
by_id[t.id] = {
|
|
'id': t.id,
|
|
'name': t.name or '未命名',
|
|
'parent_id': t.parent_id.id if t.parent_id else None,
|
|
'progress': round(t.progress or 0),
|
|
'user_id': t.user_id.name if t.user_id else False,
|
|
'stage_id': t.stage_id.name if t.stage_id else False,
|
|
'date_deadline': t.date_deadline.strftime('%Y-%m-%d') if t.date_deadline else False,
|
|
'children': [],
|
|
}
|
|
|
|
roots = []
|
|
for t in tasks:
|
|
node = by_id[t.id]
|
|
pid = node['parent_id']
|
|
if pid and pid in by_id:
|
|
by_id[pid]['children'].append(node)
|
|
else:
|
|
roots.append(node)
|
|
|
|
return {
|
|
'id': 0,
|
|
'name': '项目任务',
|
|
'children': roots,
|
|
}
|