pytbox 0.1.9__py3-none-any.whl → 0.2.1__py3-none-any.whl

pytbox/alert/alert_handler.py

@@ -20,6 +20,7 @@ class AlertHandler:
                  mail_client: MailClient=None,
                  env: Literal['dev', 'prod']='prod'
             ):
+     
         self.config = config
         self.mongo = mongo_client
         self.feishu = feishu_client
@@ -95,14 +96,15 @@ class AlertHandler:
                 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}'
+                        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}')
+                        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":

pytbox/base.py

@@ -43,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):
@@ -98,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'],

pytbox/database/victoriametrics.py

@@ -230,6 +230,37 @@ class VictoriaMetrics:
         rate = r.data[0]['value'][1]
         return int(float(rate))
     
+    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:
         '''
         查询端口状态

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/log/logger.py

@@ -152,12 +152,13 @@ class AppLogger:
                     f"**function_name**: {caller_function}"
                 ]
                 
-                self.dida.task_create(
-                    project_id="65e87d2b3e73517c2cdd9d63",
-                    title=f"自动化脚本告警: {self.app_name}",
-                    content="\n".join(dida_content_list),
-                    tags=['L-程序告警', 't-问题处理']
-                )
+                if self.dida:
+                    self.dida.task_create(
+                        project_id="65e87d2b3e73517c2cdd9d63",
+                        title=f"自动化脚本告警: {self.app_name}",
+                        content="\n".join(dida_content_list),
+                        tags=['L-程序告警', 't-问题处理']
+                    )
         
     def critical(self, message: str):
         """记录严重错误级别日志"""

{pytbox-0.1.9.dist-info → pytbox-0.2.1.dist-info}/METADATA

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

{pytbox-0.1.9.dist-info → pytbox-0.2.1.dist-info}/RECORD

@@ -1,12 +1,12 @@
-pytbox/base.py,sha256=wc2CTbICGBNvS0mBU3pRIHrIGaFdjf3Y5DIkux-_CmU,4247
+pytbox/base.py,sha256=VAAaf081rTih9_FjOvAvcihbXp-zwmbGvASnoB2mNZo,4254
 pytbox/cli.py,sha256=N775a0GK80IT2lQC2KRYtkZpIiu9UjavZmaxgNUgJhQ,160
 pytbox/dida365.py,sha256=pUMPB9AyLZpTTbaz2LbtzdEpyjvuGf4YlRrCvM5sbJo,10545
-pytbox/excel.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pytbox/excel.py,sha256=f5XBLCeJbGgxytoSwVhbk03WLTzz8Q3IJ_RZ2-r_w6A,2334
 pytbox/onepassword_connect.py,sha256=nD3xTl1ykQ4ct_dCRRF138gXCtk-phPfKYXuOn-P7Z8,3064
 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=mDg1bv9F9XOH0NNNZTlw_KaN_P4G3w8rswmDW_N_3gw,6553
+pytbox/alert/alert_handler.py,sha256=WCn4cKahv5G5BFGgmc7dX7BQ38h2kxTgxfRVTwc1O2M,6579
 pytbox/alert/ping.py,sha256=KEnnXdIRJHvR_rEHPWLBt0wz4cGwmA29Lenlak3Z_1Y,778
 pytbox/alicloud/sls.py,sha256=-r6rbCwDUQ4jwAgSVSNu7B2h1MNg20CrRIKnlXVeY9w,4159
 pytbox/categraf/build_config.py,sha256=9G85rLqkz3lchpH7ef0LbvckYHl0nRA6mHVLeUfs9Mw,6308
@@ -43,13 +43,13 @@ 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=AhJ9nCAQHKrrcL-ujeonOwEf3x2QkmT2VhoCdglqJmU,3478
-pytbox/database/victoriametrics.py,sha256=k6D6SGi7ky5pugM8iHa4ZjzTIlcnxwFw178VA6Wjbyk,11408
+pytbox/database/victoriametrics.py,sha256=othAjqn7Cmgc-cp-TK-LIX5fGLT6fkgn6RVr4b4e2Mg,13007
 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
 pytbox/feishu/helpers.py,sha256=jhSkHiUw4822QBXx2Jw8AksogZdakZ-3QqvC3lB3qEI,201
 pytbox/feishu/typing.py,sha256=3hWkJgOi-v2bt9viMxkyvNHsPgrbAa0aZOxsZYg2vdM,122
-pytbox/log/logger.py,sha256=tfqRc7VcDyI_U-NiiK4euFe8O2ro9yeXVohPmzAC_hY,8867
+pytbox/log/logger.py,sha256=t5LEvuYTDixh67JOyX0ke1ukb9kT5PEaUSsQnDDck90,8921
 pytbox/log/victorialog.py,sha256=gffEiq38adv9sC5oZeMcyKghd3SGfRuqtZOFuqHQF6E,4139
 pytbox/mail/alimail.py,sha256=njKA3PUbIaiKFaxKvUObmklmEEHg2YA-O5rpgsgT5_w,5147
 pytbox/mail/client.py,sha256=RNgWhhTXFTpD43U4p7hbmnfRdmltuZmbm890gaZTzhI,6278
@@ -63,8 +63,8 @@ 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.9.dist-info/METADATA,sha256=BrT45GpS_cUh5Xz1ClDKgHl58mcsFulmXfVzgMCAsrc,6319
-pytbox-0.1.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pytbox-0.1.9.dist-info/entry_points.txt,sha256=YaTOJ2oPjOiv2SZwY0UC-UA9QS2phRH1oMvxGnxO0Js,43
-pytbox-0.1.9.dist-info/top_level.txt,sha256=YADgWue-Oe128ptN3J2hS3GB0Ncc5uZaSUM3e1rwswE,7
-pytbox-0.1.9.dist-info/RECORD,,
+pytbox-0.2.1.dist-info/METADATA,sha256=pkJirfIq1qpvhesBTwKN49MGnAikmQsWoPUHGvswZH0,6319
+pytbox-0.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pytbox-0.2.1.dist-info/entry_points.txt,sha256=YaTOJ2oPjOiv2SZwY0UC-UA9QS2phRH1oMvxGnxO0Js,43
+pytbox-0.2.1.dist-info/top_level.txt,sha256=YADgWue-Oe128ptN3J2hS3GB0Ncc5uZaSUM3e1rwswE,7
+pytbox-0.2.1.dist-info/RECORD,,