pytbox 0.1.8__tar.gz → 0.2.0__tar.gz

{pytbox-0.1.8/src/pytbox.egg-info → pytbox-0.2.0}/PKG-INFO

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

{pytbox-0.1.8 → pytbox-0.2.0}/pyproject.toml

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

{pytbox-0.1.8 → pytbox-0.2.0}/src/pytbox/alert/alert_handler.py

@@ -7,6 +7,7 @@ from ..database.mongo import Mongo
 from ..feishu.client import Client as FeishuClient
 from ..dida365 import Dida365
 from ..utils.timeutils import TimeUtils
+from ..mail.client import MailClient
 
 
 class AlertHandler:
@@ -15,12 +16,17 @@ class AlertHandler:
                  config: dict=None,
                  mongo_client: Mongo=None,
                  feishu_client: FeishuClient=None,
-                 dida_client: Dida365=None
+                 dida_client: Dida365=None,
+                 mail_client: MailClient=None,
+                 env: Literal['dev', 'prod']='prod'
             ):
+     
         self.config = config
         self.mongo = mongo_client
         self.feishu = feishu_client
         self.dida = dida_client
+        self.mail = mail_client
+        self.env = env
 
     def send_alert(self, 
                  event_id: str=None,
@@ -34,7 +40,7 @@ class AlertHandler:
                  suggestion: str='',
                  troubleshot: str='暂无',
                  mongo_id: str=None
-                ):
+            ):
         
         if not event_id:
             event_id = str(uuid.uuid4())
@@ -62,11 +68,11 @@ class AlertHandler:
                 update = {"$set": { "resolved_time": event_time}}
                 self.mongo.collection.update_one(filter_doc, update)
                 alarm_time = self.mongo.collection.find_one(filter_doc, {'event_time': 1})['event_time']
