perf: optimize existing folder detection logic

skymkmk · Mar 5, 2023 · f747804 · f747804
1 parent 561180b
commit f747804
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/init.py b/src/init.py
@@ -27,7 +27,7 @@ def init(config: dict):
     # New save path if not exists
     if not os.path.exists(config['save_path']):
         os.makedirs(config['save_path'])
-    dirs = os.listdir(config['save_path'])
+    dirs = [i for i in os.listdir(config['save_path']) if os.path.isdir(os.path.join(config['save_path'], i))]
     tmp = []
     # Detect unrelated pages
     for i in dirs: