pytbox 0.2.4__tar.gz → 0.2.6__tar.gz

{pytbox-0.2.4/src/pytbox.egg-info → pytbox-0.2.6}/PKG-INFO

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

{pytbox-0.2.4 → pytbox-0.2.6}/pyproject.toml

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

{pytbox-0.2.4 → pytbox-0.2.6}/src/pytbox/base.py

@@ -16,6 +16,8 @@ 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
+from pytbox.notion import Notion
+from pytbox.mingdao import Mingdao
 
 
 config = load_config_by_file(path='/workspaces/pytbox/tests/alert/config_dev.toml', oc_vault_id=os.environ.get('oc_vault_id'))
@@ -96,6 +98,7 @@ vmware_test = VMwareClient(
 
 pyjira = PyJira(
     base_url=config['jira']['base_url'],
+    username=config['jira']['username'],
     token=config['jira']['token']
 )
 
@@ -111,4 +114,8 @@ sls = AliCloudSls(
 )
 
 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)
+    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)
+
+
+notion = Notion(token=config['notion']['api_secrets'], proxy=config['notion']['proxy'])
+mingdao = Mingdao(app_key=config['mingdao']['app_key'], sign=config['mingdao']['sign'])

{pytbox-0.2.4 → pytbox-0.2.6}/src/pytbox/database/victoriametrics.py

@@ -168,41 +168,19 @@ class VictoriaMetrics:
             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':
+        query = f'avg_over_time((ping_result_code{{target="{target}"}})[{last_minute}m])'
+        if self.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} 分钟是正常的!"
+            value = r.data[0]['value'][1]
+            if value == '0':
+                return ReturnResponse(code=0, msg=f"已检查 {target} 最近 {last_minute} 分钟是正常的!", data=r.data)
             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)
