| #!/usr/bin/python | |
| # -*- coding:utf-8 -*- | |
| import os | |
| from os.path import basename, splitext | |
| def get_filename(path): | |
| return basename(splitext(path)[0]) | |
| def cnt_num_files(directory, recursive=False): | |
| cnt = 0 | |
| for sub in os.listdir(directory): | |
| sub = os.path.join(directory, sub) | |
| if os.path.isfile(sub): | |
| cnt += 1 | |
| elif os.path.isdir(sub) and recursive: | |
| cnt += cnt_num_files(sub, recursive=True) | |
| return cnt |