Created
May 7, 2018 08:18
-
-
Save mgoksu/30b4c821dfae33757eae9304c2cc5bc5 to your computer and use it in GitHub Desktop.
get the file list given a folder
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import os | |
| def get_file_list(src_fol, ext): | |
| f_list = [] | |
| for subdir, _, files in os.walk(src_fol): | |
| for filename in files: | |
| _, f_ext = os.path.splitext(filename) | |
| if f_ext == ext: | |
| full_file_path = os.path.join(subdir, filename) | |
| f_list.append(full_file_path) | |
| return f_list | |
| def get_eq_file_list(src_fol, trgt_fol, in_ext, out_ext): | |
| f_src_list = [] | |
| f_trgt_list = [] | |
| for subdir, dirs, files in os.walk(src_fol): | |
| for filename in files: | |
| if not filename.endswith(in_ext): | |
| continue | |
| # create subdir for writing | |
| sub_fol = subdir.split("/")[-1] | |
| trgt_subfol = os.path.join(trgt_fol, sub_fol) | |
| if not os.path.exists(trgt_subfol): | |
| os.makedirs(trgt_subfol) | |
| src_file = os.path.join(subdir, filename) | |
| f_src_list.append(src_file) | |
| filename = filename.replace(in_ext, out_ext) | |
| # make the path | |
| trgt_file = os.path.join(trgt_subfol, filename) | |
| f_trgt_list.append(trgt_file) | |
| return f_src_list, f_trgt_list | |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment