pytbox 0.0.7__py3-none-any.whl → 0.3.2__py3-none-any.whl

pytbox/cli/main.py

@@ -5,6 +5,7 @@ Pytbox 主命令行入口
 
 import click
 from .categraf import categraf_group
+from .commands.vm import vm_group
 
 
 @click.group()
@@ -16,6 +17,7 @@ def main():
 
 # 注册子命令组
 main.add_command(categraf_group, name='categraf')
+main.add_command(vm_group, name='vm')
 
 
 if __name__ == "__main__":

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/database/victoriametrics.py

@@ -1,13 +1,16 @@
 #!/usr/bin/env python3
 
-from typing import Literal, Optional
+import time
+import json
+from typing import Literal, Optional, Dict, List
 import requests
 from ..utils.response import ReturnResponse
+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()
@@ -15,14 +18,51 @@ 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:
+        """插入指标数据。
         
-    def query(self, query: str) -> 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:
         '''
         查询指标数据
-        
+
         Args:
             query (str): 查询语句
-            
+
         Returns:
             dict: 查询结果
         '''
@@ -32,14 +72,79 @@ class VictoriaMetrics:
             timeout=self.timeout,
             params={"query": query}
         )
-        if r.json().get("status") == "success":
-            if r.json()['data']['result']:
-                return ReturnResponse(code=0, msg=f"[{query}] 查询成功!", data=r.json()['data']['result'])
+        res_json = r.json()
+        status = res_json.get("status")
+        result = res_json.get("data", {}).get("result", [])
+        is_json = output_format == 'json'
+
+        if status == "success":
+            if result:
+                code = 0
+                msg = f"[{query}] 查询成功!"
+                data = result
             else:
-                return ReturnResponse(code=2, msg=f"[{query}] 没有查询到结果", data=r.json())
+                code = 2
+                msg = f"[{query}] 没有查询到结果"
+                data = res_json
+        else:
+            code = 1
+            msg = f"[{query}] 查询失败: {res_json.get('error')}"
+            data = res_json
+
+        resp = ReturnResponse(code=code, msg=msg, data=data)
+
+        if is_json:
+            json_result = json.dumps(resp.__dict__, ensure_ascii=False)
+            return json_result
         else:
-            return ReturnResponse(code=1, msg=f"[{query}] 查询失败: {r.json().get('error')}", data=r.json())
+            return resp
+
+    def query_range(self, query):
+        '''
+        查询指标数据
 
+        Args:
+            query (str): 查询语句
+
+        Returns:
+            dict: 查询结果
+        '''
+        url = f"{self.url}/prometheus/api/v1/query_range"
+
+        data = {
+            'query': query,
+            'start': '-1d',
+            'step': '1h'
+        }
+
+        r = requests.post(url, data=data, timeout=self.timeout)
+        res_json = r.json()
+        print(res_json)
+        # status = res_json.get("status")
+        # result = res_json.get("data", {}).get("result", [])
+        # is_json = output_format == 'json'
+
+        # if status == "success":
+        #     if result:
+        #         code = 0
+        #         msg = f"[{query}] 查询成功!"
+        #         data = result
+        #     else:
+        #         code = 2
+        #         msg = f"[{query}] 没有查询到结果"
+        #         data = res_json
+        # else:
+        #     code = 1
+        #     msg = f"[{query}] 查询失败: {res_json.get('error')}"
+        #     data = res_json
+
+        # resp = ReturnResponse(code=code, msg=msg, data=data)
+
+        # if is_json:
+        #     json_result = json.dumps(resp.__dict__, ensure_ascii=False)
+        #     return json_result
+        # else:
+        #     return resp
     def get_labels(self, metric_name: str) -> ReturnResponse:
         url = f"{self.url}/api/v1/series?match[]={metric_name}"
         response = requests.get(url, timeout=self.timeout)
@@ -49,42 +154,56 @@ class VictoriaMetrics:
         else:
             return ReturnResponse(code=1, msg=f"metric name: {metric_name} 查询失败")
 
-    def check_ping_result(self, target: str, last_minute: int=10) -> ReturnResponse:
+    def check_ping_result(self, target: str, last_minute: int=10, env: str='prod', dev_file: str='') -> ReturnResponse:
         '''
         检查ping结果
+
+        Args:
+            target (str): 目标地址
+            last_minute (int, optional): 最近多少分钟. Defaults to 10.
+            env (str, optional): 环境. Defaults to 'prod'.
+            dev_file (str, optional): 开发文件. Defaults to ''.
+
+        Returns:
+            ReturnResponse: 
+                code = 0 正常, code = 1 异常, code = 2 没有查询到数据, 建议将其判断为正常
         '''
-        if target:
-            # 这里需要在字符串中保留 {}，同时插入 target，可以用双大括号转义
-            query = f"ping_result_code{{target='{target}'}}"
+        query = f'min_over_time(ping_result_code{{target="{target}"}}[{last_minute}m])'
+        # query = f'avg_over_time((ping_result_code{{target="{target}"}})[{last_minute}m])'
+        if self.env == 'dev':
+            r = load_dev_file(dev_file)
         else:
-            query = "ping_result_code"
-        
-        if last_minute:
-            query = query + f"[{last_minute}m]"
-        
-        r = self.query(query=query)
+            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=1, msg=f"已检查 {target} 最近 {last_minute} 分钟是异常的!", data=r.data)
+        else:
+            return r
+
+    def check_unreachable_ping_result(self, dev_file: str='') -> ReturnResponse:
+        '''
+        检查ping结果
+
+        Args:
+            target (str): 目标地址
+            last_minute (int, optional): 最近多少分钟. Defaults to 10.
+            env (str, optional): 环境. Defaults to 'prod'.
+            dev_file (str, optional): 开发文件. Defaults to ''.
+
+        Returns:
+            ReturnResponse: 
+                code = 0 正常, code = 1 异常, code = 2 没有查询到数据, 建议将其判断为正常
+        '''
+        query = "ping_result_code == 1"
         
-        return ReturnResponse(code=code, msg=msg, data=data)
+        if self.env == 'dev':
+            r = load_dev_file(dev_file)
+        else:
+            r = self.query(query=query)
+        return r
 
     def check_interface_rate(self,
                              direction: Literal['in', 'out'],
@@ -110,4 +229,176 @@ class VictoriaMetrics:
             query = f'(rate(snmp_interface_ifHCOutOctets{{sysName="{sysName}", ifName="{ifName}"}}[{last_minutes}m])) * 8 / 1000000'
         r = self.query(query)
         rate = r.data[0]['value'][1]
-        return int(float(rate))
+        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_interface_max_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'max_over_time((rate(snmp_interface_ifHCInOctets{{sysName="{sysname}", ifName="{ifname}"}}[{last_minutes}m]) * 8) [{last_hours}h:]) / 1e6'
+        else:
+            query = f'max_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
+
+        Args:
+            sysname (_type_): 设备名称
+            if_name (_type_): _description_
+            last_minute (_type_): _description_
+
+        Returns:
+            ReturnResponse: 
+            code: 0, msg: , data: up,down
+        '''
+        q = f"""avg_over_time(snmp_interface_ifOperStatus{{sysName="{sysname}", ifName="{if_name}"}}[{last_minute}m])"""
+        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:
+                status = 'up'
+            else:
+                status = 'down'
+            return ReturnResponse(code=0, msg=f"{sysname} {if_name} 最近 {last_minute} 分钟端口状态为 {status}", data=status)
+        else:
+            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
+    
+    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/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/feishu/endpoints.py

@@ -254,9 +254,9 @@ class MessageEndpoint(Endpoint):
             body=payload
         )
         if r.code == 0:
-            return ReturnResponse(code=0, message=f"{message_id} 回复 emoji [{emoji_type}] 成功")
+            return ReturnResponse(code=0, msg=f"{message_id} 回复 emoji [{emoji_type}] 成功")
         else:
-            return ReturnResponse(code=1, message=f"{message_id} 回复 emoji [{emoji_type}] 失败")
+            return ReturnResponse(code=1, msg=f"{message_id} 回复 emoji [{emoji_type}] 失败")
 
 class BitableEndpoint(Endpoint):
     
@@ -432,10 +432,10 @@ class BitableEndpoint(Endpoint):
             resp = self.parent.request(path=f'/bitable/v1/apps/{app_token}/tables/{table_id}/records/{record_id}',
                                         method='PUT',
                                         body=payload)
-            return ReturnResponse(code=resp.code, message=f"记录已存在, 进行更新", data=resp.data)
+            return ReturnResponse(code=resp.code, msg=f"记录已存在, 进行更新", data=resp.data)
         else:
             resp = self.add_record(app_token, table_id, fields)
-            return ReturnResponse(code=resp.code, message=f"记录不存在, 进行创建", data=resp.data)
+            return ReturnResponse(code=resp.code, msg=f"记录不存在, 进行创建", data=resp.data)
 
     def query_name_by_record_id(self, app_token: str=None, table_id: str=None, field_names: list=None, record_id: str='', name: str=''):
         response = self.query_record(app_token=app_token, table_id=table_id, field_names=field_names)
@@ -967,9 +967,9 @@ class ExtensionsEndpoint(Endpoint):
                                    body=payload)
         if response.code == 0:
             if get == 'open_id':
-                return ReturnResponse(code=0, message=f'根据用户输入的 {user_input}, 获取用户信息成功', data=response.data['user_list'][0]['user_id'])
+                return ReturnResponse(code=0, msg=f'根据用户输入的 {user_input}, 获取用户信息成功', data=response.data['user_list'][0]['user_id'])
         else:
-            return ReturnResponse(code=response.code, message=f"获取时失败, 报错请见 data 字段", data=response.data)
+            return ReturnResponse(code=response.code, msg=f"获取时失败, 报错请见 data 字段", data=response.data)
     
     def format_rich_text(self, text: str, color: Literal['red', 'green', 'yellow', 'blue'], bold: bool=False):
         if bold:

pytbox/log/logger.py

@@ -9,7 +9,7 @@ from ..database.mongo import Mongo
 from ..feishu.client import Client as FeishuClient
 from ..utils.timeutils import TimeUtils
 from ..dida365 import Dida365
-
+from ..alicloud.sls import AliCloudSls
 
 logger.remove()
 logger.add(sys.stdout, colorize=True, format="<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> | <level>{level: <8}</level> | <level>{message}</level>")
@@ -31,7 +31,6 @@ class AppLogger:
                  feishu: FeishuClient=None,
                  dida: Dida365=None,
                  enable_sls: bool=False,
-                 sls_url: str=None,
                  sls_access_key_id: str=None,
                  sls_access_key_secret: str=None,
                  sls_project: str=None,
@@ -49,10 +48,17 @@ class AppLogger:
         self.stream = stream
         self.victorialog = Victorialog(url=victorialog_url)
         self.enable_victorialog = enable_victorialog
+        self.enable_sls = enable_sls
         self.mongo = mongo
         self.feishu = feishu
         self.dida = dida
-    
+        self.sls = AliCloudSls(
+            access_key_id=sls_access_key_id,
+            access_key_secret=sls_access_key_secret,
+            project=sls_project,
+            logstore=sls_logstore
+        )
+
     def _get_caller_info(self) -> tuple[str, int, str]:
         """
         获取调用者信息
@@ -78,6 +84,9 @@ class AppLogger:
         logger.debug(f"[{caller_filename}:{caller_lineno}:{caller_function}] {message}")
         if self.enable_victorialog:
             self.victorialog.send_program_log(stream=self.stream, level="DEBUG", message=message, app_name=self.app_name, file_name=call_full_filename, line_number=caller_lineno, function_name=caller_function)
+        if self.enable_sls:
+            self.sls.put_logs(level="DEBUG", msg=message, app=self.app_name, caller_filename=caller_filename, caller_lineno=caller_lineno, caller_function=caller_function, call_full_filename=call_full_filename)
+ 
     
     def info(self, message: str='', feishu_notify: bool=False):
         """记录信息级别日志"""
@@ -85,22 +94,31 @@ class AppLogger:
         logger.info(f"[{caller_filename}:{caller_lineno}:{caller_function}] {message}")
         if self.enable_victorialog:
             r = self.victorialog.send_program_log(stream=self.stream, level="INFO", message=message, app_name=self.app_name, file_name=call_full_filename, line_number=caller_lineno, function_name=caller_function)
+        if self.enable_sls:
+            self.sls.put_logs(level="INFO", msg=message, app=self.app_name, caller_filename=caller_filename, caller_lineno=caller_lineno, caller_function=caller_function, call_full_filename=call_full_filename)
         if feishu_notify:
