import os
import shutil
from pathlib import Path

def batch_rename_files(
    folder_path: str,
    prefix: str = "file",
    suffix: str = "",
    start_num: int = 1,
    file_ext: str = "*"
) -> None:
    """
    批量重命名文件夹下的文件
    :param folder_path: 目标文件夹路径
    :param prefix: 文件名前缀（默认：file）
    :param suffix: 文件名后缀（默认：空）
    :param start_num: 序号起始值（默认：1）
    :param file_ext: 指定文件类型（如 ".xlsx"，* 表示所有文件）
    """
    # 校验文件夹是否存在
    if not os.path.exists(folder_path):
        print(f"错误：文件夹 {folder_path} 不存在！")
        return
    
    # 转换为 Path 对象，方便路径处理
    folder = Path(folder_path)
    # 获取目标文件列表
    if file_ext == "*":
        file_list = [f for f in folder.iterdir() if f.is_file()]
    else:
        file_list = list(folder.glob(f"*{file_ext}"))
    
    if not file_list:
        print(f"未找到 {file_ext} 类型的文件！")
        return
    
    # 批量重命名
    for idx, file in enumerate(file_list, start=start_num):
        # 保留原文件扩展名
        ext = file.suffix
        # 新文件名：前缀 + 序号 + 后缀 + 扩展名
        new_name = f"{prefix}{idx}{suffix}{ext}"
        new_path = folder / new_name
        
        # 避免文件名重复（如果已存在则加 _1、_2...）
        count = 1
        while new_path.exists():
            new_name = f"{prefix}{idx}{suffix}_{count}{ext}"
            new_path = folder / new_name
            count += 1
        
        # 执行重命名
        file.rename(new_path)
        print(f"重命名成功：{file.name} → {new_name}")

def organize_files_by_type(folder_path: str) -> None:
    """
    按文件类型自动整理文件到对应文件夹
    :param folder_path: 目标文件夹路径
    """
    if not os.path.exists(folder_path):
        print(f"错误：文件夹 {folder_path} 不存在！")
        return
    
    folder = Path(folder_path)
    # 定义文件类型与目标文件夹的映射（可根据需求扩展）
    file_type_mapping = {
        # 办公文档
        "Excel": [".xlsx", ".xls", ".csv"],
        "Word": [".docx", ".doc"],
        "PPT": [".pptx", ".ppt"],
        "PDF": [".pdf"],
        # 图片
        "Image": [".jpg", ".jpeg", ".png", ".gif", ".bmp"],
        # 视频
        "Video": [".mp4", ".avi", ".mov", ".mkv"],
        # 音频
        "Audio": [".mp3", ".wav", ".flac"],
        # 压缩包
        "Compress": [".zip", ".rar", ".7z"],
        # 代码文件
        "Code": [".py", ".java", ".cpp", ".js", ".html"],
        # 其他文件
        "Other": []
    }

    # 遍历文件夹下的所有文件（不含子文件夹）
    for file in folder.iterdir():
        if file.is_file():
            # 获取文件扩展名（小写，避免大小写问题）
            ext = file.suffix.lower()
            target_folder = "Other"
            
            # 匹配文件类型
            for folder_name, ext_list in file_type_mapping.items():
                if ext in [e.lower() for e in ext_list]:
                    target_folder = folder_name
                    break
            
            # 创建目标文件夹（如果不存在）
            target_path = folder / target_folder
            target_path.mkdir(exist_ok=True)
            
            # 移动文件（避免重复，重复则加序号）
            new_file_path = target_path / file.name
            count = 1
            while new_file_path.exists():
                new_name = f"{file.stem}_{count}{ext}"
                new_file_path = target_path / new_name
                count += 1
            
            shutil.move(str(file), str(new_file_path))
            print(f"整理成功：{file.name} → {target_folder}/{new_file_path.name}")

def clean_empty_folders(folder_path: str) -> None:
    """
    递归删除空文件夹
    :param folder_path: 目标文件夹路径
    """
    if not os.path.exists(folder_path):
        print(f"错误：文件夹 {folder_path} 不存在！")
        return
    
    # 递归遍历子文件夹（从最深层开始删除）
    for root, dirs, files in os.walk(folder_path, topdown=False):
        for dir_name in dirs:
            dir_path = os.path.join(root, dir_name)
            # 判断文件夹是否为空
            if not os.listdir(dir_path):
                os.rmdir(dir_path)
                print(f"删除空文件夹：{dir_path}")

# ===================== 测试/使用示例 =====================
if __name__ == "__main__":
    # 请修改为你的目标文件夹路径（Windows 路径用 \\ 或 /，Mac/Linux 用 /）
    target_folder = r"C:\\Users\\xxx\\Downloads"
    
    # 示例1：批量重命名所有 Excel 文件
    # batch_rename_files(
    #     folder_path=target_folder,
    #     prefix="每日销售数据",
    #     suffix="（天猫）",
    #     start_num=1,
    #     file_ext=".xls"
    # )
    
    # 示例2：按文件类型自动整理
    # organize_files_by_type(target_folder)
    
    # 示例3：清理整理后产生的空文件夹
    # clean_empty_folders(target_folder)