pytbox 0.1.1__tar.gz → 0.1.2__tar.gz

{pytbox-0.1.1/src/pytbox.egg-info → pytbox-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytbox
-Version: 0.1.1
+Version: 0.1.2
 Summary: A collection of Python integrations and utilities (Feishu, Dida365, VictoriaMetrics, ...)
 Author-email: mingming hou <houm01@foxmail.com>
 License-Expression: MIT

{pytbox-0.1.1 → pytbox-0.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pytbox"
-version = "0.1.1"
+version = "0.1.2"
 description = "A collection of Python integrations and utilities (Feishu, Dida365, VictoriaMetrics, ...)"
 authors = [{ name = "mingming hou", email = "houm01@foxmail.com" }]
 license = "MIT"

{pytbox-0.1.1 → pytbox-0.1.2}/src/pytbox/base.py

@@ -10,9 +10,12 @@ from pytbox.alert.alert_handler import AlertHandler
 from pytbox.log.logger import AppLogger
 from pytbox.win.ad import ADClient
 from pytbox.network.meraki import Meraki
+from pytbox.utils.env import get_env_by_os_environment
+from pytbox.vmware import VMwareClient
 
 config = load_config_by_file(path='/workspaces/pytbox/tests/alert/config_dev.toml', oc_vault_id=os.environ.get('oc_vault_id'))
 
+
 def get_mongo(collection):
     return Mongo(
         host=config['mongo']['host'],
@@ -61,4 +64,13 @@ def get_logger(app):
 #     password=config['ad']['prod']['AD_PASSWORD']
 # )
 
-meraki = Meraki(api_key=config['meraki']['api_key'], organization_id=config['meraki']['organization_id'])
+env = get_env_by_os_environment(check_key='ENV')
+meraki = Meraki(api_key=config['meraki']['api_key'], organization_id=config['meraki']['organization_id'])
+
+vmware_test = VMwareClient(
+    host=config['vmware']['test']['host'],
+    username=config['vmware']['test']['username'],
+    password=config['vmware']['test']['password'],
+    version=config['vmware']['test']['version'],
+    proxies=config['vmware']['test']['proxies']
+)

{pytbox-0.1.1 → pytbox-0.1.2}/src/pytbox/categraf/build_config.py

@@ -114,16 +114,11 @@ class BuildConfig:
                 device_templates = glob(str(jinja2_dir / f'{device_type}_*.toml.j2'))
                 if not device_templates:
                     continue
-                
                 for tmpl_path in device_templates:
                     tmpl_name = os.path.basename(tmpl_path)
-                    # 例如 h3c_system.toml.j2 -> h3c_system
                     base_name = tmpl_name.replace('.toml.j2', '')
-
                     template = self._get_template(f'input.snmp/{tmpl_name}')
-                    # 修复数据结构：模板期望的是数组，每个元素是字典
-                    # instances 是 [{"udp://10.1.1.1:161": {...}, "udp://10.1.1.2:161": {...}}, ...]
-                    render_data = template.render(instances=instances)
+                    render_data = template.render(instances=instances, config=self.instances['snmp']['config'])
 
                     target_dir = Path(self.output_dir) / 'input.snmp'
                     if not target_dir.exists():

{pytbox-0.1.1 → pytbox-0.1.2}/src/pytbox/categraf/instances.toml

@@ -25,6 +25,9 @@
 "119.29.29.29_baidu.com" = { dns_server = "119.29.29.29", domains = "www.baidu.com", labels = { name = "x", env = "prod" } }
 
 [snmp]
+[snmp.config]
+timeout = '120s'
+path = "/usr/share/snmp/mibs"
 # 支持 h3c,huawei,cisco,ruijie
 [[snmp.instances.h3c]]
 "10.1.1.1:161" = { version = 2, community = "public" }

{pytbox-0.1.1 → pytbox-0.1.2}/src/pytbox/categraf/jinja2/input.snmp/cisco_interface.toml.j2

@@ -17,9 +17,9 @@ priv_protocol = "{{detail.priv_protocol}}"
 priv_password = "{{detail.priv_password}}"
 sec_level = "{{detail.sec_level}}"
 {% endif %}
-timeout = "5s"
+timeout = "{{config.timeout}}"
 retries = 3
-path = ["/usr/share/snmp/mibs"]
+path = ["{{config.path}}"]
 translator = "gosmi"
 max_repetitions = 50
 

{pytbox-0.1.1 → pytbox-0.1.2}/src/pytbox/categraf/jinja2/input.snmp/cisco_system.toml.j2

@@ -17,9 +17,9 @@ priv_protocol = "{{detail.priv_protocol}}"
 priv_password = "{{detail.priv_password}}"
 sec_level = "{{detail.sec_level}}"
 {% endif %}
-timeout = "5s"
+timeout = "{{config.timeout}}"
 retries = 3
-path = ["/usr/share/snmp/mibs"]
+path = ["{{config.path}}"]
 translator = "gosmi"
 max_repetitions = 50
 

{pytbox-0.1.1 → pytbox-0.1.2}/src/pytbox/categraf/jinja2/input.snmp/h3c_interface.toml.j2

@@ -17,9 +17,9 @@ priv_protocol = "{{detail.priv_protocol}}"
 priv_password = "{{detail.priv_password}}"
 sec_level = "{{detail.sec_level}}"
 {% endif %}
-timeout = "5s"
+timeout = "{{config.timeout}}"
 retries = 3
-path = ["/usr/share/snmp/mibs"]
+path = ["{{config.path}}"]
 translator = "gosmi"
 max_repetitions = 50
 

{pytbox-0.1.1 → pytbox-0.1.2}/src/pytbox/categraf/jinja2/input.snmp/h3c_system.toml.j2

@@ -17,9 +17,9 @@ priv_protocol = "{{detail.priv_protocol}}"
 priv_password = "{{detail.priv_password}}"
 sec_level = "{{detail.sec_level}}"
 {% endif %}
-timeout = "5s"
+timeout = "{{config.timeout}}"
 retries = 3
-path = ["/usr/share/snmp/mibs"]
+path = ["{{config.path}}"]
 translator = "gosmi"
 max_repetitions = 50
 

{pytbox-0.1.1 → pytbox-0.1.2}/src/pytbox/categraf/jinja2/input.snmp/huawei_interface.toml.j2

@@ -17,9 +17,9 @@ priv_protocol = "{{detail.priv_protocol}}"
 priv_password = "{{detail.priv_password}}"
 sec_level = "{{detail.sec_level}}"
 {% endif %}
-timeout = "5s"
+timeout = "{{config.timeout}}"
 retries = 3
-path = ["/usr/share/snmp/mibs"]
+path = ["{{config.path}}"]
 translator = "gosmi"
 max_repetitions = 50
 

{pytbox-0.1.1 → pytbox-0.1.2}/src/pytbox/categraf/jinja2/input.snmp/huawei_system.toml.j2

@@ -17,9 +17,9 @@ priv_protocol = "{{detail.priv_protocol}}"
 priv_password = "{{detail.priv_password}}"
 sec_level = "{{detail.sec_level}}"
 {% endif %}
-timeout = "5s"
+timeout = "{{config.timeout}}"
 retries = 3
-path = ["/usr/share/snmp/mibs"]
+path = ["{{config.path}}"]
 translator = "gosmi"
 max_repetitions = 50
 

{pytbox-0.1.1 → pytbox-0.1.2}/src/pytbox/categraf/jinja2/input.snmp/ruijie_interface.toml.j2

@@ -17,9 +17,9 @@ priv_protocol = "{{detail.priv_protocol}}"
 priv_password = "{{detail.priv_password}}"
 sec_level = "{{detail.sec_level}}"
 {% endif %}
-timeout = "5s"
+timeout = "{{config.timeout}}"
 retries = 3
-path = ["/usr/share/snmp/mibs"]
+path = ["{{config.path}}"]
 translator = "gosmi"
 max_repetitions = 50
 

{pytbox-0.1.1 → pytbox-0.1.2}/src/pytbox/categraf/jinja2/input.snmp/ruijie_system.toml.j2

@@ -17,9 +17,9 @@ priv_protocol = "{{detail.priv_protocol}}"
 priv_password = "{{detail.priv_password}}"
 sec_level = "{{detail.sec_level}}"
 {% endif %}
-timeout = "5s"
+timeout = "{{config.timeout}}"
 retries = 3
-path = ["/usr/share/snmp/mibs"]
+path = ["{{config.path}}"]
 translator = "gosmi"
 max_repetitions = 50
 

pytbox-0.1.2/src/pytbox/cli/commands/vm.py

@@ -0,0 +1,22 @@
+#!/usr/bin/env python3
+
+import click
+from pytbox.database.victoriametrics import VictoriaMetrics
+from pytbox.utils.richutils import RichUtils
+
+rich_utils = RichUtils()
+
+@click.group()
+def vm_group():
+    """VictoriaMetrics 查询工具"""
+    pass
+
+
+@vm_group.command('query')
+@click.option('--url', '-u', type=str, required=True)
+@click.option('--query', '-q', type=str, required=True)
+def query(url, query):
+    """查询 VM 数据"""
+    vm_client = VictoriaMetrics(url=url)
+    r = vm_client.query(query, output_format='json')
+    rich_utils.print(msg=r)

{pytbox-0.1.1 → pytbox-0.1.2}/src/pytbox/cli/main.py

@@ -5,6 +5,7 @@ Pytbox 主命令行入口
 
 import click
 from .categraf import categraf_group
+from .commands.vm import vm_group
 
 
 @click.group()
@@ -16,6 +17,7 @@ def main():
 
 # 注册子命令组
 main.add_command(categraf_group, name='categraf')
+main.add_command(vm_group, name='vm')
 
 
 if __name__ == "__main__":

{pytbox-0.1.1 → pytbox-0.1.2}/src/pytbox/database/victoriametrics.py

@@ -1,8 +1,10 @@
 #!/usr/bin/env python3
 
+import json
 from typing import Literal, Optional
 import requests
 from ..utils.response import ReturnResponse
+from ..utils.load_vm_devfile import load_dev_file
 
 
 class VictoriaMetrics:
@@ -16,13 +18,13 @@ class VictoriaMetrics:
             'Accept': 'application/json'
         })
         
-    def query(self, query: str) -> ReturnResponse:
+    def query(self, query: str, output_format: Literal['json']=None) -> ReturnResponse:
         '''
         查询指标数据
-        
+
         Args:
             query (str): 查询语句
-            
+
         Returns:
             dict: 查询结果
         '''
@@ -32,13 +34,32 @@ class VictoriaMetrics:
             timeout=self.timeout,
             params={"query": query}
         )
