import shutil
from pathlib import Path


def rename_files(folder_path: Path):
    """
    读取目录下的所有 .mp4 文件，在文件名中添加父级文件夹名
    """
    for i, file in enumerate(folder_path.glob('*.mp4')):
        # 如果文件名中已经包含父级文件夹名，则跳过
        if folder_path.name + "_" in file.name:
            continue

        file.rename(folder_path / f'{folder_path.name}_{Path(file.name).stem}.mp4')
        print(f'{folder_path.name}_{Path(file.name).stem}.mp4')


def move_files(folder_path: Path):
    """
    将目录下的所有 .mp4 文件移动到父级目录下
    """
    for i, file in enumerate(folder_path.glob('*.mp4')):
        # 如果文件名中已经包含父级文件夹名，则跳过
        if folder_path.name + "_" in file.name:
            continue
        file.rename(folder_path.parent / f'{Path(file.name).stem}.mp4')
        print(f'{Path(file.name).stem}.mp4')


def delete_folder_contents_and_folder_itself(folder_path: Path):
    """
    删除目录及其包含的所有文件和子目录
    """
    # First, delete all files and subfolders within the directory
    for item in folder_path.iterdir():
        if item.is_file():
            item.unlink()
        elif item.is_dir():
            shutil.rmtree(item)

    # Now that the directory is empty, safely remove it
    folder_path.rmdir()
    print(f"Deleted directory: {folder_path}")


if __name__ == '__main__':
    from tool_delete_files import delete_files

    folder_path = Path(r'D:\管理认证\98-项目管理\项目管理阅读')

    # delete_files(folder_path)
    rename_files(folder_path)
    # move_files(folder_path)
    #
    # # 删除文件夹
    # delete_folder_contents_and_folder_itself(folder_path)