Eu pensei que levaria algum tempo para mostrar como você pode traduzir um objeto para ditar via dict(obj)
.
class A(object):
d = '4'
e = '5'
f = '6'
def __init__(self):
self.a = '1'
self.b = '2'
self.c = '3'
def __iter__(self):
# first start by grabbing the Class items
iters = dict((x,y) for x,y in A.__dict__.items() if x[:2] != '__')
# then update the class items with the instance items
iters.update(self.__dict__)
# now 'yield' through the items
for x,y in iters.items():
yield x,y
a = A()
print(dict(a))
# prints "{'a': '1', 'c': '3', 'b': '2', 'e': '5', 'd': '4', 'f': '6'}"
A seção principal deste código é a __iter__
função
Como os comentários explicam, a primeira coisa que fazemos é pegar os itens de classe e impedir qualquer coisa que comece com '__'.
Depois de criar isso dict
, você poderá usar a update
função dict e passar na instância __dict__
.
Isso fornecerá um dicionário completo de classe + instância de membros. Agora, tudo o que resta é iterar sobre eles e gerar os retornos.
Além disso, se você planeja usar muito isso, pode criar um @iterable
decorador de classe.
def iterable(cls):
def iterfn(self):
iters = dict((x,y) for x,y in cls.__dict__.items() if x[:2] != '__')
iters.update(self.__dict__)
for x,y in iters.items():
yield x,y
cls.__iter__ = iterfn
return cls
@iterable
class B(object):
d = 'd'
e = 'e'
f = 'f'
def __init__(self):
self.a = 'a'
self.b = 'b'
self.c = 'c'
b = B()
print(dict(b))