-        if r.json().get("status") == "success":
-            if r.json()['data']['result']:
-                return ReturnResponse(code=0, msg=f"[{query}] 查询成功!", data=r.json()['data']['result'])
+        res_json = r.json()
+        status = res_json.get("status")
+        result = res_json.get("data", {}).get("result", [])
+        is_json = output_format == 'json'
+
+        if status == "success":
+            if result:
+                code = 0
+                msg = f"[{query}] 查询成功!"
+                data = result
             else:
-                return ReturnResponse(code=2, msg=f"[{query}] 没有查询到结果", data=r.json())
+                code = 2
+                msg = f"[{query}] 没有查询到结果"
+                data = res_json
         else:
-            return ReturnResponse(code=1, msg=f"[{query}] 查询失败: {r.json().get('error')}", data=r.json())
+            code = 1
+            msg = f"[{query}] 查询失败: {res_json.get('error')}"
+            data = res_json
+
+        resp = ReturnResponse(code=code, msg=msg, data=data)
+
+        if is_json:
+            json_result = json.dumps(resp.__dict__, ensure_ascii=False)
+            return json_result
+        else:
+            return resp
 
     def get_labels(self, metric_name: str) -> ReturnResponse:
         url = f"{self.url}/api/v1/series?match[]={metric_name}"
