import os
import sys
import argparse
from datetime import datetime

class FileBatchRenamer:
    def __init__(self, directory):
        self.directory = directory
        self.rename_log = []
        
    def check_directory(self):
        """檢查目錄是否存在"""
        if not os.path.exists(self.directory):
            raise FileNotFoundError(f"目錄 '{self.directory}' 不存在")
        if not os.path.isdir(self.directory):
            raise NotADirectoryError(f"'{self.directory}' 不是一個(gè)有效的目錄")
            
    def get_files(self, extension=None, pattern=None):
        """獲取符合條件的文件列表"""
        try:
            self.check_directory()
            files = os.listdir(self.directory)
            filtered_files = []
            
            for file in files:
                file_path = os.path.join(self.directory, file)
                # 只處理文件，不處理子目錄
                if os.path.isfile(file_path):
                    # 根據(jù)擴(kuò)展名過(guò)濾
                    if extension and not file.endswith(extension):
                        continue
                    # 根據(jù)模式過(guò)濾
                    if pattern and pattern not in file:
                        continue
                    filtered_files.append(file)
                    
            return sorted(filtered_files)
        except Exception as e:
            print(f"獲取文件列表時(shí)出錯(cuò): {e}")
            return []
            
    def preview_rename(self, files, naming_pattern, start_number=1):
        """預(yù)覽重命名結(jié)果"""
        print("\n重命名預(yù)覽:")
        print("-" * 50)
        rename_map = {}
        
        for i, filename in enumerate(files, start_number):
            name, ext = os.path.splitext(filename)
            new_name = naming_pattern.format(i) + ext
            rename_map[filename] = new_name
            print(f"{filename} -> {new_name}")
            
        return rename_map
        
    def execute_rename(self, rename_map):
        """執(zhí)行重命名操作"""
        success_count = 0
        error_count = 0
        
        for old_name, new_name in rename_map.items():
            old_path = os.path.join(self.directory, old_name)
            new_path = os.path.join(self.directory, new_name)
            
            # 檢查目標(biāo)文件是否已存在
            if os.path.exists(new_path):
                print(f"警告: 文件 '{new_name}' 已存在，跳過(guò) '{old_name}'")
                error_count += 1
                continue
                
            try:
                os.rename(old_path, new_path)
                self.rename_log.append((datetime.now(), old_name, new_name))
                print(f"成功重命名: {old_name} -> {new_name}")
                success_count += 1
            except PermissionError:
                print(f"權(quán)限錯(cuò)誤: 無(wú)法重命名 '{old_name}'，請(qǐng)檢查文件權(quán)限")
                error_count += 1
            except Exception as e:
                print(f"重命名 '{old_name}' 時(shí)出錯(cuò): {e}")
                error_count += 1
                
        print(f"\n重命名完成! 成功: {success_count}, 失敗: {error_count}")
        
    def save_log(self, log_file="rename_log.txt"):
        """保存操作日志"""
        try:
            with open(os.path.join(self.directory, log_file), "w", encoding="utf-8") as f:
                f.write("文件批量重命名操作日志\n")
                f.write("=" * 50 + "\n")
                f.write(f"操作時(shí)間: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n")
                f.write(f"操作目錄: {self.directory}\n\n")
                
                for timestamp, old_name, new_name in self.rename_log:
                    f.write(f"[{timestamp.strftime('%H:%M:%S')}] {old_name} -> {new_name}\n")
                    
            print(f"操作日志已保存到: {log_file}")
        except Exception as e:
            print(f"保存日志時(shí)出錯(cuò): {e}")

def main():
    parser = argparse.ArgumentParser(description="文件批量重命名工具")
    parser.add_argument("directory", help="要處理的目錄路徑")
    parser.add_argument("-p", "--pattern", default="file_{:03d}", 
                       help="命名模式，使用 {:03d} 表示序號(hào) (默認(rèn): file_{:03d})")
    parser.add_argument("-s", "--start", type=int, default=1, 
                       help="起始序號(hào) (默認(rèn): 1)")
    parser.add_argument("-e", "--extension", 
                       help="文件擴(kuò)展名過(guò)濾 (例如: .jpg)")
    parser.add_argument("--preview", action="store_true", 
                       help="僅預(yù)覽，不執(zhí)行實(shí)際重命名")
    
    args = parser.parse_args()
    
    try:
        renamer = FileBatchRenamer(args.directory)
        files = renamer.get_files(extension=args.extension)
        
        if not files:
            print("未找到符合條件的文件")
            return
            
        print(f"找到 {len(files)} 個(gè)文件")
        
        rename_map = renamer.preview_rename(files, args.pattern, args.start)
        
        if args.preview:
            print("\n預(yù)覽模式，未執(zhí)行實(shí)際重命名操作")
            return
            
        confirm = input("\n確認(rèn)執(zhí)行重命名操作? (y/N): ")
        if confirm.lower() == 'y':
            renamer.execute_rename(rename_map)
            renamer.save_log()
        else:
            print("操作已取消")
            
    except Exception as e:
        print(f"程序執(zhí)行出錯(cuò): {e}")
        sys.exit(1)

if __name__ == "__main__":
    main()

# 基本用法，對(duì)目錄中的所有文件按默認(rèn)模式重命名
python batch_renamer.py /path/to/directory

# 指定命名模式
python batch_renamer.py /path/to/directory -p "photo_{:04d}"

# 指定起始序號(hào)
python batch_renamer.py /path/to/directory -s 100

# 只處理特定擴(kuò)展名的文件
python batch_renamer.py /path/to/directory -e .jpg

# 預(yù)覽模式（不實(shí)際執(zhí)行重命名）
python batch_renamer.py /path/to/directory --preview

python batch_renamer.py ./test_dir -p "pic_{:02d}" -e .jpg