优化

Core.py

@@ -3,7 +3,8 @@ from datetime import datetime
 import sys
 import time
 import yaml
-# from SendBot import SendToFeishu
+import requests
+from SendBot import SendToFeishu
 from media.common import run
 from media.freebuf import freebuf_main
 from media.xianzhi import xianzhi_main
@@ -15,7 +16,7 @@ from GotoSend_doonsec import Src_doonsec
 def get_params():
     with open('./config.yaml', 'r', encoding="utf-8") as file:
         config = yaml.safe_load(file)
-    sleep_time = f"{config['sleep_time']}"
+    sleep_time = int(f"{config['sleep_time']}")
     s_hour = int(f"{config['s_hour']}")
     e_hour = int(f"{config['e_hour']}")
     c_hour = int(f"{config['c_hour']}")
@@ -57,12 +58,90 @@ def main_loop():
                 pass
                 # print("正在等待执行...")
                 # print("等待间隔：", int(sleep_time))
-            time.sleep(int(sleep_time))  # 每隔35秒执行一次
+            time.sleep(sleep_time)  # 每隔35秒执行一次
         except Exception as e:
             print(f"发生错误: {e} ,程序已暂停")
             # SendToFeishu(f"发生错误: {e} ,程序已退出", "报错信息")
             exit()
 
+# 探测rss源状态
+def check_rss_status(url):
+    try:
+        response = requests.get(url, timeout=10)
+        if response.status_code == 200 and len(response.content) > 0:
+            return True
+        else:
+            return f"状态码: {response.status_code}, 内容长度: {len(response.content)}"
+    except requests.RequestException as e:
+        return f"请求异常: {str(e)}"
+
+def test_rss_source():
+    rss_info = ""
+    url_1 = check_rss_status("https://forum.butian.net/Rss")
+    if url_1 == True:
+        rss_info += "奇安信 源正常\n"
+    else:
+        rss_info += f"奇安信 源异常: {url_1}\n"
+
+    url_2 = check_rss_status("https://wechat.doonsec.com/bayes_rss.xml")
+    if url_2 == True:
+        rss_info += "洞见 源正常\n"
+    else:
+        rss_info += f"洞见 源异常: {url_2}\n"
+
+    url_3 = check_rss_status("https://www.huawei.com/cn/rss-feeds/psirt/rss")
+    if url_3 == True:
+        rss_info += "华为 源正常\n"
+    else:
+        rss_info += f"华为 源异常: {url_3}\n"
+
+    url_4 = check_rss_status("https://www.sec_wiki.com/news/rss")
+    if url_4 == True:
+        rss_info += "安全维基 源正常\n"
+    else:
+        rss_info += f"安全维基 源异常: {url_4}\n"
+
+    url_5 = check_rss_status("https://api.anquanke.com/data/v1/rss")
+    if url_5 == True:
+        rss_info += "安全客 源正常\n"
+    else:
+        rss_info += f"安全客 源异常: {url_5}\n"
+
+    url_6 = check_rss_status("https://www.4hou.com/feed")
+    if url_6 == True:
+        rss_info += "嘶吼 源正常\n"
+    else:
+        rss_info += f"嘶吼 源异常: {url_6}\n"
+
+    url_7 = check_rss_status("https://paper.seebug.org/rss/")
+    if url_7 == True:
+        rss_info += "Seebug社区 源正常\n"
+    else:
+        rss_info += f"Seebug社区 源异常: {url_7}\n"
+
+    url_8 = check_rss_status("https://www.freebuf.com/feed")
+    if url_8 == True:
+        rss_info += "FreeBuf社区 源正常\n"
+    else:
+        rss_info += f"FreeBuf社区 源异常: {url_8}\n"
+
+    url_9 = check_rss_status("https://xz.aliyun.com/feed")
+    if url_9 == True:
+        rss_info += "先知社区 源正常\n"
+    else:
+        rss_info += f"先知社区 源异常: {url_9}\n"
+
+    return rss_info
+
 if __name__ == "__main__":
     print("程序正在运行当中。")
+    rss_info = test_rss_source()
+    start_info = ""
+    start_info += "程序已启动，当前时间为：" + datetime.now().strftime("%Y-%m-%d %H:%M:%S") + "\n"
+    start_info += "程序作者：MasonLiu \t 开源地址：[GM-gitea](https://git.masonliu.com/MasonLiu/PyBot)" + "\n"
+    start_info += "时间配置：前一天 " + str(s_hour) + " 点到当天 " + str(e_hour) + " 点，每" + str(sleep_time) + "秒执行一次" + ";每日11点开始推送。" + "\n"
+    SendToFeishu(start_info, "程序信息")
+    # print(start_info)
+    SendToFeishu(rss_info, "RSS源状态")
+    # print(rss_info)
     main_loop()