Python查找不限层级Json数据中某个key或者value的路径

最新推荐文章于 2024-06-29 12:30:44 发布

原创最新推荐文章于 2024-06-29 12:30:44 发布 · 3.2k 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#python #json #无限层级

Python 专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一款用于查找JSON文件中特定key或value路径的Python脚本。该脚本能够递归地搜索JSON结构，并返回匹配项的完整路径。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近项目中有一个小需求，查找json文件中某个key或者value的路径，所以就写了一个简单的小脚本，比较粗糙。

#!/usr/bin/env python3
# -*- coding:utf-8 -*-
'''
@author: funcups
'''
from logzero import logger
import ast

class HandleJson():
    def __init__(self, data):
        if data == None:
            logger.error('请输入json格式数据')
            exit()

        if isinstance(data, str):
            try:
                self.data = ast.literal_eval(data)
            except:
                logger.error('请输入正确的json格式数据')
                exit()
        elif isinstance(data, dict):
            self.data = data

    def __paths(self, data, path=''):
        '''
        用于遍历json树
        :param data: 原始数据，或者key对应的value值
        :param path: key值字符串，默认值为''
        :return:
        '''
        if isinstance(data, dict):
            for k, v in data.items():
                tmp = path + "['%s']" % k
                yield (tmp, v)
                yield from self.__paths(v, tmp)

        if isinstance(data, list):
            for k, v in enumerate(data):
                tmp = path + '[%d]' % k
                yield (tmp, v)
                yield from self.__paths(v, tmp)

    def find_key_path(self, key):
        '''
        查找key路径
        :param key: 需要查找路径的key值
        :return: 包含key值路径的list
        '''
        result = []
        for path,value in self.__paths(self.data):
            if path.endswith("['%s']" % key):
                result.append(path)
        with open('path.txt', 'w+', encoding='utf-8') as f:
            list(map(lambda line: f.write(line + '\r'), result))
        return result

    def find_value_path(self, key):
        '''
        查找某个值的路径
        :param key: 需要查找的值，限制为字符串，数字，浮点数，布尔值
        :return:
        '''
        result = []
        for path, value in self.__paths(self.data):
            if isinstance(value, (str, int, bool, float)):
                if value == key:
                    result.append(path)
        with open('path.txt', 'w+', encoding='utf-8') as f:
            list(map(lambda line: f.write(line + '\r'), result))
        return result

if __name__ == '__main__':
    data = {'name': 'funcups'}
    hj = HandleJson(data)
    res = hj.find_key_path('name')
    print(res)

    res = hj.find_value_path('funcups')
    print(res)