pytest-dsl 0.15.3__py3-none-any.whl → 0.15.5__py3-none-any.whl

pytest_dsl/__init__.py

@@ -12,7 +12,7 @@ pytest-dsl - 基于pytest的DSL测试框架
 - 自定义关键字支持
 """
 
-__version__ = "0.1.0"
+__version__ = "0.15.5"
 
 # 核心执行器
 from pytest_dsl.core.dsl_executor import DSLExecutor
@@ -47,6 +47,20 @@ from pytest_dsl.core.variable_utils import VariableReplacer
 # 自定义关键字管理器
 from pytest_dsl.core.custom_keyword_manager import custom_keyword_manager
 
+# 远程服务器注册器
+from pytest_dsl.core.remote_server_registry import (
+    remote_server_registry, RemoteServerRegistry,
+    register_remote_server_with_variables,
+    create_database_variable_provider,
+    create_config_file_variable_provider
+)
+
+# 远程服务器配置加载器
+from pytest_dsl.core.yaml_loader import (
+    load_remote_servers_from_yaml,
+    register_remote_servers_from_config
+)
+
 # 关键字加载器
 from pytest_dsl.core.keyword_loader import (
     keyword_loader, KeywordLoader,
@@ -61,6 +75,22 @@ from pytest_dsl.core.keyword_utils import (
     generate_html_report
 )
 
+# 远程关键字功能
+try:
+    from pytest_dsl.remote import (
+        remote_keyword_manager, RemoteKeywordManager, RemoteKeywordClient,
+        register_remote_server, register_multiple_servers
+    )
+    _REMOTE_AVAILABLE = True
+except ImportError:
+    # 如果远程功能依赖不可用，设置为None
+    remote_keyword_manager = None
+    RemoteKeywordManager = None
+    RemoteKeywordClient = None
+    register_remote_server = None
+    register_multiple_servers = None
+    _REMOTE_AVAILABLE = False
+
 # 便捷导入的别名
 Executor = DSLExecutor
 Validator = DSLValidator
@@ -106,6 +136,17 @@ __all__ = [
     'Node', 'get_parser', 'get_lexer',
     'TestContext', 'global_context',
     'VariableReplacer',
+
+    # 远程关键字功能（如果可用）
+    'remote_keyword_manager', 'RemoteKeywordManager', 'RemoteKeywordClient',
+    'register_remote_server', 'register_multiple_servers',
+
+    # 远程服务器注册器
+    'remote_server_registry', 'RemoteServerRegistry',
+    'register_remote_server_with_variables',
+    'create_database_variable_provider', 
+    'create_config_file_variable_provider',
+    'load_remote_servers_from_yaml', 'register_remote_servers_from_config',
 ]
 
 # 快捷函数
@@ -137,7 +178,8 @@ def parse_dsl(content: str) -> Node:
     return parser.parse(content, lexer=lexer)
 
 
-def execute_dsl(content: str, context: dict = None, enable_hooks: bool = True) -> any:
+def execute_dsl(content: str, context: dict = None,
+                enable_hooks: bool = True) -> any:
     """执行DSL内容的便捷函数
 
     Args:
@@ -158,7 +200,8 @@ def execute_dsl(content: str, context: dict = None, enable_hooks: bool = True) -
     return executor.execute(ast)
 
 
