加入斗鱼开播下播提醒

test/douyu.py

@@ -0,0 +1,63 @@
+import requests
+import time
+
+# 配置信息
+ROOM_ID = "52876"
+CHECK_INTERVAL = 300  # 5分钟 = 300秒
+HEADERS = {
+    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
+    "Referer": f"https://www.douyu.com/{ROOM_ID}"
+}
+
+
+class DouyuMonitor:
+    def __init__(self, room_id):
+        self.room_id = room_id
+        self.api_url = f"https://www.douyu.com/betard/{room_id}"
+        self.is_live = False  # 记录当前直播状态
+
+    def check_status(self):
+        try:
+            # 发送请求
+            response = requests.get(self.api_url, headers=HEADERS, timeout=10)
+            data = response.json()
+            print(data)
+            # 提取房间信息和状态
+            room_info = data.get("room", {})
+            show_status = room_info.get("show_status")  # 1 为开播
+            room_name = room_info.get("room_name", "无标题")
+            nickname = room_info.get("nickname", "未知主播")
+
+            # 逻辑判定
+            if show_status == 1:
+                if not self.is_live:
+                    print(f"【通知】主播 {nickname} 开播了！标题：{room_name}")
+                    self.send_desktop_notification(nickname, room_name)
+                    self.is_live = True
+                else:
+                    print(f"[{time.strftime('%H:%M:%S')}] 主播正在直播中...")
+            else:
+                if self.is_live:
+                    print(f"[{time.strftime('%H:%M:%S')}] 主播下播了。")
+                self.is_live = False
+                print(f"[{time.strftime('%H:%M:%S')}] 主播尚未开播。")
+
+        except Exception as e:
+            print(f"查询出错: {e}")
+
+    def send_desktop_notification(self, streamer, title):
+        """发送系统桌面通知"""
+        print(f"斗鱼直播提醒: {streamer}" + f"你关注的主播开播啦！\n标题：{title}"
+        )
+
+
+if __name__ == "__main__":
+    monitor = DouyuMonitor(ROOM_ID)
+    print(f"开始监控直播间: {ROOM_ID}，每 5 分钟检查一次...")
+
+    while True:
+        monitor.check_status()
+        time.sleep(CHECK_INTERVAL)
+
+
+