脚本：批量移动子目录中的 PDF

"""
file: extract_files.py
author: linu
date: 2024-10-11
description: 从当前目录下的各个文件夹中抽取所有 PDF 文件，放入目标目录。
"""

import os
import shutil

from tqdm import tqdm

source_dir = "./"
dest_dir = "./dest"

# 判断目标目录是否存在，不存在则创建
if not os.path.exists(dest_dir):
    os.mkdir(dest_dir)

# 待处理的子目录：是目录，且不是目标目录
folder_list = [
    _
    for _ in os.listdir(source_dir)
    if os.path.isdir(_) and not os.path.samefile(_, dest_dir)
]

filepath_list = []
for folder in folder_list:
    # 子目录中的文件
    file_list = [
        _
        for _ in os.listdir(os.path.join(source_dir, folder))
        if os.path.isfile(os.path.join(source_dir, folder, _))
    ]
    # 只保留 PDF
    for file in file_list:
        if file.endswith(".pdf"):
            filepath_list.append((os.path.join(source_dir, folder, file), file))

loop = tqdm(filepath_list)
# 移入目标目录
for filepath, filename in loop:
    shutil.move(filepath, os.path.join(dest_dir, filename))
    loop.set_postfix({"file": filename})
loop.close()
print("Done!")