1. ホーム
  2. python

[解決済み] Enum属性の型アノテーション

2023-07-05 09:06:06

質問

このようなコードがあります。

import enum


class Color(enum.Enum):
    RED = '1'
    BLUE = '2'
    GREEN = '3'


def get_color_return_something(some_color):
    pass

の型アノテーションを適切に追加するにはどうすればよいでしょうか? some_color という変数に、Color enum から enum 属性を受け取ると仮定した場合、この関数の中でどのように型アノテーションを付ければよいでしょうか(例えば。 Color.RED )?

どのように解決するのですか?

Colorクラスのタイプヒントが動作するはずです。

def get_color_return_something(some_color: Color):
    print(some_color.value)