@@ -49,7 +70,7 @@ class VictoriaMetrics:
         else:
             return ReturnResponse(code=1, msg=f"metric name: {metric_name} 查询失败")
 
-    def check_ping_result(self, target: str, last_minute: int=10) -> ReturnResponse:
+    def check_ping_result(self, target: str, last_minute: int=10, env: str='prod', dev_file: str='') -> ReturnResponse:
         '''
         检查ping结果
         '''
@@ -62,13 +83,18 @@ class VictoriaMetrics:
         if last_minute:
             query = query + f"[{last_minute}m]"
         
-        r = self.query(query=query)
+        if env == 'dev':
+            r = load_dev_file(dev_file)
+        else:
+            r = self.query(query=query)
+ 
         if r.code == 0:
             values = r.data[0]['values']
             if len(values) == 2 and values[1] == "0":
                 code = 0
                 msg = f"已检查 {target} 最近 {last_minute} 分钟是正常的!"
             else:
+                
                 if all(str(item[1]) == "1" for item in values):
                     code = 1
                     msg = f"已检查 {target} 最近 {last_minute} 分钟是异常的!"

{pytbox-0.1.1 → pytbox-0.1.2}/src/pytbox/utils/env.py

@@ -17,14 +17,14 @@ def get_env_by_file_exist(file_path: str) -> Literal['prod', 'dev']:
         return 'dev'
 
 
