搜索作为输入提供的文件目录?

时间:2017-05-16 10:01:49

标签: python

所以我被赋予了从用户那里获取3个参数的任务,然后执行以下任务:

搜索作为输入的文件夹。 查找所有特定文件类型的扩展名。 将其打印到另一个文件夹。

是否有更简单的方法来执行此任务?尝试使用os.listdir会响应它无法找到该文件,因为它不接受变量作为输入。

Directories = [];
InitDirect = str(input('Please insert the file directory you want to search (C:\\x)'))

FileType = str(input('Please state a desired file type (.txt, .png)'))

OutDirect = str(input('Please state the output directory for the files.'))

for file in os.listdir("InitDirect"):
    if file.endswith("FileType"):
        print(os.path.join("InitDirect", file))

这是我目前的代码,虽然可能不正确。如果有人能提供帮助,那就太好了!

1 个答案:

答案 0 :(得分:1)

不需要在变量名称周围使用引号。添加""变量名称周围实际上声明了字符串,而您没有使用变量的值。将代码更改为以下内容,它应该可以正常工作。

Directories = [];
InitDirect = str(input('Please insert the file directory you want to search (C:\\x)'))

FileType = str(input('Please state a desired file type (.txt, .png)'))

OutDirect = str(input('Please state the output directory for the files.'))

for file in os.listdir(InitDirect):
    if file.endswith(FileType):
        print(os.path.join(InitDirect, file))