pytbox 0.1.5__tar.gz → 0.1.7__tar.gz

{pytbox-0.1.5/src/pytbox.egg-info → pytbox-0.1.7}/PKG-INFO

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

{pytbox-0.1.5 → pytbox-0.1.7}/pyproject.toml

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

{pytbox-0.1.5 → pytbox-0.1.7}/src/pytbox/alert/ping.py

@@ -1,6 +1,5 @@
 #!/usr/bin/env python3
 
-
 from ..database.victoriametrics import VictoriaMetrics
 from ..lib.load_config import load_config
 

{pytbox-0.1.5 → pytbox-0.1.7}/src/pytbox/alicloud/sls.py

@@ -5,19 +5,21 @@ from typing import Literal
 from aliyun.log import GetLogsRequest, LogItem, PutLogsRequest
 from aliyun.log import LogClient as SlsLogClient
 from aliyun.log.auth import AUTH_VERSION_4
-from ..utils.env import check_env
 
 
 
 class AliCloudSls:
-
-    def __init__(self, access_key_id: str=None, access_key_secret: str=None, project: str, logstore: str):
+    '''
+    pip install -U aliyun-log-python-sdk
+    '''
+    def __init__(self, access_key_id: str=None, access_key_secret: str=None, project: str=None, logstore: str=None, env: str='prod'):
         # 日志服务的服务接入点
         self.endpoint = "cn-shanghai.log.aliyuncs.com"
         # 创建 LogClient 实例，使用 V4 签名，根据实际情况填写 region，这里以杭州为例
         self.client = SlsLogClient(self.endpoint, access_key_id, access_key_secret, auth_version=AUTH_VERSION_4, region='cn-shanghai')
         self.project = project
         self.logstore = logstore
