python - 将参数限制为 python 中的特定值

Question

如果我有一个带有字符串参数的 python 函数，它需要在特定的值列表中，处理这个问题的最佳方法是什么？是否有在文档字符串或其他内容中表达这一点的约定？

例如。

def makeShape(shape):
   assert shape in ["circle", "square", "triangle"]

有没有更好的方式来表达形状应该始终是这三个值之一？

Answer 1

您可以使用 enum 来正确排序形状:

class rightShape(Enum):
    circle = 1
    square = 2
    triangle = 3

def makeShape(rightShape):
    # code in function

如果您尝试向函数传递除 circle、square 或 triangle 之外的任何内容，这将抛出一个诚实至善的错误.