Python-获取文件目录作为用户输入

Question

从用户定义的目录获取文件列表时遇到一些问题。下面的代码工作正常:

inputdirectory = r'C:/test/files'
inputfileextensions = 'txt'
files = glob.glob(inputdirectory+"*."+inputfileextensions)

但我想允许用户输入位置。我尝试过以下代码:

inputdirectory = input("Please type in the full path of the folder containing your files:    ")
inputfileextensions = input("Please type in the file extension of your files:    ")
files = glob.glob(inputdirectory+"*."+inputfileextensions)

但是这不起作用。不会出现错误消息，但文件返回为空。我尝试在目录中输入引号、正斜杠和反斜杠，但无法使其工作。我还尝试使用“r”将输入转换为原始字符串，但语法可能是错误的。有什么想法吗？

Answer 1

不太确定第一个版本对您来说如何。定义变量的方式，您应该将 glob 的输入如下所示:

inputdirectory+"*."+inputfileextensions == "C:\test\files*.txt"

查看上述值，您可以意识到这并不是您想要实现的目标。相反，您需要使用反斜杠运算符连接两个路径。像这样的东西:

os.path.join(inputdirectory, "*."+inputfileextensions) == "C:\test\files\*.txt"

通过此更改，无论输入是来自用户还是预定义的，代码都应该可以正常工作。