Python限制字符输入

Question

我尝试只允许在 Python 程序中的 .dat 文件的输入中使用字母 a、b 或 c(代码如下)，但我很难让程序正确执行此操作。

varClass = "class" + input("Which class are you in? [A/B/C]: ").lower() + ".dat"
if not re.match("^[a-c]*$", varClass):
    print("Enter the correct class number")

这是我已经拥有的，但即使输入了错误的字符后它仍然继续运行。

Answer 1

varClass = "class" + input("Which class are you in? [A/B/C]: ").lower() + ".dat"
if not re.match("^class[a-c]\.dat$", varClass):
    print("Enter the correct class number")

您的match将始终返回false，因为match从头开始匹配，并且您在开头有class。