在Python中,可以使用enum
模块来定义枚举类型。下面是一个编写通用函数的示例,该函数返回一个枚举键的类型:
from enum import Enum
def get_enum_key_type(enum_cls):
"""
返回枚举键的类型
:param enum_cls: 枚举类
:return: 键的类型
"""
if not issubclass(enum_cls, Enum):
raise ValueError("传入的参数不是枚举类")
# 获取枚举类的一个实例
enum_instance = next(iter(enum_cls))
# 返回键的类型
return type(enum_instance.value)
# 示例枚举类
class MyEnum(Enum):
KEY1 = 'value1'
KEY2 = 'value2'
# 调用函数并打印结果
print(get_enum_key_type(MyEnum))
运行以上代码,将会输出:
这表示枚举类MyEnum
的键的类型为str
。