import os
import shutil
from typing import Dict, Set  # 引入類型提示，提升代碼可讀性與維護性


def organize_files(directory: str) -> None:
    """
    整理指定目錄下的文件，按擴展名分類歸檔到對應(yīng)文件夾

    參數(shù):
        directory: str - 需要整理的目標目錄路徑（絕對路徑或相對路徑均可）

    功能說明:
        1. 支持的文件類型及目標文件夾：
           - DOCX文件（.docx）、PDF文件（.pdf）、Excel文件（.xlsx/.xls）
           - Markdown文件（.md）、TXT文件（.txt）、Python文件（.py）
        2. 自動創(chuàng)建缺失的分類文件夾
        3. 重名文件自動添加編號（如"文檔_1.docx"）
        4. 僅處理文件，跳過目錄中的子文件夾
    """
    # 1. 定義文件擴展名與目標文件夾的映射規(guī)則（便于后續(xù)擴展新類型）
    EXTENSION_TO_FOLDER: Dict[str, str] = {
        'docx': 'DOCX文件',
        'pdf': 'PDF文件',
        'xlsx': 'Excel文件',
        'xls': 'Excel文件',  # 兼容舊版Excel格式
        'md': 'Markdown文件',
        'txt': 'TXT文件',
        'py': 'Python文件'
    }

    # 2. 預(yù)創(chuàng)建所有分類文件夾（去重處理，避免重復(fù)創(chuàng)建）
    target_folders: Set[str] = set(EXTENSION_TO_FOLDER.values())  # 去重后的目標文件夾
    for folder_name in target_folders:
        folder_path: str = os.path.join(directory, folder_name)
        # 僅當文件夾不存在時創(chuàng)建，避免覆蓋已有目錄
        if not os.path.exists(folder_path):
            os.makedirs(folder_path)
            print(f"? 創(chuàng)建分類文件夾: {os.path.abspath(folder_path)}")  # 輸出絕對路徑，更直觀

    # 3. 遍歷目錄，處理所有文件
    print(f"\n?? 開始遍歷目錄: {os.path.abspath(directory)}")
    processed_count: int = 0  # 統(tǒng)計處理的文件數(shù)量
    skipped_count: int = 0    # 統(tǒng)計跳過的文件/文件夾數(shù)量

    for filename in os.listdir(directory):
        # 拼接文件的完整路徑（處理相對路徑/絕對路徑統(tǒng)一問題）
        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統(tǒng)一）
        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. 輸出整理結(jié)果匯總
    print(f"\n?? 文件整理完成！")
    print(f"? 成功處理文件數(shù)量: {processed_count}")
    print(f"??  跳過文件/文件夾數(shù)量: {skipped_count}")


def validate_directory(directory_path: str) -> bool:
    """
    校驗?zāi)夸浭欠裼行Вù嬖谇覟槟夸洠?

    參數(shù):
        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. 獲取用戶輸入的目錄路徑（交互式引導(dǎo)，替換傳統(tǒng)命令行參數(shù)）
    while True:
        target_dir: str = input("請輸入需要整理的目錄路徑: ").strip()
        if not target_dir:
            print("??  未輸入路徑，將使用當前工作目錄")
            target_dir = os.getcwd()
        # 校驗?zāi)夸浻行?，有效則退出循環(huán)，無效則繼續(xù)提示
        if validate_directory(target_dir):
            break
        print("? 目錄無效，請重新輸入。")

    # 2. 顯示確認的目標目錄（輸出絕對路徑，避免相對路徑混淆）
    print(f"??  待整理目錄: {os.path.abspath(target_dir)}")

    # 3. 執(zhí)行文件整理（調(diào)用核心邏輯，保持流程連貫）
    print("\n" + "=" * 50)
    print("?? 開始執(zhí)行文件分類整理")
    print("=" * 50)
    organize_files(target_dir)
    print("\n" + "=" * 50)
    print("?? 所有整理操作已完成！")
    print("=" * 50)

請輸入需要整理的目錄路徑: C:\Users\Test\Downloads
??  待整理目錄: C:\Users\Test\Downloads

==================================================
?? 開始執(zhí)行文件分類整理
==================================================
? 創(chuàng)建分類文件夾: C:\Users\Test\Downloads\DOCX文件
? 創(chuàng)建分類文件夾: C:\Users\Test\Downloads\PDF文件
? 創(chuàng)建分類文件夾: C:\Users\Test\Downloads\Excel文件
? 創(chuàng)建分類文件夾: C:\Users\Test\Downloads\Markdown文件
? 創(chuàng)建分類文件夾: C:\Users\Test\Downloads\TXT文件
? 創(chuàng)建分類文件夾: C:\Users\Test\Downloads\Python文件

?? 開始遍歷目錄: C:\Users\Test\Downloads
?? 移動文件: 項目報告.docx → DOCX文件/項目報告.docx
?? 移動文件: 數(shù)據(jù)統(tǒng)計.xlsx → Excel文件/數(shù)據(jù)統(tǒng)計.xlsx
?? 移動文件: 學(xué)習(xí)筆記.md → Markdown文件/學(xué)習(xí)筆記.md
?? 移動文件: 簡歷.pdf → PDF文件/簡歷.pdf
?? 移動文件: 簡歷.pdf → PDF文件/簡歷_1.pdf  # 重名文件自動編號
??  跳過不支持的文件類型: 截圖.png（擴展名: png）
??  跳過不支持的文件類型: 視頻.mp4（擴展名: mp4）

?? 文件整理完成！
? 成功處理文件數(shù)量: 5
??  跳過文件/文件夾數(shù)量: 3

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

EXTENSION_TO_FOLDER: Dict[str, str] = {
    # 原有規(guī)則...
    'jpg': '圖片文件',
    'png': '圖片文件',
    'jpeg': '圖片文件',
    'zip': '壓縮文件',
    'rar': '壓縮文件',
    '7z': '壓縮文件'
}

EXTENSION_TO_FOLDER: Dict[str, str] = {
    # 原有規(guī)則...
    'xlsx': '表格文件',
    'xls': '表格文件'  # 統(tǒng)一文件夾名稱
}

import time

# 在文件遍歷循環(huán)中添加時間篩選
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