Как в python возвратить путь к объекту в json и потом его использовать?

Question

[Сергей Бурдужа]

Всем привет.
Пишу свой скрипт для работы с json.

def addGroup(file_path, group_id = None):
    new_group_name = input("Enter group name: ")
    if new_group_name != "":
        group_slug = new_group_name.lower().replace(" ", "_")
        if sectionHasGroup(file_path, group_slug):
            print(colored("Group already exists!!!", "red"))
            return
        with open(file_path, 'r') as file:
            # read
            data = json.load(file)
            if group_id:
                group_path = getGroupPathById(file_path, group_id)
                new_tab = newTab(new_group_name)
                group_path.append(new_tab)
                new_group = newGroup(new_group_name)
                group_path.append(new_group)
                newData = json.dumps(data, indent=4)
                print(json.dumps(group_path, indent=4))
            else:
                new_tab = newTab(new_group_name)
                data[0]['fields'].append(new_tab)
                new_group = newGroup(new_group_name)
                data[0]['fields'].append(new_group)
                newData = json.dumps(data, indent=4)
                print(json.dumps(data[0]['fields'], indent=4))
            print(newData)

Если в addGroup передаю group_id, то нужно в json в нужном месте вставить код.

group_path = getGroupPathById(file_path, group_id)

И функция, которая возварщает путь

import json


def getGroupPathById(file_path, group_id):
    full_path = '';
    f = open(file_path,)
    data = json.load(f)
    for i in data:
        for j in i['fields']:
            if j['type'] == "group":
                if j['key'] == group_id:
                    index = i['fields'].index(j)
                    full_path = f"data[0]['fields'][{index}]"
                else:
                    for k in j['sub_fields']:
                        if k['type'] == "group":
                            if k['key'] == group_id:
                                index = j['sub_fields'].index(k)
                                full_path = f"data[0]['fields'][{i['fields'].index(j)}]['sub_fields'][{index}]"
                            else:
                                for l in k['sub_fields']:
                                    if l['type'] == "group":
                                        if l['key'] == group_id:
                                            index = k['sub_fields'].index(l)
                                            full_path = f"data[0]['fields'][{i['fields'].index(j)}]['sub_fields'][{j['sub_fields'].index(k)}]['sub_fields'][{index}]"
                print(f"full_path: {full_path}")
                return full_path

В addGroup вот тут ошибка
group_path.append(new_tab)

Заранее благодарен.

Comments

[chemdev]

В addGroup вот тут ошибка

Какая?

Пишу свой скрипт для работы с json.

Сформулируй точную задачу и приведи пример json. Довольно сложно понять какой путь нужен. Типо путь вида "ключ1__ключ2__элемент1__ключ3__значение"?

[Сергей Бурдужа]

chemdev,
Да какая разница как json выглядит.
Нужно поменять это

print(f"full_path: {full_path}")
                return full_path

на это

group_path = getGroupPathById(file_path, group_id)
                new_tab = newTab(new_group_name)
                group_path.append(new_tab)
                new_group = newGroup(new_group_name)
                group_path.append(new_group)

[chemdev]

Сергей Бурдужа, я ничего не понял, может другие поймут - помогут

[Илья]

По описанию задача простая, а по коду какой то ад.
По моему Вы перемудрили.
П.С. У вас написано "Я люблю phр...", часто в php используете функцию eval?

[Сергей Бурдужа]

Илья,
Добрый день.
Не часто.
Даже не знал, что в php есть eval.
Может поделитесь опытом, и покажете как правильно решать задачу?
А я хорошо запомню этот способ.

[Илья]

Сергей Бурдужа, если скинете пример json, может и смогу помочь

[Сергей Бурдужа]

Илья,
Тут и весть проект

github

[Илья]

Сергей Бурдужа,

def getGroupPathById(file_path, group_id):

    def addElement(path, value):
        path.append(value)
        return path

    def checkField(data):
        if data['type'] == "group" and data['key'] == group_id:
            return 3

        elif data['type'] == "group" and data['key'] != group_id:
            return 2

        elif data['type'] != "group":
            return 1

        else:
            return 0

    def findField(data, path=[]):
        if data.get('fields'):
            fields = data.get('fields')
            addElement(path, 'fields')
        elif data.get('sub_fields'):
            fields = data.get('sub_fields')
            addElement(path, 'sub_fields')
        else:
            return None

        for field_index, field in enumerate(fields):
            if checkField(field) == 3:
                return addElement(path, field_index)
            elif checkField(field) == 2:
                result = findField(field, addElement(path.copy(), field_index))
                if result is not None:
                    return result

    with open(file_path, 'r') as fp:
        data = json.load(fp)

    for branch_index, branch in enumerate(data):
        full_path = None
        result = findField(branch)
        if result is not None:
            full_path = addElement([], branch_index) + result
            break

    return full_path

json_path = 'json/page-home.json'
path = getGroupPathById(json_path, 'field_65a78d2d74bcf')

with open(json_path, 'r') as fp:
    data = json.load(fp)

for item in path:
    data = data[item]

print(path)
print(data)

[Сергей Бурдужа]

Илья, спасибо

Answer 1

[Сергей Бурдужа]

Нашел такое решение, не знаю насколько правильное, но работает.

if group_id:
                group_path = getGroupPathById(file_path, group_id)
                new_tab = newTab(new_group_name)
                new_group = newGroup(new_group_name)
                exec(f"{group_path}['sub_fields'].append({new_tab})")
                exec(f"{group_path}['sub_fields'].append({new_group})")
                newData = json.dumps(data, indent=4)

Comments

[Сергей Горностаев]

exec вместе с интерполяцией строк - верный способ заиметь проблем.