-def get_env_by_os_environment() -> Literal['dev', 'prod']:
+def get_env_by_os_environment(check_key: str='ENV') -> Literal['dev', 'prod']:
     '''
     根据环境变量获取环境
 
     Returns:
         Literal['dev', 'prod']: 环境，dev 表示开发环境，prod 表示生产环境
     '''
-    if os.getenv('ENV') == 'dev':
+    if os.getenv(check_key) == 'dev':
         return 'dev'
     else:
         return 'prod'

pytbox-0.1.2/src/pytbox/utils/load_config.py

@@ -0,0 +1,132 @@
+#!/usr/bin/env python3
+
+import os
+import json
+
+try:
+    import toml
+except ImportError:
+    import tomllib as toml  # Python 3.11+ fallback
+from pytbox.onepassword_connect import OnePasswordConnect
+
+
+def _replace_values(data, oc=None, jsonfile_path=None):
+    """
+    递归处理配置数据，替换 1password、password 和 jsonfile 开头的值
+    
+    Args:
+        data: 配置数据（dict, list, str 等）
+        oc: OnePasswordConnect 实例
+        jsonfile_path: JSON 文件路径
+    
+    Returns:
+        处理后的数据
+    """
+    if isinstance(data, dict):
+        result = {}
+        for k, v in data.items():
+            result[k] = _replace_values(v, oc, jsonfile_path)
+        return result
+    elif isinstance(data, list):
+        return [_replace_values(item, oc, jsonfile_path) for item in data]
+    elif isinstance(data, str):
+        # 处理 1password,item_id,field_name 格式
+        if data.startswith("1password,") and oc:
+            parts = data.split(",")
+            if len(parts) >= 3:
+                item_id = parts[1]
+                field_name = parts[2]
+                try:
+                    # 通过 item_id 获取项目，然后从字段中提取对应值
+                    item = oc.get_item(item_id)
+                    for field in item.fields:
+                        if field.label == field_name:
+                            return field.value
+                except (AttributeError, KeyError, ValueError):
+                    pass
+                return data  # 如果找不到字段，返回原始值
+        # 处理 password,item_id,field_name 格式  
+        elif data.startswith("password,") and oc:
+            parts = data.split(",")
+            if len(parts) >= 3:
+                item_id = parts[1]
+                field_name = parts[2]
+                try:
+                    # 通过 item_id 获取项目，然后从字段中提取对应值
+                    item = oc.get_item(item_id)
+                    for field in item.fields:
+                        if field.label == field_name:
+                            return field.value
+                except (AttributeError, KeyError, ValueError):
+                    pass
+                return data  # 如果找不到字段，返回原始值
+        # 处理 jsonfile,key 格式
+        elif data.startswith("jsonfile,"):
+            parts = data.split(",", 1)  # 只分割一次，防止 key 中包含逗号
+            if len(parts) >= 2:
+                key = parts[1]
+                
+                # 尝试从 JSON 文件获取值
+                if jsonfile_path and os.path.exists(jsonfile_path):
+                    try:
+                        with open(jsonfile_path, 'r', encoding='utf-8') as f:
+                            json_data = json.load(f)
+                            # 支持嵌套键，如 "db.password"
+                            value = json_data
+                            for k in key.split('.'):
+                                if isinstance(value, dict) and k in value:
+                                    value = value[k]
+                                else:
+                                    value = None
+                                    break
+                            if value is not None:
+                                return value
+                    except (json.JSONDecodeError, FileNotFoundError, KeyError):
+                        pass
+                
+                # 如果从 JSON 文件获取失败，尝试从环境变量获取
+                env_value = os.getenv(key)
+                if env_value is not None:
+                    return env_value
+                    
+                return data  # 如果都获取不到，返回原始值
+        return data
+    else:
+        return data
+
+
+def load_config_by_file(
+        path: str='/workspaces/pytbox/src/pytbox/alert/config/config.toml', 
+        oc_vault_id: str=None, 
+        jsonfile: str="/data/jsonfile.json",
+    ) -> dict:
+    '''
+    从文件加载配置，支持 1password 集成
+    
+    Args:
+        path (str, optional): 配置文件路径. Defaults to '/workspaces/pytbox/src/pytbox/alert/config/config.toml'.
+        oc_vault_id (str, optional): OnePassword vault ID，如果提供则启用 1password 集成
+        
+    Returns:
+        dict: 配置字典
+    '''
+    with open(path, 'r', encoding='utf-8') as f:
+        if path.endswith('.toml'): 
+            config = toml.load(f)
+        else:
+            # 如果不是 toml 文件，假设是其他格式，这里可以扩展
+            config = json.load(f)
+            
+        # 处理配置值替换
+        oc = None
+        if oc_vault_id:
+            oc = OnePasswordConnect(vault_id=oc_vault_id)
+        
+        # 替换配置中的特殊值（1password, password, jsonfile）
+        config = _replace_values(config, oc, jsonfile)
+        
+        return config
+
+
+if __name__ == "__main__":
+    print(load_config_by_file(path='/workspaces/pytbox/tests/alert/config_dev.toml', oc_vault_id="hcls5uxuq5dmxorw6rfewefdsa"))

