pytbox 0.1.9__tar.gz → 0.2.0__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytbox
-Version: 0.1.9
+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.9 → pytbox-0.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pytbox"
-version = "0.1.9"
+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.9 → pytbox-0.2.0}/src/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-0.1.9 → pytbox-0.2.0}/src/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-0.1.9 → pytbox-0.2.0}/src/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-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.9 → pytbox-0.2.0/src/pytbox.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytbox
-Version: 0.1.9
+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.9 → 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