107 lines
4.2 KiB
Python
107 lines
4.2 KiB
Python
from odoo import api, models
|
|
|
|
|
|
class ProjectProject(models.Model):
|
|
_inherit = 'project.project'
|
|
|
|
@api.model
|
|
def get_mindmap_tree(self, single_id=None):
|
|
"""返回 项目→阶段→任务 的树形结构（dict），供 mindmap 视图渲染。
|
|
|
|
不建中间表：直接利用已有的
|
|
project.project.type_ids（阶段）/ project.task.project_id / project.task.stage_id。
|
|
字段是否存在在服务端用 _fields 直接判断，杜绝 JS 侧写死字段导致的
|
|
Invalid field / 白屏。
|
|
"""
|
|
self = self.sudo()
|
|
|
|
# 1) 项目（单项目视角则只取当前项目）
|
|
proj_domain = [('id', '=', single_id)] if single_id else []
|
|
projects = self.search_read(proj_domain, ['name', 'type_ids'])
|
|
|
|
# 2) 阶段：经各项目 type_ids 收集、保序去重
|
|
stage_ids = []
|
|
for p in projects:
|
|
stage_ids += p['type_ids']
|
|
stage_ids = list(dict.fromkeys(stage_ids))
|
|
stages = self.env['project.task.type'].search_read(
|
|
[('id', 'in', stage_ids)], ['name', 'sequence']
|
|
) if stage_ids else []
|
|
stage_by_id = {s['id']: s for s in stages}
|
|
|
|
# 3) 任务：进度字段在部分版本/客制下不存在，服务端安全探测
|
|
task_fields = ['name', 'display_name', 'project_id', 'stage_id']
|
|
tk = self.env['project.task']
|
|
progress_field = None
|
|
if 'subtask_completion_percentage' in tk._fields:
|
|
progress_field = 'subtask_completion_percentage'
|
|
elif 'progress' in tk._fields:
|
|
progress_field = 'progress'
|
|
if progress_field:
|
|
task_fields.append(progress_field)
|
|
task_domain = [('project_id', '=', single_id)] if single_id else []
|
|
tasks = tk.search_read(task_domain, task_fields)
|
|
|
|
def _norm(v):
|
|
if v is None or v is False:
|
|
return 0
|
|
try:
|
|
v = float(v)
|
|
except (TypeError, ValueError):
|
|
return 0
|
|
# 部分字段存 0-1，有的存 0-100；统一归一到 0-100
|
|
return int(v) if v > 1 else int(v * 100)
|
|
|
|
project_nodes = {}
|
|
for p in projects:
|
|
project_nodes[p['id']] = {
|
|
'id': 'P%d' % p['id'],
|
|
'name': p['name'] or ('项目%s' % p['id']),
|
|
'kind': 'project',
|
|
'progress': 0,
|
|
'children': [],
|
|
}
|
|
|
|
for p in projects:
|
|
pn = project_nodes[p['id']]
|
|
proj_stages = [stage_by_id[i] for i in p['type_ids'] if i in stage_by_id]
|
|
proj_stages.sort(key=lambda s: s.get('sequence') or 0)
|
|
for st in proj_stages:
|
|
sn = {
|
|
'id': 'P%d_S%d' % (p['id'], st['id']),
|
|
'name': st['name'],
|
|
'kind': 'stage',
|
|
'progress': 0,
|
|
'children': [],
|
|
}
|
|
pn['children'].append(sn)
|
|
st_tasks = [
|
|
t for t in tasks
|
|
if t.get('project_id') and t['project_id'][0] == p['id']
|
|
and t.get('stage_id') and t['stage_id'][0] == st['id']
|
|
]
|
|
for t in st_tasks:
|
|
raw = t.get(progress_field) if progress_field else None
|
|
sn['children'].append({
|
|
'id': 'T%d' % t['id'],
|
|
'name': t.get('display_name') or t['name'],
|
|
'kind': 'task',
|
|
'resId': t['id'],
|
|
'progress': _norm(raw),
|
|
})
|
|
sn['progress'] = round(
|
|
sum(c['progress'] for c in sn['children']) / len(sn['children'])
|
|
) if sn['children'] else 0
|
|
pn['progress'] = round(
|
|
sum(c['progress'] for c in pn['children']) / len(pn['children'])
|
|
) if pn['children'] else 0
|
|
|
|
roots = list(project_nodes.values())
|
|
if single_id and len(roots) == 1:
|
|
return roots[0]
|
|
return {
|
|
'id': 0,
|
|
'name': '所有项目' if not single_id else '项目阶段与任务',
|
|
'children': roots,
|
|
}
|