pytbox-0.1.2/src/pytbox/utils/load_vm_devfile.py

@@ -0,0 +1,45 @@
+#!/usr/bin/env python3
+
+import json
+from pytbox.utils.response import ReturnResponse
+
+
+def load_dev_file(file_path: str) -> ReturnResponse:
+    """从开发环境文件加载数据并返回 ReturnResponse 对象
+    
+    Args:
+        file_path: JSON 文件路径
+        
+    Returns:
+        ReturnResponse: 包装后的响应对象
+    """
+    try:
+        with open(file_path, 'r', encoding='utf-8') as f:
+            data = json.load(f)
+        
+        # 如果已经是 ReturnResponse 格式的字典，直接转换
+        if isinstance(data, dict) and 'code' in data and 'msg' in data:
+            return ReturnResponse(
+                code=data.get('code', 0),
+                msg=data.get('msg', ''),
+                data=data.get('data', None)
+            )
+        else:
+            # 如果是其他格式，包装成成功响应
+            return ReturnResponse(
+                code=0,
+                msg='开发文件加载成功',
+                data=data
+            )
+    except FileNotFoundError:
+        return ReturnResponse(
+            code=4,
+            msg=f'开发文件未找到: {file_path}',
+            data=None
+        )
+    except json.JSONDecodeError as e:
+        return ReturnResponse(
+            code=1,
+            msg=f'JSON 解析错误: {str(e)}',
+            data=None
+        )

pytbox-0.1.2/src/pytbox/vmware.py

