python实现树结构的json文件_目录树转 JSON

文件夹目录树转json集合

背景：为了处理2处不同来源的数据，就都转换成json集合，方便操作。

目录树结构

----demo\

|----v1\

| |----init\

| | |----time.py

| |----order.py

| |----surface\

| | |----detail.py

| | |----order.py

| | |----user.py

| |----user.py

|----v2\

| |----country.py

| |----list.py

|----v3\

| |----languages.py

| |----surface\

| | |----detail.py

| | |----list.py

| |----user.py

以上是目录结构 首先进行读取，放到一个列表里，在读取的时候，还进行了正则匹配，匹配对应的py文件，以及文件内部的 方法。并且内部匹配到的每一条 方法生成一条记录

import sys

from pathlib import Path

import json

import re

tree_str = ''

tree_list = []

def generate_tree(pathname, n=0):

global tree_str

global tree_list

if pathname.is_file():

temp_file_name = pathname.name

temp_file_name_list = temp_file_name.split('.')

if len(temp_file_name_list) == 2 and temp_file_name_list[1] == 'py':

file = open(pathname, 'r', encoding='utf-8')

temp_action = []

for line in file.readlines():

line = line.strip()

strName = getStrName(pathname.parent, temp_file_name_list[0])

matchObj = re.match('def getAction()', line)

if matchObj:

temp_action.append('get')

matchObj = re.match('def postAction()', line)

if matchObj:

temp_action.append('post')

matchObj = re.match('def putAction()', line)

if matchObj:

temp_action.append('put')

matchObj = re.match('def deleteAction()', line)

if matchObj:

temp_action.append('delete')

if len(temp_action) > 0:

temp_strName = strName + ',' + '_'.join(temp_action)

tree_list.append(temp_strName.lower())

elif pathname.is_dir():

for cp in pathname.iterdir():

generate_tree(cp, n + 1)

def getStrName(pathname, strName):

if pathname.name != 'demo':

strName = pathname.name + '/' + strName

strName = getStrName(pathname.parent,strName)

return strName

if __name__ == '__main__':

p = Path(r'E:\www\myleetcode\demo')

generate_tree(p)

tree_list.sort()

print(tree_list)

执行结果

['v1/init/time,get', 'v1/order,get', 'v1/surface/detail,post_get', 'v1/surface/order,post', 'v1/surface/user,post', 'v1/user,get', 'v2/country,put', 'v2/list,put', 'v3/languages,get', 'v3/surface/detail,delete', 'v3/surface/list,delete', 'v3/user,get']

以上生成了初步的记录，接下来就是转换成想要的json格式

转json

塞到字典里

temp = {}

for api in tree_list:

temp_split = api.split('/')

if len(temp_split) == 2:

temp_str = temp_split[1].split(',')

temp_api = {temp_str[0] : temp_str[1]}

if temp_split[0] not in temp:

temp[temp_split[0]] = temp_api

else:

temp[temp_split[0]].update(temp_api)

elif len(temp_split) == 3:

temp_str = temp_split[2].split(',')

temp_api = {temp_str[0] : temp_str[1]}

if temp_split[0] not in temp:

temp_surface = {temp_split[1] : temp_api}

temp[temp_split[0]] = temp_surface

else:

temp_surface = temp[temp_split[0]]

if temp_split[1] not in temp_surface:

temp_surface.update({temp_split[1]: temp_api})

else:

temp_api_list = temp_surface[temp_split[1]]

temp_api_list.update(temp_api)

temp_surface[temp_split[1]] = temp_api_list

temp[temp_split[0]] = temp_surface

str_json = json.dumps(temp)

print (str_json)

json结果

{"v1": {"init": {"time": "get"}, "order": "get", "surface": {"detail": "post_get", "order": "post", "user": "post"}, "user": "get"}, "v2": {"country": "put", "list": "put"}, "v3": {"languages": "get", "surface": {"detail": "delete", "list": "delete"}, "user": "get"}}

这样做的目的是方便下一步画出树，结构图能够清晰的比较差异。

本作品采用《CC 协议》，转载必须注明作者和本文链接