pytbox 0.1.6__py3-none-any.whl → 0.1.8__py3-none-any.whl

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/base.py

@@ -16,6 +16,7 @@ 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'))
@@ -107,4 +108,8 @@ sls = AliCloudSls(
     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/database/victoriametrics.py

@@ -1,7 +1,8 @@
 #!/usr/bin/env python3
 
+import time
 import json
-from typing import Literal, Optional
+from typing import Literal, Optional, Dict, List
 import requests
 from ..utils.response import ReturnResponse
 from ..utils.load_vm_devfile import load_dev_file
@@ -9,7 +10,7 @@ from ..utils.load_vm_devfile import load_dev_file
 
 class VictoriaMetrics:
     
-    def __init__(self, url: str='', timeout: int=3) -> None:
+    def __init__(self, url: str='', timeout: int=3, env: str='prod') -> None:
         self.url = url
         self.timeout = timeout
         self.session = requests.Session()
@@ -17,7 +18,44 @@ class VictoriaMetrics:
             '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:
         '''
         查询指标数据
@@ -208,9 +246,67 @@ class VictoriaMetrics:
         '''
         q = f"""avg_over_time(snmp_interface_ifOperStatus{{sysName="{sysname}", ifName="{if_name}"}}[{last_minute}m])"""
         r = self.query(query=q)
-        status_code = r.data[0]['value'][1]
-        if status_code == 1:
-            status = 'up'
+        if r.code == 0:
+            status_code = int(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:
-            status = 'down'
-        return ReturnResponse(code=0, msg=f"{sysname} {if_name} 最近 {last_minute} 分钟端口状态为 {status}", data=status)
+            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/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.6.dist-info → pytbox-0.1.8.dist-info}/METADATA

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

{pytbox-0.1.6.dist-info → pytbox-0.1.8.dist-info}/RECORD

@@ -1,4 +1,4 @@
-pytbox/base.py,sha256=7wUhNSX8sbZ5hlgWG7BhFYtNOhVB6b-k3ULNA0vPchU,3948
+pytbox/base.py,sha256=uzrv0vlHVPOr2JjpUGd8R2Wr-iRf6SysnrAYOalKXYY,4283
 pytbox/cli.py,sha256=N775a0GK80IT2lQC2KRYtkZpIiu9UjavZmaxgNUgJhQ,160
 pytbox/dida365.py,sha256=pUMPB9AyLZpTTbaz2LbtzdEpyjvuGf4YlRrCvM5sbJo,10545
 pytbox/excel.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -7,7 +7,7 @@ pytbox/onepassword_sa.py,sha256=08iUcYud3aEHuQcUsem9bWNxdXKgaxFbMy9yvtr-DZQ,6995
 pytbox/pyjira.py,sha256=TMy34Rtu7OXRA8wpUuLsFeyIQfRNUn2ed2no00L8YSE,22470
 pytbox/vmware.py,sha256=WiH67_3-VCBjXJuh3UueOc31BdZDItiZhkeuPzoRhw4,3975
 pytbox/alert/alert_handler.py,sha256=FePPQS4LyGphSJ0QMv0_pLWaXxEqsRlcTKMfUjtsNfk,5048
-pytbox/alert/ping.py,sha256=g36X0U3U8ndZqfpVIcuoxJJ0X5gST3I_IwjTQC1roHA,779
+pytbox/alert/ping.py,sha256=KEnnXdIRJHvR_rEHPWLBt0wz4cGwmA29Lenlak3Z_1Y,778
 pytbox/alicloud/sls.py,sha256=-r6rbCwDUQ4jwAgSVSNu7B2h1MNg20CrRIKnlXVeY9w,4159
 pytbox/categraf/build_config.py,sha256=9G85rLqkz3lchpH7ef0LbvckYHl0nRA6mHVLeUfs9Mw,6308
 pytbox/categraf/instances.toml,sha256=jcJyEaqhohUcECczWArxUK4t0-rdk4vmrX21kxZlSLA,1254
@@ -43,7 +43,7 @@ pytbox/cli/formatters/__init__.py,sha256=4o85w4j-A-O1oBLvuE9q8AFiJ2C9rvB3MIKsy5V
 pytbox/cli/formatters/output.py,sha256=h5WhZlQk1rjmxEj88Jy5ODLcv6L5zfGUhks_3AWIkKU,5455
 pytbox/common/__init__.py,sha256=3JWfgCQZKZuSH5NCE7OCzKwq82pkyop9l7sH5YSNyfU,122
 pytbox/database/mongo.py,sha256=CSpHC7iR-M0BaVxXz5j6iXjMKPgXJX_G7MrjCj5Gm8Q,3478
-pytbox/database/victoriametrics.py,sha256=0QXkrnqL0rV7dGaHmtZus-MdEW3uX9ZPm0D-sS4L1DU,7606
+pytbox/database/victoriametrics.py,sha256=pKiAeZUTEXBegN8kGwkiGh1Pdm3i9DOnBhSG-la0mKk,11300
 pytbox/feishu/client.py,sha256=kwGLseGT_iQUFmSqpuS2_77WmxtHstD64nXvktuQ3B4,5865
 pytbox/feishu/endpoints.py,sha256=z3nPOZPC2JGDJlO7SusWBpRA33hZZ4Z-GBhI6F8L_u4,40240
 pytbox/feishu/errors.py,sha256=79qFAHZw7jDj3gnWAjI1-W4tB0q1_aSfdjee4xzXeuI,1179
@@ -55,16 +55,16 @@ pytbox/mail/alimail.py,sha256=njKA3PUbIaiKFaxKvUObmklmEEHg2YA-O5rpgsgT5_w,5147
 pytbox/mail/client.py,sha256=6HeKpChHGjTCYGBgQcfAhWlU_wh9wtO-bjP6TU38pGM,6120
 pytbox/mail/mail_detail.py,sha256=6u8DK-7WzYPSuX6TdicSCh2Os_9Ou6Rn9xc6WRvv85M,699
 pytbox/network/meraki.py,sha256=054E3C5KzAuXs9aPalvdAOUo6Hc5aOKZSWUaVbPquy4,6112
+pytbox/utils/cronjob.py,sha256=b17CY1fmaFTdQojicXAXHliov_JZdmT7cZhtO4v5sCo,3080
 pytbox/utils/env.py,sha256=gD2-NyL3K3Vg1B1eGeD1hRtlSHPGgF8Oi9mchuQL6_o,646
 pytbox/utils/load_config.py,sha256=R4pGerBinbewsym41hQ8Z-I5I7gepuEKODjIrli4C08,5043
 pytbox/utils/load_vm_devfile.py,sha256=GVbB-FvGb0l586SDaraz__ZaXyDWd1WxcXVw5xGfXWw,1328
-pytbox/utils/ping_checker.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 pytbox/utils/response.py,sha256=kXjlwt0WVmLRam2eu1shzX2cQ7ux4cCQryaPGYwle5g,1247
 pytbox/utils/richutils.py,sha256=OT9_q2Q1bthzB0g1GlhZVvM4ZAepJRKL6a_Vsr6vEqo,487
 pytbox/utils/timeutils.py,sha256=uSKgwt20mVcgIGKLsH2tNum8v3rcpzgmBibPvyPQFgM,20433
 pytbox/win/ad.py,sha256=-3pWfL3dElz-XoO4j4M9lrgu3KJtlhrS9gCWJBpafAU,1147
-pytbox-0.1.6.dist-info/METADATA,sha256=mxB7SNMpc2yyjOXg9nVuF9DdryE6gDfe54RIoGN4jWI,6319
-pytbox-0.1.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pytbox-0.1.6.dist-info/entry_points.txt,sha256=YaTOJ2oPjOiv2SZwY0UC-UA9QS2phRH1oMvxGnxO0Js,43
-pytbox-0.1.6.dist-info/top_level.txt,sha256=YADgWue-Oe128ptN3J2hS3GB0Ncc5uZaSUM3e1rwswE,7
-pytbox-0.1.6.dist-info/RECORD,,
+pytbox-0.1.8.dist-info/METADATA,sha256=iAknEyd73pj7U9J8ML0v1o9-q4AJvwD6OYSM4d91KAk,6319
+pytbox-0.1.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pytbox-0.1.8.dist-info/entry_points.txt,sha256=YaTOJ2oPjOiv2SZwY0UC-UA9QS2phRH1oMvxGnxO0Js,43
+pytbox-0.1.8.dist-info/top_level.txt,sha256=YADgWue-Oe128ptN3J2hS3GB0Ncc5uZaSUM3e1rwswE,7
+pytbox-0.1.8.dist-info/RECORD,,

pytbox/utils/ping_checker.py

@@ -1 +0,0 @@
-