+        self.env = env
 
     def get_logs(self, project_name, logstore_name, query, from_time, to_time): 
         logstore_index = {'line': {
@@ -57,7 +59,7 @@ class AliCloudSls:
         log_group = []
         log_item = LogItem()
         contents = [
-            ('env', check_env()),
+            ('env', self.env),
             ('level', level),
             ('app', app),
             ('msg', msg),
@@ -69,8 +71,9 @@ class AliCloudSls:
         log_item.set_contents(contents)
         log_group.append(log_item)
         request = PutLogsRequest(self.project, self.logstore, topic, "", log_group, compress=False)
-        self.client.put_logs(request)
-    
+        r = self.client.put_logs(request)
+        return r
+        
     def put_logs_for_meraki(self, alert):
         log_group = []
         log_item = LogItem()

{pytbox-0.1.5 → pytbox-0.1.7}/src/pytbox/base.py

@@ -15,6 +15,9 @@ from pytbox.vmware import VMwareClient
 from pytbox.pyjira import PyJira
 from pytbox.mail.client import MailClient
 from pytbox.mail.alimail import AliMail
+from pytbox.alicloud.sls import AliCloudSls
+from pytbox.utils.cronjob import cronjob_counter
+
 
 config = load_config_by_file(path='/workspaces/pytbox/tests/alert/config_dev.toml', oc_vault_id=os.environ.get('oc_vault_id'))
 
@@ -53,6 +56,19 @@ def get_logger(app):
         mongo=get_mongo('alert_program')
     )
 
+def get_logger_sls(app):
+    return AppLogger(
+        app_name=app, 
+        enable_sls=True,
+        feishu=feishu,
+        dida=dida,
+        mongo=get_mongo('alert_program'),
+        sls_access_key_id=config['alicloud']['account1']['access_key_id'],
+        sls_access_key_secret=config['alicloud']['account1']['access_key_secret'],
+        sls_project=config['alicloud']['account1']['project'],
+        sls_logstore=config['alicloud']['account1']['logstore']
+    )
+
 # ad_dev = ADClient(
 #     server=config['ad']['dev']['AD_SERVER'],
 #     base_dn=config['ad']['dev']['BASE_DN'],
@@ -85,4 +101,15 @@ pyjira = PyJira(
 
 mail_163 = MailClient(mail_address=config['mail']['163']['mail_address'], password=config['mail']['163']['password'])
 mail_qq = MailClient(mail_address=config['mail']['qq']['mail_address'], password=config['mail']['qq']['password'])
-ali_mail = AliMail(mail_address=config['mail']['aliyun']['mail_address'], client_id=config['mail']['aliyun']['client_id'], client_secret=config['mail']['aliyun']['client_secret'])
+ali_mail = AliMail(mail_address=config['mail']['aliyun']['mail_address'], client_id=config['mail']['aliyun']['client_id'], client_secret=config['mail']['aliyun']['client_secret'])
+
+sls = AliCloudSls(
+    access_key_id=config['alicloud']['account1']['access_key_id'],
+    access_key_secret=config['alicloud']['account1']['access_key_secret'],
+    project=config['alicloud']['account1']['project'],
+    logstore=config['alicloud']['account1']['logstore']
+)
+
+
+def get_cronjob_counter(app_type='', app='', comment=None, schedule_interval=None, schedule_cron=None):
+    return cronjob_counter(vm=vm, log=get_logger('cronjob_counter'), app_type=app_type, app=app, comment=comment, schedule_interval=schedule_interval, schedule_cron=schedule_cron)

pytbox-0.1.7/src/pytbox/database/victoriametrics.py

@@ -0,0 +1,312 @@
+#!/usr/bin/env python3
+
+import time
+import json
+from typing import Literal, Optional, Dict, List
+import requests
+from ..utils.response import ReturnResponse
+from ..utils.load_vm_devfile import load_dev_file
+
+
+class VictoriaMetrics:
+    
+    def __init__(self, url: str='', timeout: int=3, env: str='prod') -> None:
+        self.url = url
+        self.timeout = timeout
+        self.session = requests.Session()
+        self.session.headers.update({
+            'Content-Type': 'application/json',
+            'Accept': 'application/json'
+        })
+        self.env = env
+
+    def insert(self, metric_name: str = '', labels: Dict[str, str] = None, 
+               value: List[float] = None, timestamp: int = None) -> ReturnResponse:
+        """插入指标数据。
+        
+        Args:
+            metric_name: 指标名称
+            labels: 标签字典
+            value: 值列表
+            timestamp: 时间戳（毫秒），默认为当前时间
+            
+        Raises:
+            requests.RequestException: 当请求失败时抛出
+        """
+        if labels is None:
+            labels = {}
+        if value is None:
+            value = 1
+        if timestamp is None:
+            timestamp = int(time.time() * 1000)
+            
+        url = f"{self.url}/api/v1/import"
+        data = {
+            "metric": {
+                "__name__": metric_name,
+                **labels
+            },
+            "values": [value],
+            "timestamps": [timestamp]
+        }
+        
+        try:
+            response = requests.post(url, json=data, timeout=self.timeout)
+            return ReturnResponse(code=0, msg=f"数据插入成功，状态码: {response.status_code}, metric_name: {metric_name}, labels: {labels}, value: {value}, timestamp: {timestamp}")
+        except requests.RequestException as e:
+            return ReturnResponse(code=1, msg=f"数据插入失败: {e}")
+
+    def query(self, query: str=None, output_format: Literal['json']=None) -> ReturnResponse:
+        '''
+        查询指标数据
+
+        Args:
+            query (str): 查询语句
+
+        Returns:
+            dict: 查询结果
+        '''
+        url = f"{self.url}/prometheus/api/v1/query"
+        r = requests.get(
+            url, 
+            timeout=self.timeout,
+            params={"query": query}
+        )
+        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:
+                code = 2
+                msg = f"[{query}] 没有查询到结果"
+                data = res_json
+        else:
+            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 query_range(self, query):
+        '''
+        查询指标数据
+
+        Args:
+            query (str): 查询语句
+
+        Returns:
+            dict: 查询结果
+        '''
+        url = f"{self.url}/prometheus/api/v1/query_range"
+
+        data = {
+            'query': query,
+            'start': '-1d',
+            'step': '1h'
+        }
+
+        r = requests.post(url, data=data, timeout=self.timeout)
+        res_json = r.json()
+        print(res_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:
+        #         code = 2
+        #         msg = f"[{query}] 没有查询到结果"
+        #         data = res_json
+        # else:
+        #     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}"
+        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, env: str='prod', dev_file: str='') -> ReturnResponse:
+        '''
+        检查ping结果
+
+        Args:
+            target (str): 目标地址
+            last_minute (int, optional): 最近多少分钟. Defaults to 10.
+            env (str, optional): 环境. Defaults to 'prod'.
+            dev_file (str, optional): 开发文件. Defaults to ''.
+
+        Returns:
+            ReturnResponse: 
+                code = 0 正常, code = 1 异常, code = 2 没有查询到数据, 建议将其判断为正常
+        '''
+        if target:
+            # 这里需要在字符串中保留 {}，同时插入 target，可以用双大括号转义
+            query = f"ping_result_code{{target='{target}'}}"
+        else:
+            query = "ping_result_code"
+        
+        if last_minute:
+            query = query + f"[{last_minute}m]"
+        
+        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} 分钟是异常的!"
+                else:
+                    code = 0
+                    msg = f"已检查 {target} 最近 {last_minute} 分钟是正常的!"
+        elif r.code == 2:
+            code = 2
+            msg = f"没有查询到 {target} 最近 {last_minute} 分钟的ping结果!"
+        try:
+            data = r.data[0]
+        except KeyError:
+            data = r.data
+        
+        return ReturnResponse(code=code, msg=msg, data=data)
+
+    def check_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[0]['value'][1]
+        return int(float(rate))
+    
+    def check_snmp_port_status(self, sysname: str=None, if_name: str=None, last_minute: int=5) -> ReturnResponse:
+        '''
+        查询端口状态
+        status code 可参考 SNMP 文件 https://mibbrowser.online/mibdb_search.php?mib=IF-MIB
+
+        Args:
+            sysname (_type_): 设备名称
+            if_name (_type_): _description_
+            last_minute (_type_): _description_
+
+        Returns:
+            ReturnResponse: 
+            code: 0, msg: , data: up,down
+        '''
+        q = f"""avg_over_time(snmp_interface_ifOperStatus{{sysName="{sysname}", ifName="{if_name}"}}[{last_minute}m])"""
+        r = self.query(query=q)
+        if r.code == 0:
+            status_code = r.data[0]['value'][1]
+            if status_code == 1:
+                status = 'up'
+            else:
+                status = 'down'
+            return ReturnResponse(code=0, msg=f"{sysname} {if_name} 最近 {last_minute} 分钟端口状态为 {status}", data=status)
+        else:
+            return r
+
+    def insert_cronjob_run_status(self, 
+                                  app_type: Literal['alert', 'meraki', 'other']='other', 
+                                  app: str='', 
+                                  status_code: Literal[0, 1]=1, 
+                                  comment: str=None, 
+                                  schedule_interval: str=None, 
+                                  schedule_cron: str=None
+                                ) -> ReturnResponse:
+        labels = {
+            "app": app,
+            "env": self.env,
+        }
+        if app_type:
+            labels['app_type'] = app_type
+        if comment:
+            labels['comment'] = comment
+            
+        if schedule_interval:
+            labels['schedule_type'] = 'interval'
+            labels['schedule_interval'] = schedule_interval
+            
+        if schedule_cron:
+            labels['schedule_type'] = 'cron'
+            labels['schedule_cron'] = schedule_cron
+            
+        r = self.insert(metric_name="cronjob_run_status", labels=labels, value=status_code)
+        return r
+    
+    def insert_cronjob_duration_seconds(self, 
+                                        app_type: Literal['alert', 'meraki', 'other']='other', 
+                                        app: str='', 
+                                        duration_seconds: float=None, 
+                                        comment: str=None, 
+                                        schedule_interval: str=None, 
+                                        schedule_cron: str=None
+                                    ) -> ReturnResponse:
+        labels = {
+            "app": app,
+            "env": self.env
+        }
+        if app_type:
+            labels['app_type'] = app_type
+        if comment:
+            labels['comment'] = comment
+
+        if schedule_interval:
+            labels['schedule_type'] = 'interval'
+            labels['schedule_interval'] = schedule_interval
+            
+        if schedule_cron:
+            labels['schedule_type'] = 'cron'
+            labels['schedule_cron'] = schedule_cron
+        r = self.insert(metric_name="cronjob_run_duration_seconds", labels=labels, value=duration_seconds)
+        return r

{pytbox-0.1.5 → pytbox-0.1.7}/src/pytbox/log/logger.py

@@ -9,7 +9,7 @@ from ..database.mongo import Mongo
 from ..feishu.client import Client as FeishuClient
 from ..utils.timeutils import TimeUtils
 from ..dida365 import Dida365
-
+from ..alicloud.sls import AliCloudSls
 
 logger.remove()
 logger.add(sys.stdout, colorize=True, format="<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> | <level>{level: <8}</level> | <level>{message}</level>")
@@ -31,7 +31,6 @@ class AppLogger:
                  feishu: FeishuClient=None,
                  dida: Dida365=None,
                  enable_sls: bool=False,
-                 sls_url: str=None,
                  sls_access_key_id: str=None,
                  sls_access_key_secret: str=None,
                  sls_project: str=None,
@@ -49,10 +48,17 @@ class AppLogger:
         self.stream = stream
         self.victorialog = Victorialog(url=victorialog_url)
         self.enable_victorialog = enable_victorialog
+        self.enable_sls = enable_sls
         self.mongo = mongo
         self.feishu = feishu
         self.dida = dida
-    
+        self.sls = AliCloudSls(
+            access_key_id=sls_access_key_id,
+            access_key_secret=sls_access_key_secret,
+            project=sls_project,
+            logstore=sls_logstore
+        )
+
     def _get_caller_info(self) -> tuple[str, int, str]:
         """
         获取调用者信息
@@ -78,6 +84,9 @@ class AppLogger:
         logger.debug(f"[{caller_filename}:{caller_lineno}:{caller_function}] {message}")
         if self.enable_victorialog:
             self.victorialog.send_program_log(stream=self.stream, level="DEBUG", message=message, app_name=self.app_name, file_name=call_full_filename, line_number=caller_lineno, function_name=caller_function)
+        if self.enable_sls:
+            self.sls.put_logs(level="DEBUG", msg=message, app=self.app_name, caller_filename=caller_filename, caller_lineno=caller_lineno, caller_function=caller_function, call_full_filename=call_full_filename)
+ 
     
     def info(self, message: str='', feishu_notify: bool=False):
         """记录信息级别日志"""
@@ -85,22 +94,28 @@ class AppLogger:
         logger.info(f"[{caller_filename}:{caller_lineno}:{caller_function}] {message}")
         if self.enable_victorialog:
             r = self.victorialog.send_program_log(stream=self.stream, level="INFO", message=message, app_name=self.app_name, file_name=call_full_filename, line_number=caller_lineno, function_name=caller_function)
+        if self.enable_sls:
+            self.sls.put_logs(level="INFO", msg=message, app=self.app_name, caller_filename=caller_filename, caller_lineno=caller_lineno, caller_function=caller_function, call_full_filename=call_full_filename)
         if feishu_notify:
             self.feishu(message)
-    
+        
     def warning(self, message: str):
         """记录警告级别日志"""
         caller_filename, caller_lineno, caller_function, call_full_filename = self._get_caller_info()
         logger.warning(f"[{caller_filename}:{caller_lineno}:{caller_function}] {message}")
         if self.enable_victorialog:
             self.victorialog.send_program_log(stream=self.stream, level="WARN", message=message, app_name=self.app_name, file_name=call_full_filename, line_number=caller_lineno, function_name=caller_function)
-    
+        if self.enable_sls:
+            self.sls.put_logs(level="WARN", msg=message, app=self.app_name, caller_filename=caller_filename, caller_lineno=caller_lineno, caller_function=caller_function, call_full_filename=call_full_filename)
+            
     def error(self, message: str):
         """记录错误级别日志"""
         caller_filename, caller_lineno, caller_function, call_full_filename = self._get_caller_info()
         logger.error(f"[{caller_filename}:{caller_lineno}:{caller_function}] {message}")
         if self.enable_victorialog:
             self.victorialog.send_program_log(stream=self.stream, level="ERROR", message=message, app_name=self.app_name, file_name=call_full_filename, line_number=caller_lineno, function_name=caller_function)
+        if self.enable_sls:
+            self.sls.put_logs(level="ERROR", msg=message, app=self.app_name, caller_filename=caller_filename, caller_lineno=caller_lineno, caller_function=caller_function, call_full_filename=call_full_filename)
         
         if self.feishu:
             existing_message = self.mongo.collection.find_one({"message": message}, sort=[("time", -1)])
@@ -150,7 +165,8 @@ class AppLogger:
         logger.critical(f"[{caller_filename}:{caller_lineno}:{caller_function}] {message}")
         if self.enable_victorialog:
             self.victorialog.send_program_log(stream=self.stream, level="CRITICAL", message=message, app_name=self.app_name, file_name=call_full_filename, line_number=caller_lineno, function_name=caller_function)
-
+        if self.enable_sls:
+            self.sls.put_logs(level="CRITICAL", msg=message, app=self.app_name, caller_filename=caller_filename, caller_lineno=caller_lineno, caller_function=caller_function, call_full_filename=call_full_filename)
 
 # 使用示例
 if __name__ == "__main__":

pytbox-0.1.7/src/pytbox/utils/cronjob.py

@@ -0,0 +1,79 @@
+#!/usr/bin/env python3
+
+import time
+from functools import wraps
+from typing import Literal
+from ..database.victoriametrics import VictoriaMetrics
+
+
+def cronjob_counter(vm: VictoriaMetrics=None, log: str=None, app_type: Literal['']='', app='', comment=None, schedule_interval=None, schedule_cron=None):
+    """计算函数运行时间的装饰器，支持记录到 VictoriaMetrics
+    
+    Args:
+        app_type: 应用类型 ('alert', 'meraki', 'other')
+        app: 应用名称
+        comment: 备注信息
+        schedule_interval: 定时任务间隔（如 '1m', '5m'）
+        schedule_cron: cron 表达式（如 '0 */5 * * *'）
+    """
+    def decorator(func):
+        @wraps(func)
+        def wrapper(*args, **kwargs):
+            start_time = time.time()
+            
+            try:
+                result = func(*args, **kwargs)
+                end_time = time.time()
+                elapsed_time = end_time - start_time
+                
+                # 记录任务成功完成状态
+                vm.insert_cronjob_run_status(
+                    app_type=app_type,
+                    app=app,
+                    status_code=1,  # 0 表示成功完成
+                    comment=f"成功完成: {comment}" if comment else "任务成功完成",
+                    schedule_interval=schedule_interval,
+                    schedule_cron=schedule_cron
+                )
+                
+                # 记录执行耗时
+                vm.insert_cronjob_duration_seconds(
+                    app_type=app_type,
+                    app=app,
+                    duration_seconds=elapsed_time,
+                    comment=comment,
+                    schedule_interval=schedule_interval,
+                    schedule_cron=schedule_cron
+                )
+                log.info(f"{app} 任务成功完成, 耗时 {elapsed_time:.2f} 秒")
+                
+                return result
+                
+            except Exception as e:
+                end_time = time.time()
+                elapsed_time = end_time - start_time
+                
+                # 记录任务失败状态
+                error_comment = f"执行出错: {str(e)}" if not comment else f"{comment} (出错: {str(e)})"
+                vm.insert_cronjob_run_status(
+                    app_type=app_type,
+                    app=app,
+                    status_code=0,  # 1 表示失败
+                    comment=error_comment,
+                    schedule_interval=schedule_interval,
+                    schedule_cron=schedule_cron
+                )
+                
+                # 即使出错也记录耗时
+                vm.insert_cronjob_duration_seconds(
+                    app_type=app_type,
+                    app=app,
+                    duration_seconds=elapsed_time,
+                    comment=error_comment,
+                    schedule_interval=schedule_interval,
+                    schedule_cron=schedule_cron
+                )
+                log.error(f"任务失败: {error_comment}")
+                raise
+        return wrapper
+    return decorator

{pytbox-0.1.5 → pytbox-0.1.7/src/pytbox.egg-info}/PKG-INFO

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

{pytbox-0.1.5 → pytbox-0.1.7}/src/pytbox.egg-info/SOURCES.txt

@@ -64,16 +64,17 @@ src/pytbox/mail/alimail.py
 src/pytbox/mail/client.py
 src/pytbox/mail/mail_detail.py
 src/pytbox/network/meraki.py
+src/pytbox/utils/cronjob.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
 src/pytbox/utils/timeutils.py
 src/pytbox/win/ad.py
 tests/test_base.py
 tests/test_feishu.py
+tests/test_load_jsonfile.py
 tests/test_logger.py
 tests/test_onepassword_connect.py
 tests/test_onepassword_sa.py

pytbox-0.1.7/tests/test_load_jsonfile.py

@@ -0,0 +1,14 @@
+#!/usr/bin/env python3
+
+
+from pytbox.utils.load_config import load_config_by_file
+
+
+config = load_config_by_file(
+    path='/workspaces/pytbox/tests/alert/config_dev.toml',
+    # oc_vault_id="hcls5uxuq5dmxorw6rfewefdsa"
+    jsonfile="test_load_config.json"
+)
+
+
+print(config['json_test'])

pytbox-0.1.7/tests/test_logger.py

@@ -0,0 +1,20 @@
+#!/usr/bin/env python3
+
+from pytbox.base import get_logger_sls, get_logger
+
+
+log = get_logger('tests.test_logger')
+log_sls = get_logger_sls(app="tests.test_logger")
+
+def test_logger_info():
+    log.info('test_logger_info')
+    log.error('test error log2')
+
+
+def test_logger_sls():
+    log_sls.debug('test_logger_sls')
+    # log.error('test error log2')
+
+if __name__ == "__main__":
+    # test_logger_info()
+    test_logger_sls()

pytbox-0.1.7/tests/test_victoriametrics.py

@@ -0,0 +1,28 @@
+#!/usr/bin/env python3
+
+from pytbox.base import vm, get_cronjob_counter
+from pytbox.utils.cronjob import cronjob_counter
+
+
+def test_query():
+    r = vm.query('ping_average_response_ms')
+    print(r)
+
+def test_check_ping_result():
+    r = vm.check_ping_result(target='121.46.237.186', last_minute=10)
+    print(r)
+
+def test_get_labels():
+    r = vm.get_labels('ping_average_response_ms')
+    print(r)
+
+@get_cronjob_counter(app_type="tests", app="tests.test_victoriameterics", schedule_interval='5s')
+def test_check_snmp_port_status():
+    r = vm.check_snmp_port_status(sysname="shylf-prod-coresw-ce6820-182", if_name="10GE1/0/47", last_minute=10)
+    print(r)
+
+if __name__ == "__main__":
+    # test_get_labels()
+    # test_query()
+    # test_check_ping_result()
+    test_check_snmp_port_status()

pytbox-0.1.5/src/pytbox/database/victoriametrics.py

@@ -1,149 +0,0 @@
-#!/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:
-    
-    def __init__(self, url: str='', timeout: int=3) -> None:
-        self.url = url
-        self.timeout = timeout
-        self.session = requests.Session()
-        self.session.headers.update({
-            'Content-Type': 'application/json',
-            'Accept': 'application/json'
-        })
-        
-    def query(self, query: str, output_format: Literal['json']=None) -> ReturnResponse:
-        '''
-        查询指标数据
-
-        Args:
-            query (str): 查询语句
-
-        Returns:
-            dict: 查询结果
-        '''
-        url = f"{self.url}/prometheus/api/v1/query"
-        r = requests.get(
-            url, 
-            timeout=self.timeout,
-            params={"query": query}
-        )
-        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:
-                code = 2
-                msg = f"[{query}] 没有查询到结果"
-                data = res_json
-        else:
-            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}"
-        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, env: str='prod', dev_file: str='') -> ReturnResponse:
-        '''
-        检查ping结果
-
-        Args:
-            target (str): 目标地址
-            last_minute (int, optional): 最近多少分钟. Defaults to 10.
-            env (str, optional): 环境. Defaults to 'prod'.
-            dev_file (str, optional): 开发文件. Defaults to ''.
-
-        Returns:
-            ReturnResponse: 
-                code = 0 正常, code = 1 异常, code = 2 没有查询到数据, 建议将其判断为正常
-        '''
-        if target:
-            # 这里需要在字符串中保留 {}，同时插入 target，可以用双大括号转义
-            query = f"ping_result_code{{target='{target}'}}"
-        else:
-            query = "ping_result_code"
-        
-        if last_minute:
-            query = query + f"[{last_minute}m]"
-        
-        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} 分钟是异常的!"
-                else:
-                    code = 0
-                    msg = f"已检查 {target} 最近 {last_minute} 分钟是正常的!"
-        elif r.code == 2:
-            code = 2
-            msg = f"没有查询到 {target} 最近 {last_minute} 分钟的ping结果!"
-        
-        try:
-            data = r.data[0]
-        except KeyError:
-            data = r.data
-        
-        return ReturnResponse(code=code, msg=msg, data=data)
-
-    def check_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[0]['value'][1]
-        return int(float(rate))

pytbox-0.1.5/src/pytbox/utils/ping_checker.py

@@ -1 +0,0 @@
- 

pytbox-0.1.5/tests/test_logger.py

@@ -1,15 +0,0 @@
-#!/usr/bin/env python3
-
-from pytbox.base import get_logger
-
-
-
-log = get_logger('tests.test_logger')
-
-def test_logger_info():
-    log.info('test_logger_info')
-    log.error('test error log2')
-
-
-if __name__ == "__main__":
-    test_logger_info()

pytbox-0.1.5/tests/test_victoriametrics.py

@@ -1,22 +0,0 @@
-#!/usr/bin/env python3
-
-from pytbox.base import vm
-
-
-
-def test_query():
-    r = vm.query('ping_average_response_ms')
-    print(r)
-
-def test_check_ping_result():
-    r = vm.check_ping_result(target='121.46.237.186', last_minute=10)
-    print(r)
-
-def test_get_labels():
-    r = vm.get_labels('ping_average_response_ms')
-    print(r)
-
-if __name__ == "__main__":
-    # test_get_labels()
-    # test_query()
-    test_check_ping_result()