pytbox 0.0.3__py3-none-any.whl → 0.0.4__py3-none-any.whl

pytbox/database/victoriametrics.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 
+from typing import Literal, Optional
 import requests
 from ..utils.response import ReturnResponse
 
@@ -38,7 +39,16 @@ class VictoriaMetrics:
                 return ReturnResponse(code=2, msg=f"[{query}] 没有查询到结果", data=r.json())
         else:
             return ReturnResponse(code=1, msg=f"[{query}] 查询失败: {r.json().get('error')}", data=r.json())
-    
+
+    def get_labels(self, metric_name: str) -> ReturnResponse:
+        url = f"{self.url}/api/v1/series?match[]={metric_name}"
+        response = requests.get(url, timeout=self.timeout)
+        results = response.json()
+        if results['status'] == 'success':
+            return ReturnResponse(code=0, msg=f"metric name: {metric_name} 获取到 {len(results['data'])} 条数据", data=results['data'])
+        else:
+            return ReturnResponse(code=1, msg=f"metric name: {metric_name} 查询失败")
+
     def check_ping_result(self, target: str, last_minute: int=10) -> ReturnResponse:
         '''
         检查ping结果
@@ -60,7 +70,7 @@ class VictoriaMetrics:
                 msg = f"已检查 {target} 最近 {last_minute} 分钟是正常的!"
             else:
                 if all(str(item[1]) == "1" for item in values):
-                    code = 2
+                    code = 1
                     msg = f"已检查 {target} 最近 {last_minute} 分钟是异常的!"
                 else:
                     code = 0
@@ -74,4 +84,30 @@ class VictoriaMetrics:
         except KeyError:
             data = r.data
         
-        return ReturnResponse(code=code, msg=msg, data=data)
+        return ReturnResponse(code=code, msg=msg, data=data)
+
+    def query_interface_rate(self,
+                             direction: Literal['in', 'out'], 
+                             sysName: str, 
+                             ifName:str, 
+                             last_minutes: Optional[int] = None
+                            ) -> ReturnResponse:
+        """查询指定设备的入方向总流量速率（bps）。
+
+        使用 PromQL 对 `snmp_interface_ifHCInOctets` 进行速率计算并聚合到设备级别，
+        将结果从字节每秒转换为比特每秒（乘以 8）。
+
+        Args:
+            sysName: 设备 `sysName` 标签值。
+            last_minutes: 计算速率的时间窗口（分钟）。未提供时默认使用 5 分钟窗口。
+
+        Returns:
+            ReturnResponse: 查询结果包装。
+        """
+        if direction == 'in':
+            query = f'(rate(snmp_interface_ifHCInOctets{{sysName="{sysName}", ifName="{ifName}"}}[{last_minutes}m])) * 8 / 1000000'
+        else:
+            query = f'(rate(snmp_interface_ifHCOutOctets{{sysName="{sysName}", ifName="{ifName}"}}[{last_minutes}m])) * 8 / 1000000'
+        r = self.query(query)
+        rate = r.data['values'][1]
+        return int(float(rate))

{pytbox-0.0.3.dist-info → pytbox-0.0.4.dist-info}/METADATA

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

{pytbox-0.0.3.dist-info → pytbox-0.0.4.dist-info}/RECORD

@@ -8,7 +8,7 @@ pytbox/alicloud/sls.py,sha256=UR4GdI86dCKAFI2xt_1DELu7q743dpd3xrYtuNpfC5A,4065
 pytbox/common/__init__.py,sha256=3JWfgCQZKZuSH5NCE7OCzKwq82pkyop9l7sH5YSNyfU,122
 pytbox/common/base.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pytbox/database/mongo.py,sha256=QyCeSoyWoJdIFXxjDmbV5t-xU7wN3x-Ig0I8pvRmBwg,1864
-pytbox/database/victoriametrics.py,sha256=qs3IQZbcKGSIObQcwUujZEyPshq2yOIVvnc4yfTkXJ0,2642
+pytbox/database/victoriametrics.py,sha256=fh6wghcWLhQxizVljC6X22gqXr2CioB1L4qixSi33Ms,4388
 pytbox/feishu/client.py,sha256=kwGLseGT_iQUFmSqpuS2_77WmxtHstD64nXvktuQ3B4,5865
 pytbox/feishu/endpoints.py,sha256=z_nHGXAjlIyYdFnbAWY4hfzzHrENJvoNOqY9sA6-FLk,40009
 pytbox/feishu/errors.py,sha256=79qFAHZw7jDj3gnWAjI1-W4tB0q1_aSfdjee4xzXeuI,1179
@@ -21,7 +21,7 @@ pytbox/utils/load_config.py,sha256=wNCDPLH7xet5b9pUlTz6VsBejRsJZ7LP85wWMaITBYg,3
 pytbox/utils/ping_checker.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 pytbox/utils/response.py,sha256=kXjlwt0WVmLRam2eu1shzX2cQ7ux4cCQryaPGYwle5g,1247
 pytbox/utils/timeutils.py,sha256=7ZI-TU2FCUV0Gyc_QUjhSreto8SVIs54e2CZgGix5pI,1628
-pytbox-0.0.3.dist-info/METADATA,sha256=WWQcYi7uThTfbjMuhEsV9vKgqInlBZWn1VJr-0llPDQ,6006
-pytbox-0.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pytbox-0.0.3.dist-info/top_level.txt,sha256=YADgWue-Oe128ptN3J2hS3GB0Ncc5uZaSUM3e1rwswE,7
-pytbox-0.0.3.dist-info/RECORD,,
+pytbox-0.0.4.dist-info/METADATA,sha256=bgImPZ1wSfkX-ARmOiiBDaRwrrdHqf9sgzcrs5uGUcw,6006
+pytbox-0.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pytbox-0.0.4.dist-info/top_level.txt,sha256=YADgWue-Oe128ptN3J2hS3GB0Ncc5uZaSUM3e1rwswE,7
+pytbox-0.0.4.dist-info/RECORD,,