前言
我们在日常开发中,经常会遇到一些关于文件的操作,例如,实现查看目录内容的功能。类似Linux下的tree命令。统计目录下指定后缀文件的行数。
功能是将目录下所有的文件路径存入list中。可以加入后缀判断功能,搜索指定的后缀名文件。主要利用递归的方法来检索文件。
Python2.7
递归法
import os def tree_dir(path,c_path='',is_root=True): """ Get file list under path. Like 'tree' :param path Root dir :param c_path Child dir :param is_root Current is root dir """ res = [] if not os.path.exists(path): return res for f in os.listdir(path): if os.path.isfile(os.path.join(path,f)): if is_root: res.append(f) else: res.append(os.path.join(c_path,f)) else: res.extend(tree_dir(os.path.join(path,f),f,is_root=False)) return res
下面是加入后缀判断的方法。在找到文件后,判断一下是否符合后缀要求。不符合要求的文件就跳过。
def tree_dir_sur(path,is_root=True,suffix=''): """ Get file list under path. Like 'tree' :param path Root dir :param c_path Child dir :param is_root Current is root dir :param suffix Suffix of file """ res = [] if not os.path.exists(path) or not os.path.isdir(path): return res for f in os.listdir(path): if os.path.isfile(os.path.join(path,f)) and str(f).endswith(suffix): if is_root: res.append(f) else: res.append(os.path.join(c_path,f)) else: res.extend(tree_dir_sur(os.path.join(path,is_root=False,suffix=suffix)) return res if __name__ == "__main__": for p in tree_dir_sur(os.path.join('E:\ws','rnote','Python_note'),suffix='md'): print p
# -*- coding: utf-8 -*- import os def count_by_categories(path): """ Find all target files and count the lines """ if not os.path.exists(path): return c_l_dict = dict() # e.g. {category: lines} category_list = [cate for cate in os.listdir(path) if os.path.isdir(os.path.join(path,cate)) and not cate.startswith('.')] for category_dir in category_list: line_count = _sum_total_line(os.path.join(path,category_dir),'.md') if line_count > 0: c_l_dict[category_dir] = line_count return c_l_dict def _sum_total_line(path,endswith='.md'): """ Get the total lines of target files """ if not os.path.exists(path) or not os.path.isdir(path): return 0 total_lines = 0 for f in os.listdir(path): if f.endswith(endswith): with open(os.path.join(path,f)) as cur_f: total_lines += len(cur_f.readlines()) return total_lines if __name__ == '__main__': note_dir = 'E:/ws/rnote' ca_l_dict = count_by_categories(note_dir) all_lines = 0 for k in ca_l_dict.keys(): all_lines += ca_l_dict[k] print 'all lines:',str(all_lines) print ca_l_dict
all lines: 25433 {'flash_compile_git_note': 334,'Linux_note': 387,'Algorithm_note': 3637,'Comprehensive': 216,'advice': 137,'Java_note': 3013,'Android_note': 11552,'DesignPattern': 2646,'Python_note': 787,'kotlin': 184,'cpp_note': 279,'PyQt_note': 439,'reading': 686,'backend': 1136}
总结
以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作能带来一定的帮助,如果有疑问大家可以留言交流,谢谢大家对编程小技巧的支持。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。