如何为Python函数参数指定ButtonTypes类属性的专属类型？

我定义了如下ButtonTypes类：

class ButtonTypes:
    def __init__(self):
        self.textType = "text"
        self.callbackType = "callback"
        self.locationType = "location"
        self.someAnotherType = "someAnotherType"

另有一个函数create_button，其参数需为ButtonTypes类的某一属性值：

def create_button(button_type):
    pass

我希望指定该函数的参数不只是普通字符串，而必须是ButtonTypes类的属性值之一，例如想写成：

def create_button(button_type: ButtonTypes.Type):
    pass

我原本考虑在ButtonTypes类内创建Type类，再让各类型类继承它，但觉得思路有误，该如何实现？

方法1：使用enum.Enum（最推荐） #

用Python内置的枚举类来定义按钮类型，既能统一管理所有合法类型，又能通过静态类型检查工具（比如mypy）确保参数合法：

from enum import Enum

class ButtonType(Enum):
    TEXT = "text"
    CALLBACK = "callback"
    LOCATION = "location"
    SOME_ANOTHER_TYPE = "someAnotherType"

def create_button(button_type: ButtonType):
    # 需要字符串值时，通过`button_type.value`获取
    print(f"创建类型为{button_type.value}的按钮")

# 合法调用
create_button(ButtonType.TEXT)
# 非法调用会触发类型检查报错（比如传入普通字符串）
# create_button("text")

方法2：类属性+Literal类型（保留字符串使用方式） #

如果想保留直接使用字符串参数的习惯，同时限制只能传入指定的几个字符串，可以用typing.Literal定义字面量类型：

from typing import Literal

# 先把ButtonTypes改成类属性（避免实例化）
class ButtonTypes:
    TEXT_TYPE = "text"
    CALLBACK_TYPE = "callback"
    LOCATION_TYPE = "location"
    SOME_ANOTHER_TYPE = "someAnotherType"

# 基于类属性生成Literal类型，统一维护
ButtonTypeLiteral = Literal[
    ButtonTypes.TEXT_TYPE,
    ButtonTypes.CALLBACK_TYPE,
    ButtonTypes.LOCATION_TYPE,
    ButtonTypes.SOME_ANOTHER_TYPE
]

def create_button(button_type: ButtonTypeLiteral):
    pass

# 两种合法调用方式
create_button(ButtonTypes.TEXT_TYPE)
create_button("text")
# 非法调用会被类型检查工具识别
# create_button("invalid_type")

方法3：运行时参数验证（无静态类型检查需求） #

如果不需要静态类型检查，只需要在代码运行时确保参数合法，可以在函数内做验证：

class ButtonTypes:
    TEXT_TYPE = "text"
    CALLBACK_TYPE = "callback"
    LOCATION_TYPE = "location"
    SOME_ANOTHER_TYPE = "someAnotherType"

# 提前收集所有合法的按钮类型值
VALID_BUTTON_TYPES = {v for k, v in ButtonTypes.__dict__.items() if not k.startswith("__")}

def create_button(button_type):
    if button_type not in VALID_BUTTON_TYPES:
        raise ValueError(f"无效按钮类型，允许的类型为：{VALID_BUTTON_TYPES}")
    # 后续逻辑
    pass

# 合法调用
create_button(ButtonTypes.TEXT_TYPE)
# 非法调用会抛出ValueError
# create_button("invalid_type")

内容的提问来源于stack exchange，提问作者user18196171