+                return ReturnResponse(code=1, msg=f"已检查 {target} 最近 {last_minute} 分钟是异常的!", data=r.data)
+        else:
+            return r
 
     def check_unreachable_ping_result(self, dev_file: str='') -> ReturnResponse:
         '''
@@ -396,4 +374,30 @@ class VictoriaMetrics:
             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
+        return r
+    
+    def get_vmware_esxhostnames(self, vcenter: str=None) -> list:
+        '''
+        _summary_
+        '''
+        esxhostnames = []
+        query = f'vsphere_host_sys_uptime_latest{{vcenter="{vcenter}"}}'
+        metrics = self.query(query=query).data
+        for metric in metrics:
+            esxhostname = metric['metric']['esxhostname']
+            esxhostnames.append(esxhostname)
+        return esxhostnames
+    
+    def get_vmware_cpu_usage(self, vcenter: str=None, esxhostname: str=None) -> float:
+        '''
+        _summary_
+        '''
+        query = f'vsphere_host_cpu_usage_average{{vcenter="{vcenter}", esxhostname="{esxhostname}"}}'
+        return self.query(query=query).data[0]['value'][1]
+    
+    def get_vmware_memory_usage(self, vcenter: str=None, esxhostname: str=None) -> float:
+        '''
+        _summary_
+        '''
+        query = f'vsphere_host_mem_usage_average{{vcenter="{vcenter}", esxhostname="{esxhostname}"}}'
+        return self.query(query=query).data[0]['value'][1]

pytbox-0.2.6/src/pytbox/mingdao.py

@@ -0,0 +1,164 @@
+#!/usr/bin/env python3
+
+from typing import Literal, Type
+
+import requests
+
+from .utils.response import ReturnResponse
+
+
+class Mingdao:
+    '''
+    _summary_
+    '''
+    def __init__(self, app_key: str=None, sign: str=None, timeout: int=5):
+        self.base_url = "https://api.mingdao.com"
+        self.headers = {
+            'Content-Type': 'application/json;charset=UTF-8',
+            'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/113.0.0.0 Safari/537.36'
+        }
+        self.timeout = timeout
+        self.app_key = app_key
+        self.sign = sign
+
+    def _build_api_request(self, api_url: str, method: Literal['GET', 'POST'], params: dict=None, body: dict=None, api_version: Literal['v1', 'v2']='v2'):
+        body['appKey'] = self.app_key
+        body['sign'] = self.sign
+        if not api_url.startswith('/'):
+            url = f'{self.base_url}/{api_version}/{api_url}'
+        else:
+            url = f'{self.base_url}/{api_version}{api_url}'
+        params = {
+            "appKey": self.app_key,
+            "sign": self.sign,
+        }
+        return requests.request(method, url, params=params, headers=self.headers, json=body, timeout=self.timeout)
+    
+    def get_app_info(self) -> ReturnResponse:
+        '''
+        _summary_
+
+        Returns:
+            ReturnResponse: _description_
+        '''
+        r = self._build_api_request(api_url='/open/app/get', method='GET', body={}, api_version='v1')
+        return ReturnResponse(code=0, msg='获取应用信息成功', data=r.json())
+    
+    def get_work_sheet_info(self, worksheet_id: str=None, table_name: str=None, worksheet_name: str=None):
+        if worksheet_name:
+            worksheet_id = self.get_work_sheet_id_by_name(table_name=table_name, worksheet_name=worksheet_name)
+        
+        r = self._build_api_request(
+            api_url='open/worksheet/getWorksheetInfo',
+            method='POST',
+            body={
+                "worksheetId": worksheet_id,
+            }
+        )
+        return r.json()
+    
+    def get_project_info(self, worksheet_id: str, keywords: str):
+        r = self._build_api_request(
+            api_url='open/worksheet/getFilterRows',
+            method='POST',
+            body={
+                "pageIndex": 1,
+                "pageSize": 100,
+                "worksheetId": worksheet_id,
+                "keyWords": keywords,
+            }
+        )
+        return r.json()
+    
+    def get_work_sheet_id_by_name(self, table_name: str, worksheet_name: str, child_section: bool=False):
+        r = self.get_app_info()        
+        for i in r.data['data']['sections']:
+            if table_name == i['name']:
+                if child_section:
+                    for child in i['childSections'][0]['items']:
+                        if child['name'] == worksheet_name:
+                            return child['id']
+                else:
+                    for item in i['items']:
+                        if item['name'] == worksheet_name:
+                            return item['id']
+
+    def get_control_id(self, table_name: str=None, worksheet_name: str=None, control_name: str=None):
+        r = self.get_work_sheet_info(table_name=table_name, worksheet_name=worksheet_name)
+        for control in r['data']['controls']:
+            if control['controlName'] == control_name:
+                return control['controlId']
+        return None
+    
+    def get_value(self, table_name: str=None, worksheet_name: str=None, control_name: str=None, value_name: str=None):
+        control_id = self.get_control_id(table_name=table_name, worksheet_name=worksheet_name, control_name=control_name)
+  
+        r = self._build_api_request(
+            api_url='open/worksheet/getFilterRows',
+            method='POST',
+            body={
+                "pageIndex": 1,
+                "pageSize": 100,
+                "worksheetId": self.get_work_sheet_id_by_name(table_name=table_name, worksheet_name=worksheet_name),
+                # "filters": filters
+            }
+        )
+        for row in r.json()['data']['rows']:
+            if row[control_id] == value_name:
+                return row['rowid']
+
+    def get_work_record(self, 
+                        worksheet_id: str=None,
+                        project_control_id: str=None,
+                        project_value: str=None,
+                        complete_date_control_id: str=None,
+                        complete_date: Literal['Today', '上个月', 'Last7Day', 'Last30Day']=None,
+                        parse_control_id: bool=False,
+                        page_size: int=100,
+                    ):
+        
+        filters = []
+        if project_value:
+            filters.append({
+                "controlId": project_control_id,
+                "dataType": 29,
+                "spliceType": 1,
+                "filterType": 24,
+                "dateRange": 0,
+                "dateRangeType": 0,
+                "value": "",
+                "values": [
+                    project_value
+                ],
+                "minValue": "",
+                "maxValue": ""
+            })
+        
+        if complete_date:
+            if complete_date == '上个月':
+                data_range = 8
+            elif complete_date == 'Today':
+                data_range = 1
+            elif complete_date == 'Last7Day':
+                data_range = 21
+            else:
+                data_range = 1
+            
+            filters.append({
+                "controlId": complete_date_control_id,
+                "dataType": 15,
+                "spliceType": 1,
+                "filterType": 17,
+                "dateRange": data_range,
+            })
+        r = self._build_api_request(
+            api_url='open/worksheet/getFilterRows',
+            method='POST',
+            body={
+                "pageIndex": 1,
+                "pageSize": page_size,
+                "worksheetId": worksheet_id,
+                "filters": filters
+            }
+        )
+        return r.json()