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

Question

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

如果他们把.txt放在最后，文件就会被打开。

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

Answer 1

已经有字符串方法.endswith

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

如果它还没有以正确的扩展名结束，您必须考虑可能获得哪些输入并为其添加。这可能是错误类型的扩展或没有扩展。捕捉所有边缘情况可能会很棘手。