基于Python开发一个交互式文件分类整理工具

import os
import shutil
from typing import Dict, Set  # 引入类型提示，提升代码可读性与维护性


def organize_files(directory: str) -> None:
    """
    整理指定目录下的文件，按扩展名分类归档到对应文件夹

    参数:
        directory: str - 需要整理的目标目录路径（绝对路径或相对路径均可）

    功能说明:
        1. 支持的文件类型及目标文件夹：
           - DOCX文件（.docx）、PDF文件（.pdf）、Excel文件（.xlsx/.xls）
           - Markdown文件（.md）、TXT文件（.txt）、Python文件（.py）
        2. 自动创建缺失的分类文件夹
        3. 重名文件自动添加编号（如"文档_1.docx"）
        4. 仅处理文件，跳过目录中的子文件夹
    """
    # 1. 定义文件扩展名与目标文件夹的映射规则（便于后续扩展新类型）
    EXTENSION_TO_FOLDER: Dict[str, str] = {
        'docx': 'DOCX文件',
        'pdf': 'PDF文件',
        'xlsx': 'Excel文件',
        'xls': 'Excel文件',  # 兼容旧版Excel格式
        'md': 'Markdown文件',
        'txt': 'TXT文件',
        'py': 'Python文件'
    }

    # 2. 预创建所有分类文件夹（去重处理，避免重复创建）
    target_folders: Set[str] = set(EXTENSION_TO_FOLDER.values())  # 去重后的目标文件夹
    for folder_name in target_folders:
        folder_path: str = os.path.join(directory, folder_name)
        # 仅当文件夹不存在时创建，避免覆盖已有目录
        if not os.path.exists(folder_path):
            os.makedirs(folder_path)
            print(f"✅ 创建分类文件夹: {os.path.abspath(folder_path)}")  # 输出绝对路径，更直观

    # 3. 遍历目录，处理所有文件
    print(f"\n📂 开始遍历目录: {os.path.abspath(directory)}")
    processed_count: int = 0  # 统计处理的文件数量
    skipped_count: int = 0    # 统计跳过的文件/文件夹数量

    for filename in os.listdir(directory):
        # 拼接文件的完整路径（处理相对路径/绝对路径统一问题）
        file_abs_path: str = os.path.abspath(os.path.join(directory, filename))

        # 跳过子文件夹（仅处理文件）
        if os.path.isdir(file_abs_path):
            skipped_count += 1
            continue

        # 提取文件扩展名（小写处理，避免大小写敏感问题，如.PDF与.pdf统一）
        file_extension: str = filename.split('.')[-1].lower() if '.' in filename else ''

        # 4. 匹配文件类型，确定目标文件夹
        if file_extension in EXTENSION_TO_FOLDER:
            target_folder_name: str = EXTENSION_TO_FOLDER[file_extension]
            target_folder_path: str = os.path.join(directory, target_folder_name)
            # 目标文件的完整路径（初始为原文件名）
            target_file_path: str = os.path.join(target_folder_path, filename)

            # 5. 处理重名文件：自动添加编号（如"报告_1.pdf"）
            counter: int = 1
            while os.path.exists(target_file_path):
                # 拆分文件名与扩展名（处理无扩展名的特殊情况）
                name_without_ext: str = os.path.splitext(filename)[0]
                ext: str = os.path.splitext(filename)[1]
                # 生成新文件名（编号递增）
                new_filename: str = f"{name_without_ext}_{counter}{ext}"
                target_file_path = os.path.join(target_folder_path, new_filename)
                counter += 1

            # 6. 移动文件到目标文件夹
            shutil.move(file_abs_path, target_file_path)
            processed_count += 1
            # 输出移动日志（简化路径，仅显示相对目标目录的位置）
            relative_source: str = os.path.relpath(file_abs_path, directory)
            relative_target: str = os.path.relpath(target_file_path, directory)
            print(f"🔄 移动文件: {relative_source} → {relative_target}")
        else:
            # 不支持的文件类型，跳过处理
            skipped_count += 1
            print(f"⏭️  跳过不支持的文件类型: {filename}（扩展名: {file_extension}）")

    # 7. 输出整理结果汇总
    print(f"\n📊 文件整理完成！")
    print(f"✅ 成功处理文件数量: {processed_count}")
    print(f"⏭️  跳过文件/文件夹数量: {skipped_count}")