-            self.feishu(message)
-    
+            self.feishu.extensions.send_message_notify(
+                title=f"自动化脚本告警: {self.app_name}",
+                content=message
+            )
+        
     def warning(self, message: str):
         """记录警告级别日志"""
         caller_filename, caller_lineno, caller_function, call_full_filename = self._get_caller_info()
         logger.warning(f"[{caller_filename}:{caller_lineno}:{caller_function}] {message}")
         if self.enable_victorialog:
             self.victorialog.send_program_log(stream=self.stream, level="WARN", message=message, app_name=self.app_name, file_name=call_full_filename, line_number=caller_lineno, function_name=caller_function)
-    
+        if self.enable_sls:
+            self.sls.put_logs(level="WARN", msg=message, app=self.app_name, caller_filename=caller_filename, caller_lineno=caller_lineno, caller_function=caller_function, call_full_filename=call_full_filename)
+            
     def error(self, message: str):
         """记录错误级别日志"""
         caller_filename, caller_lineno, caller_function, call_full_filename = self._get_caller_info()
         logger.error(f"[{caller_filename}:{caller_lineno}:{caller_function}] {message}")
         if self.enable_victorialog:
             self.victorialog.send_program_log(stream=self.stream, level="ERROR", message=message, app_name=self.app_name, file_name=call_full_filename, line_number=caller_lineno, function_name=caller_function)
+        if self.enable_sls:
+            self.sls.put_logs(level="ERROR", msg=message, app=self.app_name, caller_filename=caller_filename, caller_lineno=caller_lineno, caller_function=caller_function, call_full_filename=call_full_filename)
         
         if self.feishu:
             existing_message = self.mongo.collection.find_one({"message": message}, sort=[("time", -1)])
@@ -137,12 +155,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):
         """记录严重错误级别日志"""
@@ -150,7 +169,8 @@ class AppLogger:
         logger.critical(f"[{caller_filename}:{caller_lineno}:{caller_function}] {message}")
         if self.enable_victorialog:
             self.victorialog.send_program_log(stream=self.stream, level="CRITICAL", message=message, app_name=self.app_name, file_name=call_full_filename, line_number=caller_lineno, function_name=caller_function)
-
+        if self.enable_sls:
+            self.sls.put_logs(level="CRITICAL", msg=message, app=self.app_name, caller_filename=caller_filename, caller_lineno=caller_lineno, caller_function=caller_function, call_full_filename=call_full_filename)
 
 # 使用示例
 if __name__ == "__main__":