-def register_keyword(name: str, parameters: list = None, source_type: str = "external",
+def register_keyword(name: str, parameters: list = None,
+                     source_type: str = "external",
                      source_name: str = "user_defined"):
     """注册关键字的装饰器
 
@@ -198,6 +241,12 @@ def check_version_compatibility():
         pass
 
 
+# 远程功能检查
+def is_remote_available() -> bool:
+    """检查远程功能是否可用"""
+    return _REMOTE_AVAILABLE
+
+
 # 初始化时进行版本检查
 check_version_compatibility()
 

pytest_dsl/cli.py

@@ -165,11 +165,18 @@ def load_yaml_variables(args):
         environment = (os.environ.get('PYTEST_DSL_ENVIRONMENT') or
                        os.environ.get('ENVIRONMENT'))
 
+        # 智能判断是否应该加载默认配置
+        # 如果用户指定了YAML文件或目录，则不自动加载默认配置
+        user_specified_files = bool(args.yaml_vars)
+        user_specified_dir = bool(args.yaml_vars_dir)
+        auto_load_default = not (user_specified_files or user_specified_dir)
+
         load_yaml_variables_from_args(
             yaml_files=args.yaml_vars,
             yaml_vars_dir=args.yaml_vars_dir,
             project_root=os.getcwd(),  # CLI模式下使用当前工作目录作为项目根目录
-            environment=environment
+            environment=environment,
+            auto_load_default=auto_load_default  # 使用智能判断的结果
         )
     except Exception as e:
         print(f"加载YAML变量失败: {str(e)}")

pytest_dsl/core/context.py

@@ -1,18 +1,42 @@
 class TestContext:
     def __init__(self):
         self._data = {}
+        self._external_providers = []  # 外部变量提供者列表
 
     def set(self, key: str, value: any) -> None:
         """设置上下文变量"""
         self._data[key] = value
 
     def get(self, key: str, default=None) -> any:
-        """获取上下文变量，如果不存在返回默认值"""
-        return self._data.get(key, default)
+        """获取上下文变量，遵循变量优先级：本地变量 > 外部提供者变量"""
+        # 1. 首先检查本地变量
+        if key in self._data:
+            return self._data[key]
+
+        # 2. 检查外部提供者（按注册顺序）
+        for provider in self._external_providers:
+            if hasattr(provider, 'get_variable'):
+                value = provider.get_variable(key)
+                if value is not None:
+                    return value
+
+        # 3. 返回默认值
+        return default
 
     def has(self, key: str) -> bool:
-        """检查上下文变量是否存在"""
-        return key in self._data
+        """检查上下文变量是否存在（包括外部提供者）"""
+        # 检查本地变量
+        if key in self._data:
+            return True
+
+        # 检查外部提供者
+        for provider in self._external_providers:
+            if hasattr(provider, 'get_variable'):
+                value = provider.get_variable(key)
+                if value is not None:
+                    return True
+
+        return False
 
     def clear(self) -> None:
         """清空上下文"""
@@ -20,4 +44,57 @@ class TestContext:
 
     def get_local_variables(self) -> dict:
         """获取所有本地变量"""
-        return self._data 
+        return self._data
+
+    def get_all_context_variables(self) -> dict:
+        """获取所有上下文变量，包括本地变量和外部提供者变量
+        
+        Returns:
+            包含所有上下文变量的字典，本地变量优先级高于外部变量
+        """
+        all_variables = {}
+        
+        # 1. 先添加外部提供者的变量
+        for provider in self._external_providers:
+            if hasattr(provider, 'get_all_variables'):
+                try:
+                    external_vars = provider.get_all_variables()
+                    if isinstance(external_vars, dict):
+                        all_variables.update(external_vars)
+                except Exception as e:
+                    print(f"警告：获取外部变量提供者变量时发生错误: {e}")
+        
+        # 2. 再添加本地变量（覆盖同名的外部变量）
+        all_variables.update(self._data)
+        
+        return all_variables
+
+    def register_external_variable_provider(self, provider) -> None:
+        """注册外部变量提供者
+
+        Args:
+            provider: 变量提供者，需要实现get_variable(key)方法
+        """
+        if provider not in self._external_providers:
+            self._external_providers.append(provider)
+
+    def sync_variables_from_external_sources(self) -> None:
+        """将外部变量提供者中的常用变量同步到本地缓存中，提高访问性能
+
+        这个方法会调用所有外部提供者的get_all_variables方法，
+        将常用变量缓存到本地_data字典中，以提高后续访问的性能。
+        注意：本地变量的优先级仍然高于外部变量。
+        """
+        for provider in self._external_providers:
+            if hasattr(provider, 'get_all_variables'):
+                try:
+                    # 获取提供者的所有变量
+                    external_vars = provider.get_all_variables()
+                    if isinstance(external_vars, dict):
+                        # 只同步那些本地还没有的变量，保持本地变量的优先级
+                        for key, value in external_vars.items():
+                            if key not in self._data:
+                                self._data[key] = value
+                except Exception as e:
+                    # 如果某个提供者同步失败，记录警告但继续处理其他提供者
+                    print(f"警告：同步外部变量提供者变量时发生错误: {e}")

pytest_dsl/core/custom_keyword_manager.py

@@ -212,7 +212,7 @@ class CustomKeywordManager:
             print(f"资源文件 {file_path} 加载失败: {str(e)}")
             raise
 
-    def _process_resource_file_content(self, content: str, 
+    def _process_resource_file_content(self, content: str,
                                        file_path: str) -> None:
         """处理资源文件内容
 
@@ -258,7 +258,7 @@ class CustomKeywordManager:
                 # 递归加载导入的资源文件
                 self.load_resource_file(imported_file)
 
-    def _register_keywords_from_ast(self, ast: Node, 
+    def _register_keywords_from_ast(self, ast: Node,
                                     source_name: str) -> None:
         """从AST中注册关键字（重构后的版本）
 
@@ -341,6 +341,12 @@ class CustomKeywordManager:
                     executor.test_context.set(
                         param_name, kwargs[param_mapping_name])
 
+            # 重要：创建变量替换器，使变量解析正常工作
+            from pytest_dsl.core.variable_utils import VariableReplacer
+            executor.variable_replacer = VariableReplacer(
+                executor.variables, executor.test_context
+            )
+
             # 执行关键字体中的语句
             result = None
             try:
@@ -357,7 +363,7 @@ class CustomKeywordManager:
 
         print(f"已注册自定义关键字: {keyword_name} 来自文件: {file_path}")
 
-    def register_keyword_from_dsl_content(self, dsl_content: str, 
+    def register_keyword_from_dsl_content(self, dsl_content: str,
                                           source_name: str = "DSL内容") -> list:
         """从DSL内容注册关键字（公共方法）
 
@@ -379,8 +385,8 @@ class CustomKeywordManager:
 
             # 收集注册前的关键字列表
             existing_keywords = (
-                set(keyword_manager._keywords.keys()) 
-                if hasattr(keyword_manager, '_keywords') 
+                set(keyword_manager._keywords.keys())
+                if hasattr(keyword_manager, '_keywords')
                 else set()
             )
 
@@ -389,8 +395,8 @@ class CustomKeywordManager:
 
             # 计算新注册的关键字
             new_keywords = (
-                set(keyword_manager._keywords.keys()) 
-                if hasattr(keyword_manager, '_keywords') 
+                set(keyword_manager._keywords.keys())
+                if hasattr(keyword_manager, '_keywords')
                 else set()
             )
             registered_keywords = list(new_keywords - existing_keywords)
@@ -405,7 +411,7 @@ class CustomKeywordManager:
             raise
 
     def register_specific_keyword_from_dsl_content(
-            self, keyword_name: str, dsl_content: str, 
+            self, keyword_name: str, dsl_content: str,
             source_name: str = "DSL内容") -> bool:
         """从DSL内容注册指定的关键字（公共方法）