如何使函数的单个参数采取多个不同的可接受的字符串值?

Question

我正在编写一个基于用户输入到函数参数的数据处理函数。

用户被限制只能提供一些可接受的区分大小写的输入。

请检查下面的函数，并让我知道我是否以正确的方式做它，或者有其他更好的或python的方式做它?

def test_func(param_1: str):
acceptable_params = ['first', 'second', 'third']
if param_1 not in acceptable_params:
raise Exception(f''{param_1}' is not an acceptable word')
print(param_1)

这对我来说很好，但我很好奇是否有更好的这样做或使它看起来更专业?

任何帮助都会很感激。由于

Answer 1

assert param_1 in ['first', 'second', 'third'], f"'{param_1}' is not an acceptable word"