python - 枚举列表中的项目，以便用户可以选择数值

Question

我正在尝试找到最直接的方法来枚举列表中的项目，这样用户就不会因为在命令行中键入长文件名而感到负担。下面的函数向用户显示文件夹中的所有 .tgz 和 .tar 文件……然后允许用户输入他想要提取的文件的名称。这对用户来说既乏味又容易出现语法错误。我希望用户只需选择与文件关联的数值(例如 .. 1、2、3 等)。有人可以给我一些指导吗？谢谢!

  dirlist=os.listdir(path)

  def show_tgz():
     for fname in dirlist:
          if fname.endswith(('.tgz','.tar')):
             print '\n'
             print fname

Answer 1

从文件列表开始:

files = [fname for fname in os.listdir(path) 
               if fname.endswith(('.tgz','.tar'))]

现在您可以从字面上枚举它们:

for item in enumerate(files):
    print "[%d] %s" % item

try:
    idx = int(raw_input("Enter the file's number"))
except ValueError:
    print "You fail at typing numbers."

try:
    chosen = files[idx]
except IndexError:
    print "Try a number in range next time."