python - 收集完整路径直到组织树上的叶子

Question

我将组织树存储为 json

{
    "name": "amos",
    "direct_reports": [
        {
            "name": "bart",
            "direct_reports": [
                {
                    "name": "colin",
                    "direct_reports": []
                },
                {
                    "name": "clara",
                    "direct_reports": []
                }
            ]
        },
        {
            "name": "bravo",
            "direct_reports": [
                {
                    "name": "cupid",
                    "direct_reports": []
                },
                {
                    "name": "clever",
                    "direct_reports": []
                }
            ]
        }
    ]
}

我需要为每个员工存储完整的“管理路径”，例如： management_chain["聪明"]={bravo,amos} management_chain["bart"]={amos}

目前我设法到达所有边缘并将它们分类为员工和经理，代码如下：

def get_herarchy(org):
    tmp_obj = {}
    tmp_obj['managers'] = []
    for emp in org['direct_reports']:
        tmp_obj['managers'].append(org['name'])
        print("manager "+org['name'])
        if len(emp['direct_reports'])>0:
            get_herarchy(emp)
        tmp_obj['name'] = emp['name']
        print(emp['name'])
    return tmp_obj

但是字典没有正确的值

Answer 1

像这样，也许：

def get_chain(org, name):
    if org['name'] == name:
        return [name]
    for emp in org['direct_reports']:
        chain = get_chain(emp, name)
        if chain:
            return [org['name']] + chain
    return None

print(get_chain(org, 'bart'))    # ['amos', 'bart']
print(get_chain(org, 'clever'))  # ['amos', 'bravo', 'clever']