import os


def get_directory():
    """獲取用戶輸入的有效目錄路徑"""
    while True:
        dir_path = input("請輸入要處理的目錄路徑：").strip()
        if os.path.isdir(dir_path):
            return os.path.normpath(dir_path)
        print("× 目錄不存在，請重新輸入")

import os

def list_files(dir_path):
    """列出目錄中的所有文件并返回列表"""
    files = [f for f in sorted(os.listdir(dir_path))
             if os.path.isfile(os.path.join(dir_path, f))]

    print("\n目錄文件列表：")
    for idx, f in enumerate(files, 1):
        print(f"{idx:02d}. {f}")
    return files

def get_rename_mode():
    """獲取用戶選擇的命名模式"""
    print("\n請選擇重命名模式：")
    print("1. 添加前綴\t2. 添加后綴")
    print("3. 替換字符串\t4. 序列號")
    while True:
        mode = input("請輸入選項 (1-4)：").strip()
        if mode in {'1', '2', '3', '4'}:
            return mode
        print("× 無效輸入，請重新選擇")

def get_sequence_params():
    """獲取序列號模式參數(shù)"""
    print("\n序列號模式參數(shù)設(shè)置：")
    try:
        start = int(input("起始數(shù)字 (默認1)：") or 1)
        digits = int(input("位數(shù) (默認3)：") or 3)
        sep = input("分隔符 (默認_)：") or "_"
        return start, digits, sep
    except ValueError:
        print("! 輸入無效，使用默認設(shè)置：起始=1，位數(shù)=3，分隔符=_")
        return 1, 3, "_"

import os


def get_directory():
    """獲取用戶輸入的有效目錄路徑"""
    while True:
        dir_path = input("請輸入要處理的目錄路徑：").strip()
        if os.path.isdir(dir_path):
            return os.path.normpath(dir_path)
        print("× 目錄不存在，請重新輸入")


def list_files(dir_path):
    """列出目錄中的所有文件并返回列表"""
    files = [f for f in sorted(os.listdir(dir_path))
             if os.path.isfile(os.path.join(dir_path, f))]

    print("\n目錄文件列表：")
    for idx, f in enumerate(files, 1):
        print(f"{idx:02d}. {f}")
    return files


def get_rename_mode():
    """獲取用戶選擇的命名模式"""
    print("\n請選擇重命名模式：")
    print("1. 添加前綴\t2. 添加后綴")
    print("3. 替換字符串\t4. 序列號")
    while True:
        mode = input("請輸入選項 (1-4)：").strip()
        if mode in {'1', '2', '3', '4'}:
            return mode
        print("× 無效輸入，請重新選擇")


def get_sequence_params():
    """獲取序列號模式參數(shù)"""
    print("\n序列號模式參數(shù)設(shè)置：")
    try:
        start = int(input("起始數(shù)字 (默認1)：") or 1)
        digits = int(input("位數(shù) (默認3)：") or 3)
        sep = input("分隔符 (默認_)：") or "_"
        return start, digits, sep
    except ValueError:
        print("! 輸入無效，使用默認設(shè)置：起始=1，位數(shù)=3，分隔符=_")
        return 1, 3, "_"


def main():
    # 獲取目錄路徑
    dir_path = get_directory()

    # 獲取文件列表
    files = list_files(dir_path)
    if not files:
        print("! 該目錄沒有可操作文件")
        return

    # 確認操作
    if input("\n是否繼續(xù)操作？(y/n) ").lower() != 'y':
        print("操作已取消")
        return

    # 獲取重命名模式
    mode = get_rename_mode()

    # 獲取模式參數(shù)
    params = {}
    if mode == '1':
        params['prefix'] = input("\n請輸入要添加的前綴：")
    elif mode == '2':
        params['suffix'] = input("\n請輸入要添加的后綴：")
    elif mode == '3':
        params['old'] = input("\n請輸入要替換的字符串：")
        params['new'] = input("請輸入替換后的新字符串：")
    elif mode == '4':
        params['start'], params['digits'], params['sep'] = get_sequence_params()

    # 二次確認
    if input("\n!! 確認執(zhí)行重命名操作？(y/n) ").lower() != 'y':
        print("操作已取消")
        return

    # 執(zhí)行重命名
    counter = 0
    current_num = params.get('start', 0)

    for filename in files:
        # 跳過隱藏文件（可選）
        if filename.startswith('.'):
            continue

        # 分割文件名和擴展名
        file_path = os.path.join(dir_path, filename)
        base, ext = os.path.splitext(filename)
        new_base = base

        # 根據(jù)模式生成新文件名
        if mode == '1':
            new_base = f"{params['prefix']}{base}"
        elif mode == '2':
            new_base = f"{base}{params['suffix']}"
        elif mode == '3':
            new_base = base.replace(params['old'], params['new'])
        elif mode == '4':
            num_format = f"{current_num:0{params['digits']}}d"
            new_base = f"{num_format}{params['sep']}{base}"
            current_num += 1

        new_name = f"{new_base}{ext}"
        new_path = os.path.join(dir_path, new_name)

        # 跳過未修改的情況
        if new_name == filename:
            continue

        # 執(zhí)行重命名
        try:
            if not os.path.exists(new_path):
                os.rename(file_path, new_path)
                print(f"√ {filename} → {new_name}")
                counter += 1
            else:
                print(f"! 沖突：{new_name} 已存在")
        except Exception as e:
            print(f"× 錯誤：重命名 {filename} 失敗 - {str(e)}")

    print(f"\n操作完成，成功重命名 {counter}/{len(files)} 個文件")


if __name__ == "__main__":
    print("=== 文件批量重命名工具 ===")
    main()