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


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

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

    功能說(shuō)明:
        1. 支持的文件類型及目標(biāo)文件夾：
           - DOCX文件（.docx）、PDF文件（.pdf）、Excel文件（.xlsx/.xls）
           - Markdown文件（.md）、TXT文件（.txt）、Python文件（.py）
        2. 自動(dòng)創(chuàng)建缺失的分類文件夾
        3. 重名文件自動(dòng)添加編號(hào)（如"文檔_1.docx"）
        4. 僅處理文件，跳過(guò)目錄中的子文件夾
    """
    # 1. 定義文件擴(kuò)展名與目標(biāo)文件夾的映射規(guī)則（便于后續(xù)擴(kuò)展新類型）
    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())  # 去重后的目標(biāo)文件夾
    for folder_name in target_folders:
        folder_path: str = os.path.join(directory, folder_name)
        # 僅當(dāng)文件夾不存在時(shí)創(chuàng)建，避免覆蓋已有目錄
        if not os.path.exists(folder_path):
            os.makedirs(folder_path)
            print(f"? 創(chuàng)建分類文件夾: {os.path.abspath(folder_path)}")  # 輸出絕對(duì)路徑，更直觀

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

    for filename in os.listdir(directory):
        # 拼接文件的完整路徑（處理相對(duì)路徑/絕對(duì)路徑統(tǒng)一問(wèn)題）
        file_abs_path: str = os.path.abspath(os.path.join(directory, filename))

        # 跳過(guò)子文件夾（僅處理文件）
        if os.path.isdir(file_abs_path):
            skipped_count += 1
            continue

        # 提取文件擴(kuò)展名（小寫(xiě)處理，避免大小寫(xiě)敏感問(wèn)題，如.PDF與.pdf統(tǒng)一）
        file_extension: str = filename.split('.')[-1].lower() if '.' in filename else ''

        # 4. 匹配文件類型，確定目標(biāo)文件夾
        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)
            # 目標(biāo)文件的完整路徑（初始為原文件名）
            target_file_path: str = os.path.join(target_folder_path, filename)

            # 5. 處理重名文件：自動(dòng)添加編號(hào)（如"報(bào)告_1.pdf"）
            counter: int = 1
            while os.path.exists(target_file_path):
                # 拆分文件名與擴(kuò)展名（處理無(wú)擴(kuò)展名的特殊情況）
                name_without_ext: str = os.path.splitext(filename)[0]
                ext: str = os.path.splitext(filename)[1]
                # 生成新文件名（編號(hào)遞增）
                new_filename: str = f"{name_without_ext}_{counter}{ext}"
                target_file_path = os.path.join(target_folder_path, new_filename)
                counter += 1

            # 6. 移動(dòng)文件到目標(biāo)文件夾
            shutil.move(file_abs_path, target_file_path)
            processed_count += 1
            # 輸出移動(dòng)日志（簡(jiǎn)化路徑，僅顯示相對(duì)目標(biāo)目錄的位置）
            relative_source: str = os.path.relpath(file_abs_path, directory)
            relative_target: str = os.path.relpath(target_file_path, directory)
            print(f"?? 移動(dòng)文件: {relative_source} → {relative_target}")
        else:
            # 不支持的文件類型，跳過(guò)處理
            skipped_count += 1
            print(f"??  跳過(guò)不支持的文件類型: {filename}（擴(kuò)展名: {file_extension}）")

    # 7. 輸出整理結(jié)果匯總
    print(f"\n?? 文件整理完成！")
    print(f"? 成功處理文件數(shù)量: {processed_count}")
    print(f"??  跳過(guò)文件/文件夾數(shù)量: {skipped_count}")


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

    參數(shù):
        directory_path: str - 待校驗(yàn)的目錄路徑

    返回:
        bool - 有效返回True，無(wú)效返回False
    """
    if not os.path.exists(directory_path):
        print(f"? 錯(cuò)誤：目錄不存在 → {directory_path}")
        return False
    if not os.path.isdir(directory_path):
        print(f"? 錯(cuò)誤：路徑不是目錄 → {directory_path}")
        return False
    return True


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

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

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

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

==================================================
?? 開(kāi)始執(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文件

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

?? 文件整理完成！
? 成功處理文件數(shù)量: 5
??  跳過(guò)文件/文件夾數(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)中添加時(shí)間篩選
file_mtime = os.path.getmtime(file_abs_path)  # 獲取文件最后修改時(shí)間（時(shí)間戳）
thirty_days_ago = time.time() - 30 * 24 * 3600  # 30天前的時(shí)間戳
if file_mtime < thirty_days_ago:
    skipped_count += 1
    print(f"??  跳過(guò)過(guò)期文件: {filename}（最后修改時(shí)間超過(guò)30天）")
    continue