Python 在函数参数中键入枚举类型

Question

我有这个 python 枚举

from enum import Enum

class TypeItem(Enum):
    ITEM_1 = 'revenue'
    ITEM_2 = 'net_income_after_taxes'

我想将它用作函数参数。该函数看起来像这样；

def function_use(type_item: TypeItem) -> None:
    return None

type_item: TypeItem 可能是错误的。 TypeItem 枚举的正确输入应该是什么？

我正在使用 python 3.7

Answer 1

我会回答我自己的问题。感谢 ForceBru 和 Georgy 的评论，正确答案已经在问题中了。

def function_use(type_item: TypeItem) -> None:
    return None

Python typing for enum type in function argument