Question

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

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

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

Answer 1

已经有字符串方法.endswith

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

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