Sandbox::https://github.com/ThinkCode/python-scripts/blob/master/10_find_files

Repository navigation

31 lines (24 loc) · 833 Bytes

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)