Undoom Uninstaller MCP

"""工具函数模块""" import os import shutil from typing import Optional, Tuple, List from datetime import datetime def format_size(size: int) -> str: """格式化文件大小 Args: size: 文件大小（字节） Returns: 格式化后的大小字符串 """ if size == 0: return "N/A" units = ['B', 'KB', 'MB', 'GB', 'TB'] for unit in units: if size < 1024.0: return f"{size:.1f} {unit}" size /= 1024.0 return f"{size:.1f} TB" def get_directory_size(path: str) -> int: """获取目录大小（优化版本，避免长时间阻塞） Args: path: 目录路径 Returns: 目录总大小（字节） """ if not path or not os.path.isdir(path): return 0 total_size = 0 try: # 限制扫描深度和文件数量，避免长时间阻塞 file_count = 0 max_files = 500 # 最多扫描500个文件 max_depth = 2 # 最大深度2层 for root, dirs, files in os.walk(path): # 计算当前深度 depth = root[len(path):].count(os.sep) if depth >= max_depth: dirs[:] = [] # 不再深入子目录 continue for filename in files: if file_count >= max_files: return total_size # 达到文件数量限制，返回当前大小 filepath = os.path.join(root, filename) try: total_size += os.path.getsize(filepath) file_count += 1 except (OSError, IOError): continue except (OSError, IOError, KeyboardInterrupt): pass return total_size def safe_remove_directory(path: str) -> Tuple[bool, str]: """安全删除目录 Args: path: 目录路径 Returns: (成功标志, 消息) """ if not path or not os.path.exists(path): return True, "路径不存在" try: if os.path.isdir(path): shutil.rmtree(path) return True, f"成功删除目录: {path}" else: os.remove(path) return True, f"成功删除文件: {path}" except Exception as e: return False, f"删除失败 {path}: {str(e)}" def format_install_date(date_str: str) -> str: """格式化安装日期 Args: date_str: 原始日期字符串（YYYYMMDD格式） Returns: 格式化后的日期字符串（YYYY-MM-DD格式） """ if not date_str or len(date_str) != 8: return "未知" try: return f"{date_str[:4]}-{date_str[4:6]}-{date_str[6:8]}" except (ValueError, IndexError): return "未知" def get_drive_letter(path: str) -> str: """获取路径的盘符 Args: path: 文件路径 Returns: 盘符字符串 """ if not path: return "未知" if len(path) >= 2 and path[1] == ':': return path[0].upper() + ":" elif path.startswith("\\\\"): # UNC路径 return "网络路径" else: return "未知" def escape_markdown(text: str) -> str: """转义Markdown特殊字符 Args: text: 原始文本 Returns: 转义后的文本 """ if not text: return "" return text.replace("|", "\\|") def truncate_text(text: str, max_length: int) -> str: """截断文本 Args: text: 原始文本 max_length: 最大长度 Returns: 截断后的文本 """ if not text or len(text) <= max_length: return text return text[:max_length-3] + "..." def get_common_residue_paths(program_name: str) -> List[str]: """获取程序常见残留路径 Args: program_name: 程序名称 Returns: 可能的残留路径列表 """ paths = [] env_vars = ["APPDATA", "LOCALAPPDATA", "PROGRAMDATA", "USERPROFILE"] for env_var in env_vars: base_path = os.environ.get(env_var, "") if base_path: if env_var == "USERPROFILE": # 用户目录下的特殊路径 paths.extend([ os.path.join(base_path, "AppData", "Local", program_name), os.path.join(base_path, "AppData", "Roaming", program_name) ]) else: paths.append(os.path.join(base_path, program_name)) return paths