Python 关于递归找出依赖用例，请教一下各位大佬

底层贫困人员 · 2021年06月16日 · 最后由 jinglebell 回复于 2021年06月17日 · 2358 次阅读

下面这段 json 为用例数据，譬如我要执行用例 1，我要先执行用例 2，执行用例 2 前提下，要先执行用例 3 4 5，执行用例 5 就先执行 6 ，总体顺序就是先执行依赖用例列表 [3，4，6，5，2 ]，最后在执行用例 1。

问题：如何实现输出依赖用例列表？如上面例子

[
    {
        "api_id": 1,
        "name": "用例1",
        "step": [
            2
        ]
    },
{
        "api_id": 2,
        "name": "用例2",
        "step": [
            3,
            4,
            5
        ]
    },
{
        "api_id": 3,
        "name": "用例3",
        "step": null
    },
{
        "api_id": 4,
        "name": "用例4",
        "step": null
    },{
        "api_id": 5,
        "name": "用例5",
        "step":[6]
    },
{
        "api_id": 6,
        "name": "用例6",
        "step": null
    }
]

目前代码实现，不确定是否有隐藏 bug

class Cases(object):

    @classmethod
    def get_steps(cls, case_data, steps, steps_list):
        steps = list(reversed(steps))
        for i in steps:
            for index, case in enumerate(case_data):
                if case['api_id'] == i:
                    steps_list.insert(0, case_data[index])
                    if case['step'] is not None:
                        cls.get_steps(case_data, case['step'], steps_list)
        return steps_list

steps = [2]
steps_data = Cases.get_steps(data,steps,[])
print(json.dumps(steps_data,ensure_ascii=False))

共收到 9 条回复时间点赞

陈恒捷 #1 · 2021年06月17日

1、是不是需要做下循环依赖的校验？比如 1 依赖 2，2 依赖 3，3 依赖 1 这种。
2、这里最深的 if 已经是第四层了，嵌套略深，而且也缺少对 step id 没有对应 api_id 这种异常情况的检测，目前只是直接忽略返回空列表，而非抛异常。
3、有一个场景没有描述，比如 1 依赖 2、3，2 和 3 都依赖 4。此时应该生成的是 [4, 2, 3, 1] ，还是 [4, 2, 4, 3, 1] ？按照目前实现，会出现的应该是后者。

建议按照这几个信息，分块编写相关逻辑：