Selectively dump object attributes with PyYAML

Обновить

April 2019

Просмотры

971 раз

1

Я могу использовать YAML сваливать иерархию объектов питона, так:

import yaml

class C():
    def __init__(self, x, y):
        self.x = x
        self.y = y

class D():
    def __init__(self, c, d):
        self.c = c
        self.d = d

d = D(c=C(x=1, y='hi'),
      d='bye')

print yaml.dump(d)

производит вывод:

!!python/object:__main__.D
c: !!python/object:__main__.C {x: 1, y: hi}
d: bye

Но я хочу , чтобы выборочно скрыть некоторые атрибуты. Поэтому предположим , у меня есть функция , attribs_to_dump(obj)которая для любого объекта возвращает список имен атрибутов , которые я хочу сбросить, например:

def attribs_to_dump(obj):
    if obj.__class__ == C: return ['x']
    if obj.__class__ == D: return ['c']

Мой вопрос, как я могу подключить attribs_to_dumpк yaml.dumpтак , что я получаю следующий вывод?

!!python/object:__main__.D
c: !!python/object:__main__.C {x: 1}

Там это осложняющий фактор: Я хочу , чтобы достичь эффекта, зацепив в YAML , как он ползет по иерархии объектов , а не путем предварительной обработки иерархии объектов себе. Причина заключается в том, что не все объекты в иерархии легко поддается самоанализ из - за setattr/getattr/__dict__магии , которая присутствует в некоторых библиотеках я использую: - (...

Вся помощь ценится!

1 ответы

1

This is an interesting question, I enjoyed solving it, thanks :)

from copy import deepcopy


class C(object):
    def __init__(self, x, y):
        self.x = x
        self.y = y

class D(object):
    def __init__(self, c, d):
        self.c = c
        self.d = d

d = D(
    c=C(x=1, y='hi'),
    d='bye'
)


FILTER_PARAMS = (
    #(class_reference, process_recursively, ['attributes', 'on', 'whitelist'])
    (D, True, ['c']),
    (C, False, ['x']),
)

def attr_filter(obj, filter_params):
    for attr in dir(obj):
        if attr.startswith('__'):
            # ignore builtins
            continue

        attr_val = obj.__getattribute__(attr)
        # loop through filter params
        for (cls, do_recursive, whitelist) in filter_params:
            if isinstance(obj, cls) and attr in whitelist:
                # filter class matches the current obj's class and
                # current attribute is on the whitelist
                if do_recursive:
                    # must process this attribute the same way as the parent
                    setattr(obj, attr, attr_filter(attr_val, filter_params))                
                # break will avoid the execution of the else clause meaning
                # the attribute was on the white list so don't delete it
                break
        else:
            # delete the current attribute of the instance as it was
            # not on the whitelist
            delattr(obj, attr)

    return obj


# do a deepcopy of the object you pass in, so the original will be preserved
filtered_instance = attr_filter(deepcopy(d), FILTER_PARAMS)
print dir(filtered_instance)
print dir(filtered_instance.c)

# now pass filtered_instance to yaml's dump function