From 5e886308e60ac20ec7427d645370b7330ae88268 Mon Sep 17 00:00:00 2001 From: Jordan Date: Sat, 5 Dec 2020 18:32:00 -0700 Subject: use os.scandir() over os.walk(), better performance --- lib/books.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/lib/books.py b/lib/books.py index 86ab4b6..83d096d 100644 --- a/lib/books.py +++ b/lib/books.py @@ -31,10 +31,9 @@ class Books: ''' Return list of directories recursively discovered in :path: ''' - ret = list() - for root, dirs, _ in os.walk(path): - for d in dirs: - ret.append(os.path.join(root, d)) + s = os.scandir(path) + ret = [x.path for x in s if not x.name.startswith('.') and x.is_dir()] + s.close() return ret -- cgit v1.2.3-54-g00ecf