接入企业微信报警

Boris-code · Apr 16, 2021 · ee28d8a · ee28d8a
1 parent 69e5f0b
commit ee28d8a
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 8 deletions.
diff --git a/feapder/core/scheduler.py b/feapder/core/scheduler.py
@@ -446,6 +446,10 @@ def send_msg(self, msg, level="debug", message_prefix=""):
                 msg, message_prefix=message_prefix, title=self._spider_name
             )
 
+        if setting.WECHAT_WARNING_URL:
+            keyword = "feapder报警系统\n"
+            tools.wechat_warning(keyword + msg, message_prefix=message_prefix)
+
     def spider_begin(self):
         """
         @summary: start_monitor_task 方式启动，此函数与spider_end不在同一进程内，变量不可共享

diff --git a/feapder/setting.py b/feapder/setting.py
@@ -99,7 +99,7 @@
 ITEM_FILTER_ENABLE = False  # item 去重
 REQUEST_FILTER_ENABLE = False  # request 去重
 
-# 报警 支持钉钉及邮件，二选一即可
+# 报警 支持钉钉、企业微信、邮件
 # 钉钉报警
 DINGDING_WARNING_URL = ""  # 钉钉机器人api
 DINGDING_WARNING_PHONE = ""  # 报警人 支持列表，可指定多个

diff --git a/feapder/templates/project_template/setting.py b/feapder/templates/project_template/setting.py
@@ -75,7 +75,7 @@
 # ITEM_FILTER_ENABLE = False  # item 去重
 # REQUEST_FILTER_ENABLE = False  # request 去重
 #
-# # 报警 支持钉钉及邮件，二选一即可
+# 报警 支持钉钉、企业微信、邮件
 # # 钉钉报警
 # DINGDING_WARNING_URL = ""  # 钉钉机器人api
 # DINGDING_WARNING_PHONE = ""  # 报警人 支持列表，可指定多个

diff --git a/feapder/utils/tools.py b/feapder/utils/tools.py
@@ -2304,7 +2304,7 @@ def wechat_warning(
     """企业微信报警"""
     if isinstance(user_phone, str):
         user_phone = [user_phone] if user_phone else []
-        
+
     if all_users is True or not user_phone:
         user_phone = ["@all"]
 
@@ -2314,13 +2314,10 @@ def wechat_warning(
     if is_in_rate_limit(rate_limit, url, user_phone, message_prefix or message):
         log.info("报警时间间隔过短，此次报警忽略。 内容 {}".format(message))
         return
-    
+
     data = {
         "msgtype": "text",
-        "text": {
-            "content": message,
-            "mentioned_mobile_list": user_phone
-        }
+        "text": {"content": message, "mentioned_mobile_list": user_phone},
     }
 
     headers = {"Content-Type": "application/json"}