python使用参数对嵌套字典进行取值

因一些特殊需求须要以参数的形式获取字典中特定的值，网上搜了一下并无特别好的实现(并无太认真去找~),因此本身实现了一个，以供你们参考:) 。

话很少说，直接上代码:

def dict_get(dic, locators, default=None):

        ''' :param dic: 输入须要在其中取值的原始字典 <dict> :param locators: 输入取值定位器, 如:['result', 'msg', '-1', 'status'] <list> :param default: 进行取值中报错时所返回的默认值 (default: None) :return: 返回根据参数locators找出的值 '''

        if not isinstance(dic, dict) or not isinstance(locators, list):
            return default

        value = None

        for locator in locators:
            if not type(value) in [dict, list] and isinstance(locator, str) and not can_convert_to_int(locator):
                try:
                    value = dic[locator]
                except KeyError:
                    return default
                continue
            if isinstance(value, dict):
                try:
                    value = dict_get(value, [locator])
                except KeyError:
                    return default
                continue
            if isinstance(value, list) and can_convert_to_int(locator):
                try:
                    value = value[int(locator)]
                except IndexError:
                    return default
                continue

        return value

    def can_convert_to_int(input):
        try:
            int(input)
            return True
        except BaseException:
            return False
复制代码

Best Practice

好的咱们来进行一次简单的最佳实践:)

if __name__ == '__main__':
    dict_test = {"result": {"code": "110002", "msg": [{'status': 'ok'}, {'status': 'failed'}]}}
    result = dict_get(dict_test, ['result', 'msg', '-1', 'status'])
    print(result)
复制代码

下面是控制台的输出，你们能够看到输出是符合预期结果的:)

failed

Process finished with exit code 0


复制代码

欢迎你们扫码关注个人公众号「智能自动化测试」，回复：测试进阶教程，便可免费得到 进阶教程 ~