Iteruj po atrybutach obiektów w Pythonie

Mam obiekt Pythona z kilkoma atrybutami i metodami. Chcę iterować po atrybutach obiektów.

class my_python_obj(object):
    attr1='a'
    attr2='b'
    attr3='c'

    def method1(self, etc, etc):
        #Statements

Chcę wygenerować słownik zawierający wszystkie atrybuty obiektów i ich bieżące wartości, ale chcę to zrobić w sposób dynamiczny (więc jeśli później dodam kolejny atrybut, nie muszę również pamiętać o aktualizacji mojej funkcji).

W php zmienne mogą być używane jako klucze, ale obiekty w Pythonie nie są obsługiwane i jeśli używam do tego notacji z kropką, tworzy nowy atrybut o nazwie moja var, co nie jest moim zamiarem.

Żeby było jaśniej:

def to_dict(self):
    '''this is what I already have'''
    d={}
    d["attr1"]= self.attr1
    d["attr2"]= self.attr2
    d["attr3"]= self.attr3
    return d

·

def to_dict(self):
    '''this is what I want to do'''
    d={}
    for v in my_python_obj.attributes:
        d[v] = self.v
    return d

Aktualizacja: Przez atrybuty mam na myśli tylko zmienne tego obiektu, a nie metody.

Zakładając, że masz klasę taką jak

>>> class Cls(object):
...     foo = 1
...     bar = 'hello'
...     def func(self):
...         return 'call me'
...
>>> obj = Cls()

wywołanie dirobiektu zwraca wszystkie atrybuty tego obiektu, w tym specjalne atrybuty Pythona. Chociaż niektóre atrybuty obiektów są wywoływalne, takie jak metody.

>>> dir(obj)
['__class__', '__delattr__', '__dict__', '__doc__', '__format__', '__getattribute__', '__hash__', '__init__', '__module__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__sizeof__', '__str__', '__subclasshook__', '__weakref__', 'bar', 'foo', 'func']

Możesz zawsze odfiltrować specjalne metody, używając rozumienia listy.

>>> [a for a in dir(obj) if not a.startswith('__')]
['bar', 'foo', 'func']

lub jeśli wolisz mapy / filtry.

>>> filter(lambda a: not a.startswith('__'), dir(obj))
['bar', 'foo', 'func']

Jeśli chcesz odfiltrować metody, możesz użyć wbudowanego callablejako sprawdzenia.

>>> [a for a in dir(obj) if not a.startswith('__') and not callable(getattr(obj, a))]
['bar', 'foo']

Możesz również sprawdzić różnicę między twoją klasą a jej obiektem instancji przy użyciu.

>>> set(dir(Cls)) - set(dir(object))
set(['__module__', 'bar', 'func', '__dict__', 'foo', '__weakref__'])

generalnie umieść __iter__metodę w swojej klasie i iteruj przez atrybuty obiektu lub umieść tę klasę mixin w swojej klasie.

class IterMixin(object):
    def __iter__(self):
        for attr, value in self.__dict__.iteritems():
            yield attr, value

Twoja klasa:

>>> class YourClass(IterMixin): pass
...
>>> yc = YourClass()
>>> yc.one = range(15)
>>> yc.two = 'test'
>>> dict(yc)
{'one': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14], 'two': 'test'}

Jak już wspomniano w niektórych odpowiedziach / komentarzach, obiekty Pythona już przechowują słownik ich atrybutów (metody nie są uwzględnione). Można to uzyskać w postaci __dict__, ale lepszym sposobem jest użycie vars(dane wyjściowe są jednak takie same). Zwróć uwagę, że modyfikowanie tego słownika spowoduje zmodyfikowanie atrybutów instancji! Może to być przydatne, ale oznacza również, że powinieneś ostrożnie korzystać z tego słownika. Oto krótki przykład:

class A():
    def __init__(self, x=3, y=2, z=5):
        self.x = x
        self._y = y
        self.__z__ = z

    def f(self):
        pass

a = A()
print(vars(a))
# {'x': 3, '_y': 2, '__z__': 5}
# all of the attributes of `a` but no methods!

# note how the dictionary is always up-to-date
a.x = 10
print(vars(a))
# {'x': 10, '_y': 2, '__z__': 5}

# modifying the dictionary modifies the instance attribute
vars(a)["_y"] = 20
print(vars(a))
# {'x': 10, '_y': 20, '__z__': 5}

Używanie dir(a)jest dziwnym, jeśli nie wręcz złym podejściem do tego problemu. Dobrze, jeśli naprawdę potrzebujesz iteracji po wszystkich atrybutach i metodach klasy (w tym metodach specjalnych, takich jak __init__). Jednak wydaje się, że nie jest to to, czego chcesz, a nawet zaakceptowana odpowiedź idzie źle, stosując trochę kruchego filtrowania, aby spróbować usunąć metody i pozostawić tylko atrybuty; możesz zobaczyć, jak to się nie powiedzie dla klasy Azdefiniowanej powyżej.

(Używanie __dict__zostało zrobione w kilku odpowiedziach, ale wszystkie definiują niepotrzebne metody zamiast używać ich bezpośrednio. Jedynie komentarz sugeruje użycie vars).

Obiekty w Pythonie przechowują swoje atrybuty (w tym funkcje) w formie dict __dict__. Możesz (ale generalnie nie powinieneś) tego używać do bezpośredniego dostępu do atrybutów. Jeśli potrzebujesz tylko listy, możesz również wywołać dir(obj), co zwraca iterowalną ze wszystkimi nazwami atrybutów, do których możesz następnie przekazać getattr.

Jednak konieczność robienia czegokolwiek z nazwami zmiennych jest zwykle złym projektem. Dlaczego nie przechowywać ich w kolekcji?

class Foo(object):
    def __init__(self, **values):
        self.special_values = values

Następnie możesz iterować po kluczach za pomocą for key in obj.special_values:

class someclass:
        x=1
        y=2
        z=3
        def __init__(self):
           self.current_idx = 0
           self.items = ["x","y","z"]
        def next(self):
            if self.current_idx < len(self.items):
                self.current_idx += 1
                k = self.items[self.current_idx-1]
                return (k,getattr(self,k))
            else:
                raise StopIteration
        def __iter__(self):
           return self

następnie nazwij to iterowalnym

s=someclass()
for k,v in s:
    print k,"=",v

Prawidłowa odpowiedź brzmi: nie powinieneś. Jeśli chcesz tego typu rzeczy, użyj po prostu dyktu lub będziesz musiał jawnie dodać atrybuty do jakiegoś kontenera. Możesz to zautomatyzować, ucząc się o dekoratorach.

W szczególności, nawiasem mówiąc, metoda method1 w twoim przykładzie jest równie dobra jak atrybut.

W przypadku Pythona 3.6

class SomeClass:

    def attr_list(self, should_print=False):

        items = self.__dict__.items()
        if should_print:
            [print(f"attribute: {k}    value: {v}") for k, v in items]

        return items

Jestem pewien, że Johan Cleeze pochwaliłby twój dogmatyzm;). Zostawiam tę odpowiedź, wciąż ją obwiniaj To właściwie czyni mnie bardziej powiernym. Zostaw komentarz kurczaki!

W przypadku Pythona 3.6

class SomeClass:

    def attr_list1(self, should_print=False):

        for k in self.__dict__.keys():
            v = self.__dict__.__getitem__(k)
            if should_print:
                print(f"attr: {k}    value: {v}")

    def attr_list(self, should_print=False):

        b = [(k, v) for k, v in self.__dict__.items()]
        if should_print:
            [print(f"attr: {a[0]}    value: {a[1]}") for a in b]
        return b