import os
import time
from typing import List, Set

class TempFileCleaner:
    """專業(yè)的臨時文件清理工具"""
    
    # 常見臨時文件擴展名集合
    TEMP_EXTENSIONS = {
        '.tmp', '.temp', '.~', '.bak', 
        '.old', '.log', '.cache', '.dmp'
    }
    
    def __init__(self, root_dir: str, exclude_dirs: List[str] = None):
        """
        初始化清理工具
        
        :param root_dir: 要清理的根目錄
        :param exclude_dirs: 要排除的目錄列表
        """
        self.root_dir = os.path.abspath(root_dir)
        self.exclude_dirs = set(os.path.abspath(d) for d in (exclude_dirs or []))
        self.deleted_files = []
        self.failed_deletions = []
        self.total_bytes = 0
    
    def is_temp_file(self, filename: str) -> bool:
        """判斷文件是否為臨時文件"""
        lower_name = filename.lower()
        return (lower_name.endswith(tuple(self.TEMP_EXTENSIONS)) or
                lower_name.startswith('~$') or
                lower_name.startswith('temp_'))
    
    def should_exclude(self, filepath: str) -> bool:
        """檢查文件是否在排除目錄中"""
        for excluded in self.exclude_dirs:
            if filepath.startswith(excluded):
                return True
        return False
    
    def clean_directory(self, dir_path: str):
        """清理指定目錄中的臨時文件"""
        try:
            for entry in os.listdir(dir_path):
                full_path = os.path.join(dir_path, entry)
                
                if self.should_exclude(full_path):
                    continue
                
                if os.path.isdir(full_path):
                    self.clean_directory(full_path)
                elif os.path.isfile(full_path) and self.is_temp_file(entry):
                    self._attempt_file_deletion(full_path)
        except PermissionError:
            self.failed_deletions.append(f"權限不足: {dir_path}")
        except Exception as e:
            self.failed_deletions.append(f"錯誤處理 {dir_path}: {str(e)}")
    
    def _attempt_file_deletion(self, filepath: str):
        """嘗試刪除文件并記錄結(jié)果"""
        try:
            file_size = os.path.getsize(filepath)
            os.remove(filepath)
            self.deleted_files.append(filepath)
            self.total_bytes += file_size
        except Exception as e:
            self.failed_deletions.append(f"刪除失敗 {filepath}: {str(e)}")
    
    def run_cleanup(self) -> dict:
        """執(zhí)行清理操作并返回結(jié)果報告"""
        start_time = time.time()
        self.clean_directory(self.root_dir)
        
        return {
            'root_directory': self.root_dir,
            'total_deleted': len(self.deleted_files),
            'total_freed': self._format_bytes(self.total_bytes),
            'failed_attempts': len(self.failed_deletions),
            'execution_time': f"{time.time() - start_time:.2f}秒",
            'deleted_files': self.deleted_files,
            'failed_deletions': self.failed_deletions
        }
    
    @staticmethod
    def _format_bytes(size: int) -> str:
        """格式化字節(jié)大小為易讀字符串"""
        for unit in ['B', 'KB', 'MB', 'GB']:
            if size < 1024.0:
                return f"{size:.2f} {unit}"
            size /= 1024.0
        return f"{size:.2f} TB"

# 使用示例
if __name__ == "__main__":
    # 配置要清理的目錄和排除目錄
    cleaner = TempFileCleaner(
        root_dir="C:/Projects",
        exclude_dirs=["C:/Projects/ImportantDocs", "C:/Projects/node_modules"]
    )
    
    # 執(zhí)行清理并獲取報告
    report = cleaner.run_cleanup()
    
    # 打印摘要報告
    print("\n=== 清理報告 ===")
    print(f"根目錄: {report['root_directory']}")
    print(f"刪除文件數(shù): {report['total_deleted']}")
    print(f"釋放空間: {report['total_freed']}")
    print(f"失敗操作: {report['failed_attempts']}")
    print(f"執(zhí)行時間: {report['execution_time']}")