@@ -0,0 +1,120 @@
+#!/usr/bin/env python3
+
+
+from typing import Any, Dict, Optional, Union, Literal
+import urllib3
+urllib3.disable_warnings()
+import requests
+from requests.auth import HTTPBasicAuth
+
+from .utils.response import ReturnResponse
+
+
+class VMwareClient:
+    """VMware vSphere Automation API 客户端。
+    
+    支持多种认证方式：
+    1. Basic Auth - HTTP 基础认证
+    2. API Key Auth - 使用会话 ID 认证
+    3. Federated Identity Auth - 联合身份认证（Bearer Token）
+    """
+    
+    def __init__(
+        self,
+        host: str=None,
+        username: str=None,
+        password: str=None,
+        version: Literal['6.7', '7.0']='6.7',
+        proxies: dict=None,
+        verify_ssl: bool = False,
+        timeout: int = 30,
+    ) -> None:
+        """初始化 VMware 客户端。
+        
+        Args:
+            host: vCenter Server 主机地址（例如：https://vcenter.example.com）
+            verify_ssl: 是否验证 SSL 证书
+            timeout: 请求超时时间（秒）
+            proxy_host: 代理服务器主机地址
+            proxy_port: 代理服务器端口
+            proxy_username: 代理认证用户名（可选）
+            proxy_password: 代理认证密码（可选）
+        """
+        self.host = host
+        self.username = username
+        self.password = password
+        self.version = version
+        self.proxies = proxies
+        self.verify_ssl = verify_ssl
+        self.timeout = timeout
+        self.session_id: Optional[str] = None
+    
+        self.headers = {
+            "vmware-api-session-id": self.get_session()
+        }
+ 
+    def get_session(self) -> str:
+        """获取 VMware vSphere API 会话 ID。
+        
+        Returns:
+            会话 ID 字符串
+        """
+        if self.version == '6.7':
+            url = f"{self.host}/rest/com/vmware/cis/session"
+        else:
+            url = f"{self.host}/api/session"
+   
+        response = requests.post(
+            url, 
+            auth=HTTPBasicAuth(self.username, self.password), 
+            timeout=self.timeout, 
+            verify=self.verify_ssl, 
+            proxies=self.proxies
+        )
+        
+        if response.status_code == 200 or response.status_code == 201:
+            # vSphere API 通常直接返回 session ID 字符串
+            session_id = response.json()
+            try:
+                return session_id['value']
+            except Exception:
+                return session_id
+        else:
+            return f"认证失败: {response.status_code} - {response.text}"
+    
+    def get_vm_list(self) -> ReturnResponse:
+        '''
+        _summary_
+
+        Returns:
+            ReturnResponse: _description_
+        '''
+        if self.version == '6.7':
+            url = f"{self.host}/rest/vcenter/vm"
+        else:
+            url = f"{self.host}/api/vcenter/vm"
+        response = requests.get(url, headers=self.headers, timeout=self.timeout, verify=False, proxies=self.proxies)
+        if response.status_code == 200:
+            if self.version == '6.7':
+                data = response.json().get('value')
+            else:
+                data = response.json()
+            return ReturnResponse(code=0, msg=f'成功获取到 {len(data)} 台虚拟机', data=data)
+        else:
+            return ReturnResponse(code=1, msg='error', data=response.json())
+   
+    def get_vm(self, vm_id):
+        if self.version == '6.7':
+            url = f"{self.host}/rest/vcenter/vm/{vm_id}"
+        else:
+            url = f"{self.host}/api/vcenter/vm/{vm_id}"
+        response = requests.get(url, headers=self.headers, timeout=self.timeout, verify=False, proxies=self.proxies)
+        if response.status_code == 200:
+            return ReturnResponse(code=0, msg='成功获取到虚拟机', data=response.json())
+        else:
+            return ReturnResponse(code=1, msg=f"{response.status_code}, {response.text}", data=response.json())
+   
+# 使用示例
+if __name__ == "__main__":
+    pass
+    

