auto-coder 0.1.398__py3-none-any.whl → 0.1.400__py3-none-any.whl

autocoder/sdk/utils/formatters.py

@@ -0,0 +1,216 @@
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+"""
+Auto-Coder SDK 格式化工具
+
+提供各种输出格式化功能。
+"""
+
+import json
+from typing import Any, AsyncIterator, Dict
+
+from ..models.messages import Message
+from ..models.responses import StreamEvent
+
+
+def format_output(content: Any, output_format: str = "text") -> str:
+    """
+    格式化输出内容
+    
+    Args:
+        content: 要格式化的内容
+        output_format: 输出格式 (text, json)
+        
+    Returns:
+        str: 格式化后的内容
+    """
+    if output_format == "json":
+        if isinstance(content, (dict, list)):
+            return json.dumps(content, ensure_ascii=False, indent=2)
+        elif hasattr(content, 'to_dict'):
+            return json.dumps(content.to_dict(), ensure_ascii=False, indent=2)
+        else:
+            return json.dumps({"content": str(content)}, ensure_ascii=False, indent=2)
+    else:
+        # text format
+        return str(content)
+
+
+async def format_stream_output(stream: AsyncIterator[Any], output_format: str = "stream-json") -> AsyncIterator[str]:
+    """
+    格式化流式输出
+    
+    Args:
+        stream: 输入流
+        output_format: 输出格式
+        
+    Yields:
+        str: 格式化后的内容
+    """
+    if output_format == "stream-json":
+        # 发送开始事件
+        start_event = StreamEvent.start_event()
+        yield start_event.to_json()
+        
+        try:
+            async for item in stream:
+                if isinstance(item, Message):
+                    # 转换Message为StreamEvent
+                    if item.is_assistant_message():
+                        content_event = StreamEvent.content_event(item.content)
+                        yield content_event.to_json()
+                elif hasattr(item, 'to_dict'):
+                    # 直接输出有to_dict方法的对象
+                    yield json.dumps(item.to_dict(), ensure_ascii=False)
+                else:
+                    # 包装普通内容为content事件
+                    content_event = StreamEvent.content_event(str(item))
+                    yield content_event.to_json()
+            
+            # 发送结束事件
+            end_event = StreamEvent.end_event()
+            yield end_event.to_json()
+            
+        except Exception as e:
+            # 发送错误事件
+            error_event = StreamEvent.error_event(str(e))
+            yield error_event.to_json()
+    else:
+        # 其他格式，直接输出
+        async for item in stream:
+            yield format_output(item, "text")
+
+
+def format_table_output(data: list, headers: list = None) -> str:
+    """
+    格式化表格输出
+    
+    Args:
+        data: 表格数据
+        headers: 表头
+        
+    Returns:
+        str: 格式化的表格
+    """
+    if not data:
+        return "No data to display"
+    
+    if headers is None:
+        if isinstance(data[0], dict):
+            headers = list(data[0].keys())
+        else:
+            headers = [f"Column {i+1}" for i in range(len(data[0]) if data[0] else 0)]
+    
+    # 计算列宽
+    col_widths = []
+    for i, header in enumerate(headers):
+        max_width = len(header)
+        for row in data:
+            if isinstance(row, dict):
+                cell_value = str(row.get(header, ""))
+            elif isinstance(row, (list, tuple)) and i < len(row):
+                cell_value = str(row[i])
+            else:
+                cell_value = ""
+            max_width = max(max_width, len(cell_value))
+        col_widths.append(max_width + 2)  # 添加padding
+    
+    # 构建表格
+    lines = []
+    
+    # 表头
+    header_line = "|".join(header.center(col_widths[i]) for i, header in enumerate(headers))
+    lines.append(header_line)
+    
+    # 分隔线
+    separator = "|".join("-" * width for width in col_widths)
+    lines.append(separator)
+    
+    # 数据行
+    for row in data:
+        if isinstance(row, dict):
+            row_values = [str(row.get(header, "")) for header in headers]
+        elif isinstance(row, (list, tuple)):
+            row_values = [str(row[i]) if i < len(row) else "" for i in range(len(headers))]
+        else:
+            row_values = [str(row)]
+        
+        row_line = "|".join(value.ljust(col_widths[i]) for i, value in enumerate(row_values))
+        lines.append(row_line)
+    
+    return "\n".join(lines)
+
+
+def format_error_output(error: Exception, verbose: bool = False) -> str:
+    """
+    格式化错误输出
+    
+    Args:
+        error: 异常对象
+        verbose: 是否显示详细信息
+        
+    Returns:
+        str: 格式化的错误信息
+    """
+    if verbose:
+        import traceback
+        return f"Error: {str(error)}\n\nTraceback:\n{traceback.format_exc()}"
+    else:
+        return f"Error: {str(error)}"
+
+
+def format_progress_output(current: int, total: int, description: str = "") -> str:
+    """
+    格式化进度输出
+    
+    Args:
+        current: 当前进度
+        total: 总数
+        description: 描述
+        
+    Returns:
+        str: 格式化的进度信息
+    """
+    if total == 0:
+        percentage = 100
+    else:
+        percentage = min(100, int((current / total) * 100))
+    
+    bar_length = 30
+    filled_length = int(bar_length * percentage // 100)
+    bar = "█" * filled_length + "-" * (bar_length - filled_length)
+    
+    return f"{description} [{bar}] {percentage}% ({current}/{total})"
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+

autocoder/sdk/utils/io_utils.py

@@ -0,0 +1,302 @@
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+"""
+Auto-Coder SDK IO工具
+
+提供输入输出相关的工具函数。
+"""
+
+import sys
+import os
+from pathlib import Path
+from typing import Optional
+
+
+def read_stdin() -> str:
+    """
+    从标准输入读取内容
+    
+    Returns:
+        str: 读取的内容
+    """
+    try:
+        if sys.stdin.isatty():
+            # 如果是交互式终端，返回空字符串
+            return ""
+        else:
+            # 从管道或重定向读取
+            return sys.stdin.read().strip()
+    except Exception:
+        return ""
+
+
+def write_stdout(content: str) -> None:
+    """
+    写入标准输出
+    
+    Args:
+        content: 要写入的内容
+    """
+    try:
+        sys.stdout.write(content)
+        sys.stdout.flush()
+    except Exception:
+        pass
+
+
+def write_stderr(content: str) -> None:
+    """
+    写入标准错误输出
+    
+    Args:
+        content: 要写入的内容
+    """
+    try:
+        sys.stderr.write(content)
+        sys.stderr.flush()
+    except Exception:
+        pass
+
+
+def ensure_directory(dir_path: str) -> bool:
+    """
+    确保目录存在，如果不存在则创建
+    
+    Args:
+        dir_path: 目录路径
+        
+    Returns:
+        bool: 是否成功创建或已存在
+    """
+    try:
+        Path(dir_path).mkdir(parents=True, exist_ok=True)
+        return True
+    except Exception:
+        return False
+
+
+def read_file_safe(file_path: str, encoding: str = "utf-8") -> Optional[str]:
+    """
+    安全读取文件内容
+    
+    Args:
+        file_path: 文件路径
+        encoding: 编码格式
+        
+    Returns:
+        Optional[str]: 文件内容，如果读取失败返回None
+    """
+    try:
+        with open(file_path, 'r', encoding=encoding) as f:
+            return f.read()
+    except Exception:
+        return None
+
+
+def write_file_safe(file_path: str, content: str, encoding: str = "utf-8") -> bool:
+    """
+    安全写入文件内容
+    
+    Args:
+        file_path: 文件路径
+        content: 要写入的内容
+        encoding: 编码格式
+        
+    Returns:
+        bool: 是否写入成功
+    """
+    try:
+        # 确保目录存在
+        ensure_directory(os.path.dirname(file_path))
+        
+        with open(file_path, 'w', encoding=encoding) as f:
+            f.write(content)
+        return True
+    except Exception:
+        return False
+
+
+def append_file_safe(file_path: str, content: str, encoding: str = "utf-8") -> bool:
+    """
+    安全追加文件内容
+    
+    Args:
+        file_path: 文件路径
+        content: 要追加的内容
+        encoding: 编码格式
+        
+    Returns:
+        bool: 是否追加成功
+    """
+    try:
+        # 确保目录存在
+        ensure_directory(os.path.dirname(file_path))
+        
+        with open(file_path, 'a', encoding=encoding) as f:
+            f.write(content)
+        return True
+    except Exception:
+        return False
+
+
+def file_exists(file_path: str) -> bool:
+    """
+    检查文件是否存在
+    
+    Args:
+        file_path: 文件路径
+        
+    Returns:
+        bool: 文件是否存在
+    """
+    return Path(file_path).exists() and Path(file_path).is_file()
+
+
+def directory_exists(dir_path: str) -> bool:
+    """
+    检查目录是否存在
+    
+    Args:
+        dir_path: 目录路径
+        
+    Returns:
+        bool: 目录是否存在
+    """
+    return Path(dir_path).exists() and Path(dir_path).is_dir()
+
+
+def get_file_size(file_path: str) -> Optional[int]:
+    """
+    获取文件大小
+    
+    Args:
+        file_path: 文件路径
+        
+    Returns:
+        Optional[int]: 文件大小（字节），如果文件不存在返回None
+    """
+    try:
+        return Path(file_path).stat().st_size
+    except Exception:
+        return None
+
+
+def list_files_in_directory(dir_path: str, pattern: str = "*") -> list:
+    """
+    列出目录中的文件
+    
+    Args:
+        dir_path: 目录路径
+        pattern: 文件名模式
+        
+    Returns:
+        list: 文件路径列表
+    """
+    try:
+        dir_path_obj = Path(dir_path)
+        if not dir_path_obj.exists() or not dir_path_obj.is_dir():
+            return []
+        
+        return [str(path) for path in dir_path_obj.glob(pattern) if path.is_file()]
+    except Exception:
+        return []
+
+
+def create_temp_file(content: str = "", suffix: str = ".tmp") -> Optional[str]:
+    """
+    创建临时文件
+    
+    Args:
+        content: 文件内容
+        suffix: 文件后缀
+        
+    Returns:
+        Optional[str]: 临时文件路径，如果创建失败返回None
+    """
+    try:
+        import tempfile
+        
+        with tempfile.NamedTemporaryFile(mode='w', suffix=suffix, delete=False) as f:
+            f.write(content)
+            return f.name
+    except Exception:
+        return None
+
+
+def delete_file_safe(file_path: str) -> bool:
+    """
+    安全删除文件
+    
+    Args:
+        file_path: 文件路径
+        
+    Returns:
+        bool: 是否删除成功
+    """
+    try:
+        Path(file_path).unlink()
+        return True
+    except Exception:
+        return False
+
+
+def copy_file_safe(src_path: str, dst_path: str) -> bool:
+    """
+    安全复制文件
+    
+    Args:
+        src_path: 源文件路径
+        dst_path: 目标文件路径
+        
+    Returns:
+        bool: 是否复制成功
+    """
+    try:
+        import shutil
+        
+        # 确保目标目录存在
+        ensure_directory(os.path.dirname(dst_path))
+        
+        shutil.copy2(src_path, dst_path)
+        return True
+    except Exception:
+        return False
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+