31 lines
838 B
Python
31 lines
838 B
Python
|
import fnmatch
|
||
|
|
import os
|
||
|
|
|
||
|
|
# constants
|
||
|
|
PATH = '/../../../..'
|
||
|
|
PATTERN = '*.py'
|
||
|
|
|
||
|
|
|
||
|
|
def get_file_names(filepath, pattern):
|
||
|
|
matches = []
|
||
|
|
if os.path.exists(filepath):
|
||
|
|
for root, dirnames, filenames in os.walk(filepath):
|
||
|
|
for filename in fnmatch.filter(filenames, pattern):
|
||
|
|
# matches.append(os.path.join(root, filename)) # full path
|
||
|
|
matches.append(os.path.join(filename)) # just file name
|
||
|
|
if matches:
|
||
|
|
print "Found {} files:".format(len(matches))
|
||
|
|
output_files(matches)
|
||
|
|
else:
|
||
|
|
print "No files found."
|
||
|
|
else:
|
||
|
|
print "Sorry that path does not exist. Try again."
|
||
|
|
|
||
|
|
|
||
|
|
def output_files(list_of_files):
|
||
|
|
for filename in list_of_files:
|
||
|
|
print filename
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
all_files = get_file_names(PATH, PATTERN)
|