自作クラスの要素をlistやtupleのようにunpackしたい

Question

自作クラスの要素を、下記コード内コメントアウトのようにunpackしたいです。
どうすればよいでしょうか?

class Edge:
    def __init__(self, to, cap, rev):
        self.to = to
        self.cap = cap
        self.rev = rev


def main():
    edges = [Edge(i, i, i) for i in range(5)]
    for edge in edges:
        print(edge.to, edge.cap, edge.rev)
    # for to, cap, rev in edges:  # こういうunpackがしたい
    #     print(to, cap, rev)


if __name__ == "__main__":
    main()

Answer 1

__iter__ メソッドを実装すると出来ます。

class Edge:
    def __init__(self, to, cap, rev):
        self.to = to
        self.cap = cap
        self.rev = rev
    
    def __iter__(self):
        yield from (self.to, self.cap, self.rev)

see also collections.namedtuple, dataclasses.astuple

Comments

[退会済みユーザー]

ありがとうございます。
何か上手いやり方があったはずなのになーと思っていたら、namedtupleでした