如何在用户输入结束时检查某些字符?Python

Question

我的程序要求用户输入要打开的文件的名称。

如果他们将.txt放在最后，则将打开文件。

如何使程序防止非.txt扩展，以便我的程序无论如何都会添加它，以便输入正确的程序。

Answer 1

已经有字符串方法.endswith

>>> 'test.txt'.endswith('.txt')
True
>>> 'test.png'.endswith('.txt')
False

如果它尚未以正确的扩展名结尾，则您必须考虑可能获得哪些输入并为其添加。这可能是错误的扩展名或没有扩展。这可能会很棘手地捕获所有边缘案例。