1 使用

通过os.walk()扫描制定目录下的所有文件以及子文件夹

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
'''
扫描目录下的第一层文件,不包括子文件夹中的文件
'''
def get_dir_lists(root_dir):
root_dir = r"J:\电子书"
dir_tree = os.walk(root_dir)
sub_dir = dir_tree.__next__()[1]
file_list = []
for sub in sub_dir:
for _sub in os.walk(root_dir + "\\" + str(sub)):
_book_list = _sub[-1]
if _sub[1] is not None:
for book in _book_list:
if ".pdf" in book:
file_list.append(book)
return file_list

写在最后

欢迎大家关注鄙人的公众号【麦田里的守望者zhg】,让我们一起成长,谢谢。
微信公众号