Python Developer’s Meetup Nepal #13 Question Solution
Q: Write a python script that recursively walks all sub-directories and searches all files with extension *.png or *.PNG and append them to the list.
Solution:
import os # list variable to store the .png and .PNG file png_file_list = [] # Recursive function to traverse all the sub-dirctory and check for png files def traverse_directory(dir_path): for child in os.listdir(dir_path): path = os.path.join(dir_path, child) if os.path.isdir(path): traverse_directory(path) else: # split the path and store file name only file_name = path.split('\\')[-1] # split the file name and store extension file_name_extension = file_name.split('.')[-1] # check if the file name has extension .png or .PNG if(file_name_extension == 'png' or file_name_extension == 'PNG'): png_file_list.append(file_name) # Traverse current directory for files and folders traverse_directory(".") # Print the final list print(png_file_list)
Advertisements