import os
def remove_by_ext(path,ext):
'''Walks through directories removing files with specified extension'''
for (path,dirs,files) in os.walk(path):
for file_name in files:
file_name_lower=file_name.lower()
if file_name_lower[-len(ext):] == ext.lower():
os.remove(path+'\\'+file_name)
print('File removed: %s' %file_name)
path = raw_input('Input Path: ')
ext = raw_input('Input Extension: ')
remove_by_ext(path,ext)