{pytbox-0.1.1 → pytbox-0.1.2/src/pytbox.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytbox
-Version: 0.1.1
+Version: 0.1.2
 Summary: A collection of Python integrations and utilities (Feishu, Dida365, VictoriaMetrics, ...)
 Author-email: mingming hou <houm01@foxmail.com>
 License-Expression: MIT

{pytbox-0.1.1 → pytbox-0.1.2}/src/pytbox.egg-info/SOURCES.txt

@@ -6,6 +6,7 @@ src/pytbox/cli.py
 src/pytbox/dida365.py
 src/pytbox/onepassword_connect.py
 src/pytbox/onepassword_sa.py
+src/pytbox/vmware.py
 src/pytbox.egg-info/PKG-INFO
 src/pytbox.egg-info/SOURCES.txt
 src/pytbox.egg-info/dependency_links.txt
@@ -41,6 +42,7 @@ src/pytbox/cli/__init__.py
 src/pytbox/cli/main.py
 src/pytbox/cli/categraf/__init__.py
 src/pytbox/cli/categraf/commands.py
+src/pytbox/cli/commands/vm.py
 src/pytbox/cli/common/__init__.py
 src/pytbox/cli/common/options.py
 src/pytbox/cli/common/utils.py
@@ -61,6 +63,7 @@ src/pytbox/mail/client.py
 src/pytbox/network/meraki.py
 src/pytbox/utils/env.py
 src/pytbox/utils/load_config.py
+src/pytbox/utils/load_vm_devfile.py
 src/pytbox/utils/ping_checker.py
 src/pytbox/utils/response.py
 src/pytbox/utils/richutils.py
@@ -71,4 +74,5 @@ tests/test_feishu.py
 tests/test_logger.py
 tests/test_onepassword_connect.py
 tests/test_onepassword_sa.py
-tests/test_victoriametrics.py
+tests/test_victoriametrics.py
+tests/test_vmware.py

pytbox-0.1.2/tests/test_vmware.py

@@ -0,0 +1,6 @@
+#!/usr/bin/env python3
+
+from pytbox.base import vmware_test
+
+r = vmware_test.get_vm_list()
+print(r)

pytbox-0.1.1/src/pytbox/utils/load_config.py

@@ -1,86 +0,0 @@
-#!/usr/bin/env python3
-
-
-try:
-    import toml
-except ImportError:
-    import tomllib as toml  # Python 3.11+ fallback
-from pytbox.onepassword_connect import OnePasswordConnect
-
-
-def _replace_1password_values(data, oc):
-    """
-    递归处理配置数据，替换 1password 和 password 开头的值
-    
-    Args:
-        data: 配置数据（dict, list, str 等）
-        oc: OnePasswordConnect 实例
-    
-    Returns:
-        处理后的数据
-    """
-    if isinstance(data, dict):
-        result = {}
-        for k, v in data.items():
-            result[k] = _replace_1password_values(v, oc)
-        return result
-    elif isinstance(data, list):
-        return [_replace_1password_values(item, oc) for item in data]
-    elif isinstance(data, str):
-        # 处理 1password,item_id,field_name 格式
-        if data.startswith("1password,"):
-            parts = data.split(",")
-            if len(parts) >= 3:
-                item_id = parts[1]
-                field_name = parts[2]
-                # 通过 item_id 获取项目，然后从字段中提取对应值
-                item = oc.get_item(item_id)
-                for field in item.fields:
-                    if field.label == field_name:
-                        return field.value
-                return data  # 如果找不到字段，返回原始值
-        # 处理 password,item_id,field_name 格式  
-        elif data.startswith("password,"):
-            parts = data.split(",")
-            if len(parts) >= 3:
-                item_id = parts[1]
-                field_name = parts[2]
-                # 通过 item_id 获取项目，然后从字段中提取对应值
-                item = oc.get_item(item_id)
-                for field in item.fields:
-                    if field.label == field_name:
-                        return field.value
-                return data  # 如果找不到字段，返回原始值
-        return data
-    else:
-        return data
-
-
-def load_config_by_file(path: str='/workspaces/pytbox/src/pytbox/alert/config/config.toml', oc_vault_id: str=None) -> dict:
-    '''
-    从文件加载配置，支持 1password 集成
-    
-    Args:
-        path (str, optional): 配置文件路径. Defaults to '/workspaces/pytbox/src/pytbox/alert/config/config.toml'.
-        oc_vault_id (str, optional): OnePassword vault ID，如果提供则启用 1password 集成
-        
-    Returns:
-        dict: 配置字典
-    '''
-    with open(path, 'r', encoding='utf-8') as f:
-        if path.endswith('.toml'):
-            config = toml.load(f)
-        else:
-            # 如果不是 toml 文件，假设是其他格式，这里可以扩展
-            import json
-            config = json.load(f)
-            
-        if oc_vault_id:
-            oc = OnePasswordConnect(vault_id=oc_vault_id)
-            config = _replace_1password_values(config, oc)
-            
-        return config
-
-
-if __name__ == "__main__":
-    print(load_config_by_file(path='/workspaces/pytbox/tests/alert/config_dev.toml', oc_vault_id="hcls5uxuq5dmxorw6rfewefdsa"))