Выбор файла из указанного каталога в Python

Question

Я хочу выбрать файл по номерам в левой части списка, но я не могу пойти дальше, чем это:

import os
path="/root/Desktop"
dirList=os.listdir(path)
for fname in dirList:
    print fname

selected = raw_input("Select a file above: ")

Что мне делать?

Пример:

http://img502.imageshack.us/img502/4407/listingy.png

Заранее спасибо ..

Answer 1

Вы должны использовать перечисление для списка, а затем обрабатывать ошибки ввода. В идеале это будет функция, и вместо break вы просто вернете выбранный файл.

import os

path="/root/Desktop"
dirList=os.listdir(path)

for i, fname in enumerate(dirList):
    print "%d) %s" % (i + 1, fname)

while True:
    try:
        selectedInt = int(raw_input("Select a file above: "))
        selected = dirList[selectedInt - 1]
        break
    except Exception:
        print "Error: Please enter a number between 1 and %d" % len(dirList)

Answer 2

for i, fname in enumerate(dirList):
    print "%s) %s" % (i + 1, fname)

selectedInt = int(raw_input("Select a file above: "))
selected = dirList[selectedInt - 1]

Однако учтите, что исправление ошибок не выполняется.Вы должны ловить случаи, когда ввод не является целым числом.

Answer 3

Создайте словарь, в котором ключ является индексом, а значение - именем файла (используйте enumerate (dirList), чтобы получить кортеж (index, filename) за одну итерацию).Затем просто вытяните соответствующее значение на основе ввода (клавиша).

Answer 4

Попробуйте это:

import os

listed_files = []
for i, fname in enumerate(os.listdir("/root/Desktop")):
    print i, ': ', fname
    listed_files.append(fname)

selected = raw_input("Select a file above: ")
open("/root/Desktop/%s" % listed_files[int(selected)])

Answer 5

Вы знаете имя файла по определенному индексу, так что вы можете просто сделать что-то вроде этого:

import os
path="/root/Desktop"
dirList=os.listdir(path)

for index in range(0,len(dirList)):
    print str(index+1) + ": " + dirList[index]

selected = raw_input("Select a file above: ")
print "You selected filename: " + dirList[selected];

редактировать: ой, чтобы замедлить

Answer 6

Вы можете попробовать следующее:

import os
path="/root/Desktop"
dirList=os.listdir(path)
for i in range(0,len(dirList)): # generate an index an loop over it
    print "%d)" % (i+1), dirList[i] # print a selection number matching each file

selected = raw_input("Select a file above: ")
selected = int(selected) # cast the input to int

print "You have selected:", dirList[selected-1] # you can get the corresponding entry!

Это должно сработать:)