class TempFileCleaner:
    TEMP_EXTENSIONS = {'.tmp', '.temp', '.~', '.bak', '.old', '.log', '.cache', '.dmp'}
    
    def __init__(self, root_dir: str, exclude_dirs: List[str] = None):
        self.root_dir = os.path.abspath(root_dir)
        self.exclude_dirs = set(os.path.abspath(d) for d in (exclude_dirs or []))

def is_temp_file(self, filename: str) -> bool:
    lower_name = filename.lower()
    return (lower_name.endswith(tuple(self.TEMP_EXTENSIONS)) or
            lower_name.startswith('~$') or
            lower_name.startswith('temp_'))

def should_exclude(self, filepath: str) -> bool:
    for excluded in self.exclude_dirs:
        if filepath.startswith(excluded):
            return True
    return False

def clean_directory(self, dir_path: str):
    try:
        for entry in os.listdir(dir_path):
            full_path = os.path.join(dir_path, entry)
            
            if self.should_exclude(full_path):
                continue
            
            if os.path.isdir(full_path):
                self.clean_directory(full_path)
            elif os.path.isfile(full_path) and self.is_temp_file(entry):
                self._attempt_file_deletion(full_path)

def _attempt_file_deletion(self, filepath: str):
    try:
        file_size = os.path.getsize(filepath)
        os.remove(filepath)
        self.deleted_files.append(filepath)
        self.total_bytes += file_size
    except Exception as e:
        self.failed_deletions.append(f"刪除失敗 {filepath}: {str(e)}")

def run_cleanup(self) -> dict:
    start_time = time.time()
    self.clean_directory(self.root_dir)
    
    return {
        'root_directory': self.root_dir,
        'total_deleted': len(self.deleted_files),
        'total_freed': self._format_bytes(self.total_bytes),
        'failed_attempts': len(self.failed_deletions),
        'execution_time': f"{time.time() - start_time:.2f}秒",
        'deleted_files': self.deleted_files,
        'failed_deletions': self.failed_deletions
    }

@staticmethod
def _format_bytes(size: int) -> str:
    for unit in ['B', 'KB', 'MB', 'GB']:
        if size < 1024.0:
            return f"{size:.2f} {unit}"
        size /= 1024.0
    return f"{size:.2f} TB"

@classmethod
def from_config(cls, config_path: str):
    with open(config_path) as f:
        config = json.load(f)
    return cls(
        root_dir=config['root_dir'],
        exclude_dirs=config.get('exclude_dirs', [])
    )

import logging

logging.basicConfig(
    filename='cleaner.log',
    level=logging.INFO,
    format='%(asctime)s - %(levelname)s - %(message)s'
)

# 在刪除操作中添加日志
logging.info(f"已刪除: {filepath} ({self._format_bytes(file_size)})")

from concurrent.futures import ThreadPoolExecutor

def clean_directory(self, dir_path: str):
    with ThreadPoolExecutor(max_workers=4) as executor:
        for entry in os.listdir(dir_path):
            full_path = os.path.join(dir_path, entry)
            if self.should_exclude(full_path):
                continue
            if os.path.isdir(full_path):
                executor.submit(self.clean_directory, full_path)
            elif self.is_temp_file(entry):
                self._attempt_file_deletion(full_path)

def __init__(self, root_dir: str, exclude_dirs: List[str] = None, dry_run: bool = False):
    self.dry_run = dry_run  # 新增參數(shù)

def _attempt_file_deletion(self, filepath: str):
    try:
        file_size = os.path.getsize(filepath)
        if not self.dry_run:  # 安全模式檢查
            os.remove(filepath)
        self.deleted_files.append(filepath)
        self.total_bytes += file_size
    except Exception as e:
        self.failed_deletions.append(f"刪除失敗 {filepath}: {str(e)}")

cp file_to_delete /backup/file_to_delete.bak && rm file_to_delete

lsof /path/to/file  # 查看文件是否被進程打開