def validate_directory(directory_path: str) -> bool:
    """
    校验目录是否有效（存在且为目录）

    参数:
        directory_path: str - 待校验的目录路径

    返回:
        bool - 有效返回True，无效返回False
    """
    if not os.path.exists(directory_path):
        print(f"❌ 错误：目录不存在 → {directory_path}")
        return False
    if not os.path.isdir(directory_path):
        print(f"❌ 错误：路径不是目录 → {directory_path}")
        return False
    return True


if __name__ == "__main__":
    # 1. 获取用户输入的目录路径（交互式引导，替换传统命令行参数）
    while True:
        target_dir: str = input("请输入需要整理的目录路径: ").strip()
        if not target_dir:
            print("ℹ️  未输入路径，将使用当前工作目录")
            target_dir = os.getcwd()
        # 校验目录有效性，有效则退出循环，无效则继续提示
        if validate_directory(target_dir):
            break
        print("❌ 目录无效，请重新输入。")

    # 2. 显示确认的目标目录（输出绝对路径，避免相对路径混淆）
    print(f"ℹ️  待整理目录: {os.path.abspath(target_dir)}")

    # 3. 执行文件整理（调用核心逻辑，保持流程连贯）
    print("\n" + "=" * 50)
    print("📋 开始执行文件分类整理")
    print("=" * 50)
    organize_files(target_dir)
    print("\n" + "=" * 50)
    print("🎉 所有整理操作已完成！")
    print("=" * 50)

请输入需要整理的目录路径: C:\Users\Test\Downloads
ℹ️  待整理目录: C:\Users\Test\Downloads

==================================================
📋 开始执行文件分类整理
==================================================
✅ 创建分类文件夹: C:\Users\Test\Downloads\DOCX文件
✅ 创建分类文件夹: C:\Users\Test\Downloads\PDF文件
✅ 创建分类文件夹: C:\Users\Test\Downloads\Excel文件
✅ 创建分类文件夹: C:\Users\Test\Downloads\Markdown文件
✅ 创建分类文件夹: C:\Users\Test\Downloads\TXT文件
✅ 创建分类文件夹: C:\Users\Test\Downloads\Python文件

📂 开始遍历目录: C:\Users\Test\Downloads
🔄 移动文件: 项目报告.docx → DOCX文件/项目报告.docx
🔄 移动文件: 数据统计.xlsx → Excel文件/数据统计.xlsx
🔄 移动文件: 学习笔记.md → Markdown文件/学习笔记.md
🔄 移动文件: 简历.pdf → PDF文件/简历.pdf
🔄 移动文件: 简历.pdf → PDF文件/简历_1.pdf  # 重名文件自动编号
⏭️  跳过不支持的文件类型: 截图.png（扩展名: png）
⏭️  跳过不支持的文件类型: 视频.mp4（扩展名: mp4）

📊 文件整理完成！
✅ 成功处理文件数量: 5
⏭️  跳过文件/文件夹数量: 3

==================================================
🎉 所有整理操作已完成！
==================================================

EXTENSION_TO_FOLDER: Dict[str, str] = {
    # 原有规则...
    'jpg': '图片文件',
    'png': '图片文件',
    'jpeg': '图片文件',
    'zip': '压缩文件',
    'rar': '压缩文件',
    '7z': '压缩文件'
}

EXTENSION_TO_FOLDER: Dict[str, str] = {
    # 原有规则...
    'xlsx': '表格文件',
    'xls': '表格文件'  # 统一文件夹名称
}

import time

# 在文件遍历循环中添加时间筛选
file_mtime = os.path.getmtime(file_abs_path)  # 获取文件最后修改时间（时间戳）
thirty_days_ago = time.time() - 30 * 24 * 3600  # 30天前的时间戳
if file_mtime < thirty_days_ago:
    skipped_count += 1
    print(f"⏭️  跳过过期文件: {filename}（最后修改时间超过30天）")
    continue