-    
+            
             content = [
                 f'**事件名称**: {event_name}',
                 f'**告警时间**: {TimeUtils.convert_timeobj_to_str(timeobj=event_time, timezone_offset=0) if event_type == "trigger" else TimeUtils.convert_timeobj_to_str(timeobj=alarm_time, timezone_offset=8)}',
-                f'**事件内容**: {event_content}',
+                f'**事件内容**: {event_content + " 已恢复" if event_type == "resolved" else event_content}',
                 f'**告警实例**: {entity_name}',
                 f'**建议**: {suggestion}',
                 f'**故障排查**: {troubleshot}',
@@ -78,17 +84,31 @@ class AlertHandler:
     
             if self.config['feishu']['enable_alert']:
                 self.feishu.extensions.send_message_notify(
+                    receive_id=self.config['feishu']['receive_id'],
                     color='red' if event_type == "trigger" else 'green',
-                    title=event_content,
+                    title=event_content + " 已恢复" if event_type == "resolved" else event_content,
                     priority=priority,
-                    sub_title="",
+                    sub_title='测试告警, 无需处理' if self.env == 'dev' else '',
                     content='\n'.join(content)
                 )
             
+            if self.config['mail']['enable_mail']:
+                if event_type == "trigger":
+                    self.mail.send_mail(
+                        receiver=[self.config['mail']['mail_address']],
+                        subject=f"{self.config['mail']['subject_trigger']}, {event_content}",
+                        contents=f"event_content:{event_content}, alarm_time: {str(event_time)}, event_id: {event_id}, alarm_name: {event_name}, entity_name: {entity_name}, priority: {priority}, automate_ts: {troubleshot}, suggestion: {suggestion}"
+                    )
+                else:
+                    self.mail.send_mail(
+                        receiver=[self.config['mail']['mail_address']],
+                        subject=f"{self.config['mail']['subject_resolved']}, {event_content}",
+                        contents=f"event_content:{event_content}, alarm_time: {str(TimeUtils.get_now_time_mongo())}, event_id: {event_id}, alarm_name: {event_name}, entity_name: {entity_name}, priority: {priority}, automate_ts: {troubleshot}, suggestion: {suggestion}"
+                    )
+            
             if self.config['dida']['enable_alert']:
                 if event_type == "trigger":
                     res = self.dida.task_create(
-                        
                         project_id=self.config['dida']['alert_project_id'],
                         title=event_content,
                         content='\n'.join(content),

{pytbox-0.1.8 → pytbox-0.2.0}/src/pytbox/base.py

@@ -8,7 +8,6 @@ from pytbox.feishu.client import Client as FeishuClient
 from pytbox.dida365 import Dida365
 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
@@ -44,6 +43,7 @@ dida = Dida365(
     access_token=config['dida']['access_token']
 )
 
+
 alert_handler = AlertHandler(config=config, mongo_client=get_mongo('alert_test'), feishu_client=feishu, dida_client=dida)
 
 def get_logger(app):
@@ -99,9 +99,9 @@ pyjira = PyJira(
     token=config['jira']['token']
 )
 
-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'])
+# 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'])
 
 sls = AliCloudSls(
     access_key_id=config['alicloud']['account1']['access_key_id'],
@@ -110,6 +110,5 @@ sls = AliCloudSls(
     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.8 → pytbox-0.2.0}/src/pytbox/database/mongo.py

@@ -80,7 +80,7 @@ class Mongo:
         alarm_list = []
         for result in results:
             duration_minute = '持续 ' + str(int((result['resolved_time'] - result['event_time']).total_seconds() / 60)) + ' 分钟'
-            alarm_list.append('触发告警: ' + TimeUtils.convert_timeobj_to_str(timeobj=result['event_time']) + ' ' + duration_minute)
+            alarm_list.append('触发时间: ' + TimeUtils.convert_timeobj_to_str(timeobj=result['event_time']) + ' ' + duration_minute)
 
         alarm_str = '\n'.join(alarm_list)
         

{pytbox-0.1.8 → pytbox-0.2.0}/src/pytbox/database/victoriametrics.py

@@ -230,7 +230,38 @@ class VictoriaMetrics:
         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:
+    def check_interface_avg_rate(self,
+                                 direction: Literal['in', 'out'],
+                                 sysname: str, 
+                                 ifname:str, 
+                                 last_hours: Optional[int] = 24,
+                                 last_minutes: Optional[int] = 5,
+                                ) -> ReturnResponse:
+        '''
+        _summary_
+
+        Args:
+            direction (Literal['in', 'out']): _description_
+            sysname (str): _description_
+            ifname (str): _description_
+            last_hours (Optional[int], optional): _description_. Defaults to 24.
+            last_minutes (Optional[int], optional): _description_. Defaults to 5.
+
+        Returns:
+            ReturnResponse: _description_
+        '''
+        if direction == 'in':
+            query = f'avg_over_time(rate(snmp_interface_ifHCInOctets{{sysName="{sysname}", ifName="{ifname}"}}[{last_minutes}m]) * 8 [{last_hours}h:]) / 1e6'
+        else:
+            query = f'avg_over_time(rate(snmp_interface_ifHCOutOctets{{sysName="{sysname}", ifName="{ifname}"}}[{last_minutes}m]) * 8 [{last_hours}h:]) / 1e6'
+        r = self.query(query)
+        try:
+            rate = r.data[0]['value'][1]
+            return ReturnResponse(code=0, msg=f"查询 {sysname} {ifname} 最近 {last_hours} 小时平均速率为 {round(float(rate), 2)} Mbit/s", data=round(float(rate), 2))
+        except KeyError:
+            return ReturnResponse(code=1, msg=f"查询 {sysname} {ifname} 最近 {last_hours} 小时平均速率为 0 Mbit/s")
+    
+    def check_snmp_port_status(self, sysname: str=None, if_name: str=None, last_minute: int=5, dev_file: str=None) -> ReturnResponse:
         '''
         查询端口状态
         status code 可参考 SNMP 文件 https://mibbrowser.online/mibdb_search.php?mib=IF-MIB
@@ -245,7 +276,10 @@ class VictoriaMetrics:
             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 self.env == 'dev':
+            r = load_dev_file(dev_file)
+        else:
+            r = self.query(query=q)
         if r.code == 0:
             status_code = int(r.data[0]['value'][1])
             if status_code == 1:

pytbox-0.2.0/src/pytbox/excel.py

@@ -0,0 +1,64 @@
+#!/usr/bin/env python3
+
+from openpyxl.styles import Alignment
+from openpyxl.styles import PatternFill
+from openpyxl.styles.differential import DifferentialStyle
+from openpyxl.styles import Font
+
+
+class ExcelFormat:
+    def __init__(self, ws):
+        self.ws = ws
+        
+    def set_column(self):
+        for column in self.ws.columns:
+            max_length = 0
+            column = [cell for cell in column]
+            for cell in column:
+                try:
+                    if len(str(cell.value)) > max_length:
+                        max_length = len(cell.value)
+                except:
+                    pass
+            adjusted_width = (max_length + 10)
+            self.ws.column_dimensions[column[0].column_letter].width = adjusted_width
+            
+    def set_rows_center(self):
+        # 将所有单元格的文字居中
+        for row in self.ws.iter_rows():
+            
+            for cell in row:
+                cell.alignment = Alignment(horizontal='center', vertical='center')
+        for row in self.ws.iter_rows():
+            self.ws.row_dimensions[row[0].row].height = 24
+    
+    def set_freeze_first_row(self):
+        """设置首行锁定/冻结首行"""
+        # 冻结首行，从第二行开始滚动
+        self.ws.freeze_panes = 'A2'
+    
+    def set_freeze_first_column(self):
+        """设置首列锁定/冻结首列"""
+        # 冻结首列，从第二列开始滚动
+        self.ws.freeze_panes = 'B1'
+    
+    def set_first_row_bold_color(self, font_color='FF0000FF'):
+        """设置首行字体加粗并改变颜色
+        
+        Args:
+            font_color (str): 字体颜色的十六进制代码，默认为蓝色(FF0000FF)
+                            格式：AARRGGBB 或 RRGGBB
+                            例如：'FF0000FF'(蓝色), 'FFFF0000'(红色), 'FF008000'(绿色)
+        """
+        # 遍历首行的所有单元格
+        for cell in self.ws[1]:
+            if cell.value is not None:  # 只对有内容的单元格设置样式
+                # 设置字体为粗体并改变颜色
+                cell.font = Font(bold=True, color=font_color)
+    
+    def set_freeze_first_row_and_column(self):
+        """同时冻结首行和首列"""
+        # 冻结首行首列，从第二行第二列开始滚动
+        self.ws.freeze_panes = 'B2'
+    
+    

{pytbox-0.1.8 → pytbox-0.2.0}/src/pytbox/mail/client.py

@@ -27,6 +27,10 @@ class MailClient:
         elif 'foxmail.com' in mail_address:
             self.smtp_address = 'smtp.qq.com'
             self.imap_address = 'imap.qq.com'
+        
+        elif 'mail' in mail_address and 'cn' in mail_address:
+            self.smtp_address = "smtpdm.aliyun.com"
+            self.imap_address = ""
             
         else:
             raise ValueError(f'不支持的邮箱地址: {mail_address}')

{pytbox-0.1.8 → pytbox-0.2.0/src/pytbox.egg-info}/PKG-INFO

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

{pytbox-0.1.8 → pytbox-0.2.0}/tests/test_victoriametrics.py

@@ -21,8 +21,14 @@ 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)
 
+
+r = vm.check_interface_avg_rate(direction='in', sysname='whcq-prod-coresw-s6720-254', ifname='XGigabitEthernet0/0/47', last_hours=24, last_minutes=5)
+print(r)
+
 if __name__ == "__main__":
     # test_get_labels()
     # test_query()
     # test_check_ping_result()
-    test_check_snmp_port_status()
+    # test_check_snmp_port_status()
+    # test_check_interface_avg_rate()
+    pass