Skip to content

Commit aace5de

Browse files
committed
fix(lib): correctly caching and simplified structure
1 parent 738c54f commit aace5de

File tree

1 file changed

+7
-27
lines changed

1 file changed

+7
-27
lines changed

manim_slides/slide.py

Lines changed: 7 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
MutableMapping,
1010
Optional,
1111
Sequence,
12-
Set,
1312
Tuple,
1413
ValuesView,
1514
)
@@ -478,23 +477,12 @@ def __save_slides(self, use_cache: bool = True) -> None:
478477
"""
479478
self.__add_last_slide()
480479

481-
self.__output_folder.mkdir(parents=True, exist_ok=True)
482-
483480
files_folder = self.__output_folder / "files"
484-
files_folder.mkdir(exist_ok=True)
485481

486482
scene_name = str(self)
487483
scene_files_folder = files_folder / scene_name
488484

489-
old_animation_files: Set[Path] = set()
490-
491-
if not scene_files_folder.exists():
492-
scene_files_folder.mkdir()
493-
elif not use_cache:
494-
shutil.rmtree(scene_files_folder)
495-
scene_files_folder.mkdir()
496-
else:
497-
old_animation_files.update(scene_files_folder.iterdir())
485+
scene_files_folder.mkdir(parents=True, exist_ok=True)
498486

499487
files = []
500488
for src_file in tqdm(
@@ -510,24 +498,16 @@ def __save_slides(self, use_cache: bool = True) -> None:
510498
# but animations before a will have a None src_file
511499
continue
512500

513-
filename = src_file.name
514-
rev_filename = (
515-
src_file.parent / f"{src_file.stem}_reversed{src_file.suffix}"
516-
)
517-
dst_file = scene_files_folder / filename
501+
dst_file = scene_files_folder / src_file.name
502+
rev_file = scene_files_folder / f"{src_file.stem}_reversed{src_file.suffix}"
503+
518504
# We only copy animation if it was not present
519-
if filename in old_animation_files:
520-
old_animation_files.remove(filename)
521-
else:
505+
if not use_cache or not dst_file.exists():
522506
shutil.copyfile(src_file, dst_file)
523507

524508
# We only reverse video if it was not present
525-
if rev_filename in old_animation_files:
526-
old_animation_files.remove(rev_filename)
527-
else:
528-
rev_file = scene_files_folder / rev_filename
529-
if not rev_file.exists():
530-
reverse_video_file(src_file, rev_file)
509+
if not use_cache or not rev_file.exists():
510+
reverse_video_file(src_file, rev_file)
531511

532512
files.append(dst_file)
533513

0 commit comments

Comments
 (0)