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


複製代碼

歡迎你們掃碼關注個人公衆號「智能自動化測試」，回覆：測試進階教程，便可免費得到 進階教程 ~