使用python将文件列表添加到字典中

Question

我编写了此脚本，它将显示特定目录中的所有文件，并让用户输入他想要编辑的文件。

import os 

path = "/home/luai/Desktop/python/test" 
dirs = os.listdir(path) 

print "Here is a list of all files: " 

for files in dirs: 
    print files 

filename = raw_input ("which file to update: ") 
if filename in dirs: 
    inputFile = open(filename, 'r') 
else: 
    print "no match found" 
    sys.exit() 
    inputFile.close() 

和工作，但事情是，我希望用户只输入数字或字母开的，而不是写文件的全名的文件。有任何想法吗 ？

谢谢。

Answer 1

我只想在与索引的listdir输出回路：

import os 

path = "/home/luai/Desktop/python/test" 
dirs = os.listdir(path) 

print "Here is a list of all files: " 

for filenumber in range(len(dirs)): 
    print filenumber, dirs[filenumber] 

filenumber = raw_input ("Number of file which to update: ") 
filenumber = int(filenumber) 

if dirs[filenumber] in dirs: 
    inputFile = open(os.path.join(path, dirs[filenumber])) 
    inputFile.close() #this line was at a weird place? 
else: 
    print "no match found" 
    sys.exit() 

但是，这仍然有listdir还为您提供目录的问题，你不能open那些。 所以最好的是只从path获得文件。

import os 

path = "/home/luai/Desktop/python/test" 
files = os.walk(path).next()[2] 


print "Here is a list of all files: " 

for filenumber in range(len(files)): 
    print filenumber, files[filenumber] 

filenumber = raw_input ("Number of file which to update: ") 
filenumber = int(filenumber) 

if files[filenumber] in files: 
    inputFile = open(os.path.join(path, files[filenumber])) 
    inputFile.close() 
else: 
    print "no match found" 
    sys.exit() 

Answer 2

您可以创建一个字典，其中包含值作为文件名和密钥的任意位置。根据用户输入，检查密钥是否存在，并打开相应的文件名称值。

Answer 3

鲁本有正确的想法，但它可以多一点优雅：

chosen_file_index = -1 # Intentionally invalid index 

while chosen_file_index not in range(0, len(filenames)-1): 

    filenames = os.listdir(path) 

    if not filenames: 
      print "No files found" 
      break 

    for file_index, filename in enumerate(filenames): 
     print "{index}. {filename}".format(index=file_index, filename=filename) 

    chosen_file_index = raw_input("Which file to update:") 

chosen_filename = filenames[chosen_file_index) 

所以在这里我添加的想法，系统会不断地问，直到用户选择从列表中的有效条目并处理了没有文件的情况。