Skip to content

Commit 42ceb38

Browse files
committed
parallel cleanup
1 parent 127aebc commit 42ceb38

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/main.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
from pathlib import Path
99

1010
import pymupdf
11+
from joblib import Parallel, delayed
1112
from natsort import natsorted, ns
1213
from PIL import Image
1314

@@ -113,8 +114,11 @@ def merge(base: Path, root: str, files: list[str]) -> None:
113114
),
114115
)
115116

116-
for root, _, files in os.walk(pdfs / "todo"):
117-
cleanup(root, files)
117+
# this can be parallel
118+
Parallel(n_jobs=-1)(
119+
delayed(cleanup)(root, files) for root, _, files in os.walk(pdfs / "todo")
120+
)
118121

122+
# but the order here matters
119123
for root, _, files in os.walk(pdfs / "done"):
120124
merge(pdfs, root, files)

0 commit comments

Comments
 (0)