reproto 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

utils/builtin_proto.py

@@ -0,0 +1,262 @@
+"""
+Google Protobuf 内置类型管理模块
+处理内置proto文件的检测、拷贝和导入语句生成
+"""
+
+import os
+import shutil
+from pathlib import Path
+from typing import Dict, Set, Optional
+from utils.logger import logger
+
+
+def find_include_directory() -> Optional[Path]:
+    """
+    查找include目录，支持开发环境和安装包环境
+    
+    Returns:
+        include目录的路径，如果找不到则返回None
+    """
+    # 尝试多个可能的位置
+    possible_locations = [
+        # 1. 相对于当前文件的位置（开发环境）
+        Path(__file__).parent.parent / "include",
+        
+        # 2. 相对于包安装位置（安装包环境）
+        Path(__file__).parent.parent.parent / "include",
+        
+        # 3. 在site-packages中的位置
+        Path(__file__).parent / "include",
+        
+        # 4. 使用pkg_resources查找（如果可用）
+    ]
+    
+    # 尝试使用pkg_resources（推荐方式）
+    try:
+        import pkg_resources
+        try:
+            include_path = Path(pkg_resources.resource_filename('reproto', 'include'))
+            if include_path.exists():
+                possible_locations.insert(0, include_path)
+        except (pkg_resources.DistributionNotFound, FileNotFoundError):
+            pass
+    except ImportError:
+        pass
+    
+    # 尝试使用importlib.resources（Python 3.9+推荐方式）
+    try:
+        import importlib.resources as resources
+        try:
+            # 对于Python 3.9+
+            if hasattr(resources, 'files'):
+                include_ref = resources.files('reproto') / 'include'
+                if include_ref.is_dir():
+                    possible_locations.insert(0, Path(str(include_ref)))
+        except Exception:
+            pass
+    except ImportError:
+        pass
+    
+    # 按优先级检查每个位置
+    for location in possible_locations:
+        if location.exists() and location.is_dir():
+            # 验证这确实是包含proto文件的include目录
+            wrappers_file = location / "google" / "protobuf" / "wrappers.proto"
+            if wrappers_file.exists():
+                logger.debug(f"找到include目录: {location}")
+                return location
+    
+    logger.error("无法找到include目录")
+    return None
+
+
+class BuiltinProtoManager:
+    """Google Protobuf 内置proto文件管理器"""
+    
+    # 内置类型到proto文件的映射
+    _BUILTIN_TYPE_TO_FILE = {
+        # Wrapper types - 都在 wrappers.proto 中
+        'google.protobuf.BoolValue': 'google/protobuf/wrappers.proto',
+        'google.protobuf.Int32Value': 'google/protobuf/wrappers.proto',
+        'google.protobuf.Int64Value': 'google/protobuf/wrappers.proto',
+        'google.protobuf.UInt32Value': 'google/protobuf/wrappers.proto',
+        'google.protobuf.UInt64Value': 'google/protobuf/wrappers.proto',
+        'google.protobuf.FloatValue': 'google/protobuf/wrappers.proto',
+        'google.protobuf.DoubleValue': 'google/protobuf/wrappers.proto',
+        'google.protobuf.StringValue': 'google/protobuf/wrappers.proto',
+        'google.protobuf.BytesValue': 'google/protobuf/wrappers.proto',
+        
+        # Well-known types
+        'google.protobuf.Any': 'google/protobuf/any.proto',
+        'google.protobuf.Timestamp': 'google/protobuf/timestamp.proto',
+        'google.protobuf.Duration': 'google/protobuf/duration.proto',
+        'google.protobuf.Empty': 'google/protobuf/empty.proto',
+        'google.protobuf.Struct': 'google/protobuf/struct.proto',
+        'google.protobuf.Value': 'google/protobuf/struct.proto',
+        'google.protobuf.ListValue': 'google/protobuf/struct.proto',
+        'google.protobuf.NullValue': 'google/protobuf/struct.proto',
+        'google.protobuf.FieldMask': 'google/protobuf/field_mask.proto',
+        
+        # API types
+        'google.protobuf.Api': 'google/protobuf/api.proto',
+        'google.protobuf.Method': 'google/protobuf/api.proto',
+        'google.protobuf.Mixin': 'google/protobuf/api.proto',
+        
+        # Type definitions
+        'google.protobuf.Type': 'google/protobuf/type.proto',
+        'google.protobuf.Field': 'google/protobuf/type.proto',
+        'google.protobuf.Enum': 'google/protobuf/type.proto',
+        'google.protobuf.EnumValue': 'google/protobuf/type.proto',
+        'google.protobuf.Option': 'google/protobuf/type.proto',
+        'google.protobuf.Syntax': 'google/protobuf/type.proto',
+        
+        # Source context
+        'google.protobuf.SourceContext': 'google/protobuf/source_context.proto',
+    }
+    
+    def __init__(self, include_dir: str, output_dir: str):
+        """
+        初始化内置proto管理器
+        
+        Args:
+            include_dir: include目录路径
+            output_dir: 输出目录路径
+        """
+        self.include_dir = Path(include_dir)
+        self.output_dir = Path(output_dir)
+        self._copied_files: Set[str] = set()
+    
+    def is_builtin_type(self, proto_type: str) -> bool:
+        """
+        检查是否为内置类型
+        
+        Args:
+            proto_type: proto类型名称
+            
+        Returns:
+            是否为内置类型
+        """
+        return proto_type in self._BUILTIN_TYPE_TO_FILE
+    
+    def get_import_path(self, proto_type: str) -> Optional[str]:
+        """
+        获取内置类型的导入路径
+        
+        Args:
+            proto_type: proto类型名称
+            
+        Returns:
+            导入路径，如果不是内置类型则返回None
+        """
+        return self._BUILTIN_TYPE_TO_FILE.get(proto_type)
+    
+    def ensure_builtin_proto_file(self, proto_type: str) -> bool:
+        """
+        确保内置proto文件存在于输出目录中
+        
+        Args:
+            proto_type: proto类型名称
+            
+        Returns:
+            是否成功处理
+        """
+        if not self.is_builtin_type(proto_type):
+            return False
+        
+        import_path = self.get_import_path(proto_type)
+        if not import_path:
+            return False
+        
+        # 如果已经拷贝过，直接返回
+        if import_path in self._copied_files:
+            return True
+        
+        # 源文件路径
+        source_file = self.include_dir / import_path
+        if not source_file.exists():
+            logger.error(f"内置proto文件不存在: {source_file}")
+            return False
+        
+        # 目标文件路径
+        target_file = self.output_dir / import_path
+        
+        # 创建目标目录
+        target_file.parent.mkdir(parents=True, exist_ok=True)
+        
+        try:
+            # 拷贝文件
+            shutil.copy2(source_file, target_file)
+            self._copied_files.add(import_path)
+            logger.info(f"📄 拷贝内置proto文件: {import_path}")
+            return True
+        except Exception as e:
+            logger.error(f"拷贝内置proto文件失败: {source_file} -> {target_file}, 错误: {e}")
+            return False
+    
+    def process_dependencies(self, dependencies: Set[str]) -> Dict[str, str]:
+        """
+        处理依赖列表，返回需要的导入语句映射
+        
+        Args:
+            dependencies: 依赖类型集合
+            
+        Returns:
+            类型名到导入路径的映射
+        """
+        imports = {}
+        
+        for dep_type in dependencies:
+            if self.is_builtin_type(dep_type):
+                import_path = self.get_import_path(dep_type)
+                if import_path and self.ensure_builtin_proto_file(dep_type):
+                    imports[dep_type] = import_path
+        
+        return imports
+    
+    def get_copied_files(self) -> Set[str]:
+        """
+        获取已拷贝的文件列表
+        
+        Returns:
+            已拷贝文件路径的集合
+        """
+        return self._copied_files.copy()
+
+
+# 全局实例（在需要时初始化）
+_builtin_manager: Optional[BuiltinProtoManager] = None
+
+
+def get_builtin_manager(include_dir: str = None, output_dir: str = None) -> BuiltinProtoManager:
+    """
+    获取全局内置proto管理器实例
+    
+    Args:
+        include_dir: include目录路径（可选，如果不提供会自动查找）
+        output_dir: 输出目录路径（首次调用时必需）
+        
+    Returns:
+        BuiltinProtoManager实例
+    """
+    global _builtin_manager
+    
+    if _builtin_manager is None:
+        if output_dir is None:
+            raise ValueError("首次调用get_builtin_manager时必须提供output_dir")
+        
+        # 如果没有提供include_dir，自动查找
+        if include_dir is None:
+            include_path = find_include_directory()
+            if include_path is None:
+                raise ValueError("无法找到include目录，请手动指定include_dir参数")
+            include_dir = str(include_path)
+        
+        _builtin_manager = BuiltinProtoManager(include_dir, output_dir)
+    
+    return _builtin_manager
+
+
+def reset_builtin_manager():
+    """重置全局内置proto管理器（主要用于测试）"""
+    global _builtin_manager
+    _builtin_manager = None 

