Как искать файлы с известным расширением, как .py?

Question

как искать файлы с известным расширением, например .py ??

fext = raw_input("Put file extension to search: ")
dir = raw_input("Dir to search in: ")

##Search for the file and get the right one's

Answer 1

Я считаю, что вы хотите сделать что-то похожее на это: /dir/to/search/*.extension?

Это называется glob, и вот как его использовать:

import glob
files = glob.glob('/path/*.extension')

Редактировать: и вот документация: http://docs.python.org/library/glob.html

Answer 2

import os
root="/home"
ext = raw_input("Put file extension to search: ")
path = raw_input("Dir to search in: ")
for r,d,f in os.walk(path):
    for files in f:
         if files.endswith(ext):
              print "found: ",os.path.join(r,files)

Answer 3

Вы можете написать так же просто, как:

import os
ext = raw_input("Put file extension to search: ")
path = raw_input("Dir to search in: ")
matching_files = [os.path.join(path, x) for x in os.listdir(path) if x.endswith(ext)]

Answer 4

нерекурсивно:

for x in os.listdir(dir):
    if x.endswith(fext):
        filename = os.path.join(dir, x)
        # do your stuff here