Pythonではイテレータを簡単に作れます。もしかしたら何かに使えるかもしれないので、素振りします。

環境

Python
- 3.12

ゴール

PythonのFCC（First Class Collection）にて、For文を使用する際にFCCのインスタンスのまま使用できること。

from typing import List

class User():
  # ... 省略

class Users():
  value: List[User]

  # コンストラクタ省略


users = Users([])
# この記載ができることがゴール
[user for user in users]
# ※ もちろん、対応前でも値を取得すれば元の通りにアクセスできます。
[user for user in users.value]

実装

iterメソッドか、yield fromで内部の値にアクセスすることで、外側からは意識せずともFor文で使用できます。

from typing import List, Iterable

class Users():
  value: List[User]

  def __iter__(self) -> Iterable[User]:
    # イテレータプロトコルを実装する
    return iter(self.value)
    # yield from でListのイテレータを呼ぶこともできる
    yield from self.value