utils/type_utils.py

@@ -169,9 +169,17 @@ class TypeMapper:
         """获取类型所需的导入语句"""
         imports = []
         if cls.is_well_known_type(proto_type):
-            # 提取具体的well-known type名称
-            type_name = proto_type.split('.')[-1].lower()
-            imports.append(f"google/protobuf/{type_name}.proto")
+            # 使用内置proto管理器获取正确的导入路径
+            try:
+                from utils.builtin_proto import get_builtin_manager
+                builtin_manager = get_builtin_manager()
+                import_path = builtin_manager.get_import_path(proto_type)
+                if import_path:
+                    imports.append(import_path)
+            except (ImportError, ValueError):
+                # 如果内置管理器不可用，使用旧逻辑作为后备
+                type_name = proto_type.split('.')[-1].lower()
+                imports.append(f"google/protobuf/{type_name}.proto")
         return imports
     
     @classmethod
@@ -227,7 +235,7 @@ class NamingConverter:
     @staticmethod
     def to_snake_case(camel_str: str) -> str:
         """
-        将CamelCase转换为snake_case
+        将CamelCase转换为snake_case，同时处理$符号
         
         Args:
             camel_str: 驼峰命名字符串
@@ -238,11 +246,18 @@ class NamingConverter:
         if not camel_str:
             return camel_str
             
+        # 首先处理$符号：将$替换为_，处理内部类和匿名类
+        s0 = camel_str.replace('$', '_')
+        
         # 处理连续大写字母：XMLParser -> XML_Parser
-        s1 = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', camel_str)
+        s1 = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', s0)
         # 处理小写字母后跟大写字母：userId -> user_Id
         s2 = re.sub('([a-z0-9])([A-Z])', r'\1_\2', s1)
-        return s2.lower()
+        
+        # 清理多余的下划线：将连续的下划线替换为单个下划线
+        s3 = re.sub('_+', '_', s2)
+        
+        return s3.lower()
     
     @staticmethod
     def to_pascal_case(snake_str: str) -> str:
@@ -281,6 +296,24 @@ class NamingConverter:
             
         # 第一个单词保持小写，其余单词首字母大写
         return components[0].lower() + ''.join(word.capitalize() for word in components[1:] if word)
+    
+    @staticmethod
+    def clean_proto_name(name: str) -> str:
+        """
+        清理proto名称中的$符号，用于消息和枚举名称
+        
+        Args:
+            name: 原始名称（可能包含$符号）
+            
+        Returns:
+            清理后的名称
+        """
+        if not name:
+            return name
+            
+        # 将$替换为空字符串，这样Models$Device变成ModelsDevice
+        # 这符合proto命名规范，避免语法错误
+        return name.replace('$', '')
 
 
 class FieldNameProcessor: