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

core/info_decoder.py

@@ -16,9 +16,19 @@ import re
 from typing import Optional, List, Dict, Tuple
 from pathlib import Path
 
-from models.message_definition import MessageDefinition, FieldDefinition, OneofDefinition
-from utils.logger import get_logger
-from utils.type_utils import type_mapper, naming_converter
+# 智能导入：同时支持相对导入（包环境）和绝对导入（开发环境）
+try:
+    # 相对导入（包环境）
+    from ..models.message_definition import MessageDefinition, FieldDefinition, OneofDefinition
+    from ..utils.logger import get_logger
+    from ..utils.type_utils import type_mapper, naming_converter
+    from ..parsing.java_parser import JavaParser
+except ImportError:
+    # 绝对导入（开发环境）
+    from models.message_definition import MessageDefinition, FieldDefinition, OneofDefinition
+    from utils.logger import get_logger
+    from utils.type_utils import type_mapper, naming_converter
+    from parsing.java_parser import JavaParser
 
 
 class InfoDecoder:
@@ -40,7 +50,6 @@ class InfoDecoder:
         self.java_source_analyzer = java_source_analyzer
         
         # 导入JavaParser
-        from parsing.java_parser import JavaParser
         self.java_parser = JavaParser()
         
         # Protobuf字段类型映射表

core/reconstructor.py

@@ -15,14 +15,31 @@ from pathlib import Path
 from collections import deque
 from typing import Set, Dict, List, Optional, Tuple
 
-from parsing.java_parser import JavaParser
-from parsing.enum_parser import EnumParser
-from core.info_decoder import InfoDecoder
-from generation.proto_generator import ProtoGenerator
-from models.message_definition import MessageDefinition, EnumDefinition, EnumValueDefinition
-from utils.logger import get_logger
-from utils.file_cache import get_file_cache
-from utils.type_utils import type_mapper, naming_converter
+# 智能导入：同时支持相对导入（包环境）和绝对导入（开发环境）
+try:
+    # 相对导入（包环境）
+    from ..parsing.java_parser import JavaParser
+    from ..parsing.enum_parser import EnumParser
+    from .info_decoder import InfoDecoder
+    from ..generation.proto_generator import ProtoGenerator
+    from ..models.message_definition import MessageDefinition, EnumDefinition, EnumValueDefinition
+    from ..utils.logger import get_logger
+    from ..utils.file_cache import get_file_cache
+    from ..utils.type_utils import type_mapper, naming_converter, TypeMapper, NamingConverter
+    from ..utils.builtin_proto import get_builtin_manager
+    from ..utils.type_index import get_type_index
+except ImportError:
+    # 绝对导入（开发环境）
+    from parsing.java_parser import JavaParser
+    from parsing.enum_parser import EnumParser
+    from core.info_decoder import InfoDecoder
+    from generation.proto_generator import ProtoGenerator
+    from models.message_definition import MessageDefinition, EnumDefinition, EnumValueDefinition
+    from utils.logger import get_logger
+    from utils.file_cache import get_file_cache
+    from utils.type_utils import type_mapper, naming_converter, TypeMapper, NamingConverter
+    from utils.builtin_proto import get_builtin_manager
+    from utils.type_index import get_type_index
 
 
 class JavaSourceAnalyzer:
@@ -359,7 +376,6 @@ class ProtoReconstructor:
         self.info_decoder.java_source_analyzer = self.java_source_analyzer
         
         # 初始化内置proto管理器
-        from utils.builtin_proto import get_builtin_manager
         self.builtin_manager = get_builtin_manager(output_dir=str(output_dir))
         
         # 🚀 性能优化：移除未使用的索引系统，简化代码
@@ -404,7 +420,6 @@ class ProtoReconstructor:
         self._generate_all_proto_files()
         
         # 5. 输出性能统计信息
-        from utils.file_cache import get_file_cache
         file_cache = get_file_cache()
         file_cache.print_stats()
         
@@ -1269,8 +1284,6 @@ class ProtoReconstructor:
             是否为基础字段类型
         """
         # 🚀 性能优化：使用缓存的类型检查器，避免重复计算
-        from utils.type_utils import TypeMapper
-        
         # 直接使用统一的基础类型检查，无需额外逻辑
         return TypeMapper.is_java_basic_type(type_name)
 
@@ -1353,8 +1366,6 @@ class ProtoReconstructor:
             实际的完整类型名
         """
         # 🚀 优化：使用统一的类型检查器
-        from utils.type_utils import TypeMapper
-        
         if TypeMapper.is_java_basic_type(inferred_type):
             self.logger.debug(f"    跳过基础类型: {inferred_type}")
             return None
@@ -1365,7 +1376,6 @@ class ProtoReconstructor:
         
         try:
             # 使用索引系统进行快速查找，避免文件IO
-            from utils.type_index import get_type_index
             type_index = get_type_index(self.sources_dir)
             
             # 构造可能的完整类名
@@ -1523,5 +1533,4 @@ class ProtoReconstructor:
             蛇形命名字符串
         """
         # 🚀 优化：使用统一的命名转换器，避免重复实现
-        from utils.type_utils import NamingConverter
         return NamingConverter.to_snake_case(camel_str) 

generation/__init__.py

@@ -1,3 +1,20 @@
+"""
+代码生成模块
+
+负责将解析得到的消息定义转换为Protobuf .proto文件
+"""
+
+# 智能导入：同时支持相对导入（包环境）和绝对导入（开发环境）
+try:
+    # 相对导入（包环境）
+    from .proto_generator import ProtoGenerator
+except ImportError:
+    # 绝对导入（开发环境）
+    from generation.proto_generator import ProtoGenerator
+
+__all__ = [
+    'ProtoGenerator'
+]
 
 
 

generation/proto_generator.py

@@ -10,8 +10,17 @@ Author: AI Assistant
 
 import re
 from typing import Dict, Set, List, Union
-from models.message_definition import MessageDefinition, FieldDefinition, EnumDefinition, EnumValueDefinition
-from utils.type_utils import type_mapper, naming_converter, field_name_processor
+
+# 智能导入：同时支持相对导入（包环境）和绝对导入（开发环境）
+try:
+    # 相对导入（包环境）
+    from ..models.message_definition import MessageDefinition, FieldDefinition, EnumDefinition, EnumValueDefinition
+    from ..utils.type_utils import type_mapper, naming_converter, field_name_processor
+    from ..utils.logger import get_logger
+except ImportError:
+    # 绝对导入（开发环境）
+    from models.message_definition import MessageDefinition, FieldDefinition, EnumDefinition, EnumValueDefinition
+    from utils.type_utils import type_mapper, naming_converter, field_name_processor
 
 
 class ProtoGenerator:
@@ -70,7 +79,6 @@ class ProtoGenerator:
             return '\n'.join(lines)
             
         except Exception as e:
-            from utils.logger import get_logger
             logger = get_logger("proto_generator")
             logger.error(f"❌ 生成proto文件失败 {message_def.name if message_def else 'Unknown'}: {e}")
             raise
@@ -118,7 +126,6 @@ class ProtoGenerator:
             return '\n'.join(lines)
             
         except Exception as e:
-            from utils.logger import get_logger
             logger = get_logger("proto_generator")
             logger.error(f"❌ 生成枚举proto文件失败 {enum_def.name if enum_def else 'Unknown'}: {e}")
             raise
@@ -290,7 +297,6 @@ class ProtoGenerator:
                 return f'  {field_type} {field.name} = {field.tag};'
                 
         except Exception as e:
-            from utils.logger import get_logger
             logger = get_logger("proto_generator")
             logger.error(f"❌ 生成字段定义失败 {field.name if field else 'Unknown'}: {e}")
             raise

main.py

@@ -16,17 +16,33 @@ Author: AI Assistant
 
 import sys
 import argparse
+import traceback
 from pathlib import Path
 
-# 导入项目模块
+# 确保项目根目录在Python路径中
 import os
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
-from core.reconstructor import ProtoReconstructor
-from utils.logger import setup_logger, get_logger
 
+# 智能导入：同时支持相对导入（包环境）和绝对导入（开发环境）
+try:
+    # 相对导入（包环境）
+    from .core.reconstructor import ProtoReconstructor
+    from .utils.logger import setup_logger, get_logger
+    from .utils.report_utils import print_results_summary
+except ImportError:
+    # 绝对导入（开发环境）
+    from core.reconstructor import ProtoReconstructor
+    from utils.logger import setup_logger, get_logger
+    from utils.report_utils import print_results_summary
 
-def parse_arguments():
-    """解析命令行参数"""
+
+def parse_arguments() -> argparse.Namespace:
+    """
+    解析命令行参数
+    
+    Returns:
+        解析后的命令行参数对象
+    """
     parser = argparse.ArgumentParser(
         description='从JADX反编译的Java源码重构Protobuf .proto文件',
         formatter_class=argparse.RawDescriptionHelpFormatter,
@@ -72,8 +88,19 @@ def parse_arguments():
     return parser.parse_args()
 
 
-def validate_arguments(args):
-    """验证命令行参数"""
+def validate_arguments(args: argparse.Namespace) -> tuple[Path, str, Path]:
+    """
+    验证命令行参数的有效性
+    
+    Args:
+        args: 解析后的命令行参数
+        
+    Returns:
+        验证后的路径元组: (sources_path, root_class, output_path)
+        
+    Raises:
+        SystemExit: 当参数无效时退出程序
+    """
     logger = get_logger("main")
     
     # 验证源码目录
@@ -92,13 +119,13 @@ def validate_arguments(args):
         logger.error("应该是完整的类名，如: com.example.Model")
         sys.exit(1)
     
-    # 输出目录可以不存在，会自动创建
+    # 验证输出目录
     output_path = Path(args.output_dir)
     if output_path.exists() and not output_path.is_dir():
         logger.error(f"输出路径存在但不是目录: {output_path}")
         sys.exit(1)
     
-    # 验证日志目录
+    # 创建日志目录
     log_path = Path(args.log_dir)
     try:
         log_path.mkdir(parents=True, exist_ok=True)
@@ -109,11 +136,19 @@ def validate_arguments(args):
     return sources_path.resolve(), args.root_class, output_path.resolve()
 
 
-def main():
-    """主函数"""
+def main() -> None:
+    """
+    主函数：协调整个重构过程
+    
+    处理流程：
+    1. 解析和验证命令行参数
+    2. 初始化日志系统
+    3. 创建重构器并执行重构
+    4. 输出结果统计信息
+    """
     args = None
     try:
-        # 解析参数
+        # 解析和验证参数
         args = parse_arguments()
         
         # 初始化日志系统
@@ -130,62 +165,25 @@ def main():
         logger.info(f"📁 日志目录: {args.log_dir}")
         logger.info(f"🎯 根类: {root_class}")
         
-        # 创建重构器并执行
+        # 创建重构器并执行重构
         reconstructor = ProtoReconstructor(sources_dir, output_dir)
         reconstructor._verbose = args.verbose  # 传递verbose标志
         results = reconstructor.reconstruct_from_root(root_class)
         
-        # 输出详细的结果统计
-        if results:
-            # 统计成功和失败的数量
-            success_count = len(results)
-            failed_count = len(reconstructor.failed_classes) if hasattr(reconstructor, 'failed_classes') else 0
-            total_attempted = success_count + failed_count
-            
-            logger.success("✅ 重构完成!")
-            logger.info(f"📊 处理统计: 共尝试处理 {total_attempted} 个类型")
-            
-            message_count = sum(1 for r in results.values() if hasattr(r, 'fields'))
-            enum_count = sum(1 for r in results.values() if hasattr(r, 'values'))
-            
-            logger.info(f"   - ✅ 成功: {success_count} 个 (消息: {message_count}, 枚举: {enum_count})")
-            
-            # 显示失败的类
-            if hasattr(reconstructor, 'failed_classes') and reconstructor.failed_classes:
-                logger.warning(f"   - ❌ 失败: {failed_count} 个")
-                for failed_class, reason in reconstructor.failed_classes.items():
-                    logger.warning(f"     • {failed_class}: {reason}")
-            
-            # 显示跳过的类
-            if hasattr(reconstructor, 'skipped_classes') and reconstructor.skipped_classes:
-                skipped_count = len(reconstructor.skipped_classes)
-                logger.info(f"   - ⏭️  跳过: {skipped_count} 个 (基础类型或已处理)")
-                if args.verbose:
-                    for skipped_class, reason in reconstructor.skipped_classes.items():
-                        logger.info(f"     • {skipped_class}: {reason}")
-        else:
-            logger.error("❌ 没有生成任何proto文件!")
-            logger.error("请检查:")
-            logger.error("  1. 根类名是否正确")
-            logger.error("  2. Java源码目录是否包含对应的文件")
-            logger.error("  3. 类是否为protobuf消息类")
-            
-            # 显示详细的失败信息
-            if hasattr(reconstructor, 'failed_classes') and reconstructor.failed_classes:
-                logger.error("失败的类:")
-                for failed_class, reason in reconstructor.failed_classes.items():
-                    logger.error(f"  • {failed_class}: {reason}")
-            
-            sys.exit(1)
+        # 输出结果统计
+        print_results_summary(reconstructor, results, logger, args.verbose)
         
     except KeyboardInterrupt:
+        # 处理用户中断
         if args:
             logger = get_logger("main")
             logger.warning("⚠️  操作被用户中断")
         else:
             print("\n⚠️  操作被用户中断")
         sys.exit(1)
+        
     except Exception as e:
+        # 处理其他异常
         if args:
             logger = get_logger("main")
             logger.error(f"❌ 重构失败: {e}")
@@ -194,7 +192,6 @@ def main():
         else:
             print(f"\n❌ 重构失败: {e}")
             if hasattr(args, 'verbose') and args.verbose:
-                import traceback
                 traceback.print_exc()
         sys.exit(1)
 

models/__init__.py

@@ -1,27 +1,34 @@
-from dataclasses import dataclass, field
-from typing import List, Dict, Set
+"""
+模型定义模块
 
-@dataclass
-class FieldDefinition:
-    """表示一个 Protobuf 消息中的字段。"""
-    name: str
-    type: str
-    tag: int
-    rule: str # "optional", "repeated", or "oneof"
+包含Protobuf消息、字段、枚举等数据结构的定义
+所有具体的类定义都在相应的子模块中
+"""
 
-@dataclass
-class OneofDefinition:
-    """表示一个 Protobuf oneof 块。"""
-    name: str
-    fields: List[FieldDefinition] = field(default_factory=list)
+# 智能导入：同时支持相对导入（包环境）和绝对导入（开发环境）
+try:
+    # 相对导入（包环境）
+    from .message_definition import (
+        MessageDefinition, 
+        FieldDefinition, 
+        OneofDefinition, 
+        EnumDefinition, 
+        EnumValueDefinition
+    )
+except ImportError:
+    # 绝对导入（开发环境）
+    from models.message_definition import (
+        MessageDefinition, 
+        FieldDefinition, 
+        OneofDefinition, 
+        EnumDefinition, 
+        EnumValueDefinition
+    )
 
-@dataclass
-class MessageDefinition:
-    """表示一个完整的 Protobuf 消息的定义。"""
-    name: str
-    package: str
-    info_string: str = ""
-    objects: List[str] = field(default_factory=list)
-    fields: List[FieldDefinition] = field(default_factory=list)
-    oneofs: Dict[str, OneofDefinition] = field(default_factory=dict)
-    dependencies: Set[str] = field(default_factory=set) 
+__all__ = [
+    'MessageDefinition',
+    'FieldDefinition', 
+    'OneofDefinition',
+    'EnumDefinition',
+    'EnumValueDefinition'
+] 

parsing/__init__.py

@@ -1,3 +1,25 @@
+"""
+解析器模块
+
+包含各种源码解析器：
+- Java解析器：解析Java源码，提取字段标签和类型信息
+- 枚举解析器：专门处理Java枚举类的解析
+"""
+
+# 智能导入：同时支持相对导入（包环境）和绝对导入（开发环境）
+try:
+    # 相对导入（包环境）
+    from .java_parser import JavaParser
+    from .enum_parser import EnumParser
+except ImportError:
+    # 绝对导入（开发环境）
+    from parsing.java_parser import JavaParser
+    from parsing.enum_parser import EnumParser
+
+__all__ = [
+    'JavaParser',
+    'EnumParser'
+]
 
 
 

parsing/enum_parser.py

@@ -8,8 +8,16 @@
 import re
 import os
 from typing import List, Optional, Dict, Tuple
-from models.message_definition import EnumDefinition, EnumValueDefinition
-from utils.logger import get_logger
+
+# 智能导入：同时支持相对导入（包环境）和绝对导入（开发环境）
+try:
+    # 相对导入（包环境）
+    from ..models.message_definition import EnumDefinition, EnumValueDefinition
+    from ..utils.logger import get_logger
+except ImportError:
+    # 绝对导入（开发环境）
+    from models.message_definition import EnumDefinition, EnumValueDefinition
+    from utils.logger import get_logger
 
 
 class EnumParser:

parsing/java_parser.py

@@ -11,7 +11,13 @@ import re
 from pathlib import Path
 from typing import Optional, Tuple, List
 
-from utils.logger import get_logger
+# 智能导入：同时支持相对导入（包环境）和绝对导入（开发环境）
+try:
+    # 相对导入（包环境）
+    from ..utils.logger import get_logger
+except ImportError:
+    # 绝对导入（开发环境）
+    from utils.logger import get_logger
 
 
 class JavaParser:

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "reproto"
-version = "0.1.2"
+version = "0.1.3"
 description = "一个强大的逆向工程工具, 能够从任何使用Google Protobuf Lite的Android应用中自动重构出完整的.proto文件结构."
 readme = "README.md"
 license = "Proprietary"