(target_dir)
| 625 | extract_images_from_docx(f, png_dir)""" |
| 626 | |
| 627 | def clean_up_folder(target_dir): |
| 628 | if not os.path.isdir(target_dir): |
| 629 | prRed(f"'{target_dir}' is not a valid directory.") |
| 630 | return |
| 631 | |
| 632 | for filename in os.listdir(target_dir): |
| 633 | file_path = os.path.join(target_dir, filename) |
| 634 | |
| 635 | if os.path.isfile(file_path): |
| 636 | ext = os.path.splitext(filename)[1].lower().lstrip(".") |
| 637 | |
| 638 | # Group jpeg and jpg into same folder |
| 639 | if ext in ("jpeg", "jpg"): |
| 640 | ext_folder = "jpg" |
| 641 | else: |
| 642 | ext_folder = ext or "unknown" |
| 643 | |
| 644 | # Create destination folder if it doesn't exist |
| 645 | dest_folder = os.path.join(target_dir, ext_folder) |
| 646 | os.makedirs(dest_folder, exist_ok=True) |
| 647 | |
| 648 | # Move the file |
| 649 | shutil.move(file_path, os.path.join(dest_folder, filename)) |
| 650 | |
| 651 | |
| 652 |
no test coverage detected