Как переработать код алгоритма поиска объекта?

Question

[Артем Кондрацкий]

Написал такой алгоритм поиска, сначало, ищется подходящая папка, исходя из идентификатора получаемого объекта (obj), затем идет поиск по названию подходящей папки среди локального хранилища. Схема рабочая, но больше похоже на говно-код, может быть можно как-то ее более изящно и правильно переписать? Было бы славно объединить это в одно "действие"

FOLDERS = {
    "Geometry": {
        "color": #...,
        "icon":  #...,
        "ids": [
            5163, 
            5167, 
            5173,
            5168,
         ]
    }
}

folder_name = next(iter(filter(
        lambda name: obj.GetType() in FOLDERS[name]['ids'],
FOLDERS)), None)

folder = next(iter(filter(
       lambda folder: folder.GetName() == folder_name,
created_folders)), None)

created_folder - массив с хранящимися объектами (папками), объекты не обычные, а особый класс, который отдает API

функция GetType отдает int, это идентификатор объекта, такой на подобии как и в массиве
функция GetName отдает string, это именя объекта

Comments

[Vladislav]

Я бы переписал folder_name и folder как отдельные функции.

[Артем Кондрацкий]

а толку? можно конечно написать отдельную функцию (условно search) и передавать в нее переменные, тем самым немного сократить код, но не является нужным решением

[o5a]

Артем Кондрацкий, "объединить в одно" при такой структуре данных формально можно, но это просто получится более длинная строка, с худшей читаемостью. Ведь в любом случае приходится перебирать значения в одном словаре (FOLDERS), а затем в другом.
Сделать удобнее можно, если преобразовать этот словарь FOLDERS в нужный вид, где каждому id будет соответствовать название, т.е.

folder_names = {5163: 'Geometry', 5167: Geometry, ...}

тогда последующий поиск можно делать напрямую

folder = next(iter(filter(
       lambda folder: folder.GetName() == folder_names[obj.GetType()],
created_folders)), None)

Answer 1

[Vaskey]

Можно упростить ваш алгоритм поиска объекта, объединив два шага (поиск папки по идентификатору объекта и поиск папки по имени) в один. Вместо того, чтобы хранить идентификаторы папок в словаре, можно использовать имена папок как ключи и хранить идентификаторы объектов в списке значений. Тогда вы можете использовать имя папки, чтобы найти соответствующий идентификатор объекта, а затем использовать это значение для поиска папки в массиве created_folders. Вот пример кода:

FOLDERS = {
    "Geometry": {
        "color": #...,
        "icon":  #...,
        "ids": [
            5163, 
            5167, 
            5173,
            5168,
         ]
    }
}

obj_type = obj.GetType()
folder_name = next((name for name, folder in FOLDERS.items() if obj_type in folder['ids']), None)
folder = next((folder for folder in created_folders if folder.GetType() == folder_name), None)

В этом примере мы сначала определяем идентификатор объекта, используя метод GetType(). Затем мы находим имя папки, которая соответствует этому идентификатору, с помощью генератора списков и функции next(). Затем мы находим папку, которая соответствует этому имени, с помощью того же подхода.

Этот подход более простой и легко читаемый, чем использование словаря FOLDERS с идентификаторами папок в качестве значений. Также он более гибкий, потому что вы можете легко добавлять новые папки и объекты в вашу программу, просто добавляя их имена и идентификаторы в словарь FOLDERS.

Comments

[Артем Кондрацкий]

Спасибо, так гораздо приятнее смотрится

[hint000]

ChatGPT detected

[Артем Кондрацкий]

hint000, хах, я тоже об этом подумал