how to search for file's has a known file extension like .py?

Question

how to search for file's has a known file extension like .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

I believe you want to do something like similar to this: /dir/to/search/*.extension ?

This is called glob and here is how to use it:

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

Edit: and here is the documentation: 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

Non-recursive:

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

Answer 4

You can write is as simple as:

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)]