加入斗鱼开播下播提醒
This commit is contained in:
liuwei
4
plugins/douyu/__init__.py
Normal file
4
plugins/douyu/__init__.py
Normal file
@@ -0,0 +1,4 @@
|
||||
from .main import DouyuPlugin
|
||||
|
||||
def get_plugin():
|
||||
return DouyuPlugin()
|
||||
6
plugins/douyu/config.toml
Normal file
6
plugins/douyu/config.toml
Normal file
@@ -0,0 +1,6 @@
|
||||
[Douyu]
|
||||
enable = true
|
||||
command = ["订阅斗鱼", "取消订阅斗鱼", "斗鱼订阅列表"]
|
||||
check_interval_minutes = 5
|
||||
api_url_template = "https://www.douyu.com/betard/{room_id}"
|
||||
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"
|
||||
254
plugins/douyu/main.py
Normal file
254
plugins/douyu/main.py
Normal file
@@ -0,0 +1,254 @@
|
||||
import asyncio
|
||||
import json
|
||||
from typing import Dict, Any, List, Optional, Tuple, Set
|
||||
|
||||
import aiohttp
|
||||
from loguru import logger
|
||||
|
||||
from base.plugin_common.message_plugin_interface import MessagePluginInterface
|
||||
from base.plugin_common.plugin_interface import PluginStatus
|
||||
from db.connection import DBConnectionManager
|
||||
from utils.decorator.async_job import async_job
|
||||
from utils.decorator.plugin_decorators import plugin_stats_decorator
|
||||
from utils.decorator.points_decorator import plugin_points_cost
|
||||
from utils.robot_cmd.robot_command import Feature, PermissionStatus, GroupBotManager
|
||||
from wechat_ipad import WechatAPIClient
|
||||
|
||||
|
||||
class DouyuRedisManager:
|
||||
def __init__(self, db_manager: DBConnectionManager):
|
||||
self.redis = db_manager.get_redis_connection()
|
||||
self.prefix = "bot:douyu:"
|
||||
|
||||
def add_group_room(self, group_id: str, room_id: str) -> bool:
|
||||
key = f"{self.prefix}group:{group_id}:rooms"
|
||||
return self.redis.sadd(key, room_id) >= 0
|
||||
|
||||
def remove_group_room(self, group_id: str, room_id: str) -> bool:
|
||||
key = f"{self.prefix}group:{group_id}:rooms"
|
||||
return self.redis.srem(key, room_id) >= 0
|
||||
|
||||
def list_group_rooms(self, group_id: str) -> List[str]:
|
||||
key = f"{self.prefix}group:{group_id}:rooms"
|
||||
rooms = self.redis.smembers(key) or set()
|
||||
result = []
|
||||
for r in rooms:
|
||||
result.append(r.decode("utf-8") if isinstance(r, bytes) else r)
|
||||
return sorted(result)
|
||||
|
||||
def all_subscribed_rooms(self) -> Set[str]:
|
||||
groups = GroupBotManager.get_group_list()
|
||||
rooms: Set[str] = set()
|
||||
for gid in groups:
|
||||
for r in self.list_group_rooms(gid):
|
||||
rooms.add(r)
|
||||
return rooms
|
||||
|
||||
def groups_for_room(self, room_id: str) -> List[str]:
|
||||
groups = GroupBotManager.get_group_list()
|
||||
res = []
|
||||
for gid in groups:
|
||||
if room_id in set(self.list_group_rooms(gid)):
|
||||
res.append(gid)
|
||||
return res
|
||||
|
||||
def get_room_status(self, room_id: str) -> Optional[Dict[str, Any]]:
|
||||
key = f"{self.prefix}room_status:{room_id}"
|
||||
data = self.redis.get(key)
|
||||
if not data:
|
||||
return None
|
||||
if isinstance(data, bytes):
|
||||
data = data.decode("utf-8")
|
||||
try:
|
||||
return json.loads(data)
|
||||
except Exception:
|
||||
return None
|
||||
|
||||
def set_room_status(self, room_id: str, status: Dict[str, Any]) -> bool:
|
||||
key = f"{self.prefix}room_status:{room_id}"
|
||||
return self.redis.set(key, json.dumps(status, ensure_ascii=False), ex=86400)
|
||||
|
||||
|
||||
class DouyuPlugin(MessagePluginInterface):
|
||||
FEATURE_KEY = "DOUYU_MONITOR"
|
||||
FEATURE_DESCRIPTION = "🎮 斗鱼开播提醒 [订阅斗鱼 房间号, 取消订阅斗鱼 房间号]"
|
||||
|
||||
@property
|
||||
def name(self) -> str:
|
||||
return "斗鱼直播"
|
||||
|
||||
@property
|
||||
def version(self) -> str:
|
||||
return "1.0.0"
|
||||
|
||||
@property
|
||||
def description(self) -> str:
|
||||
return "斗鱼主播开播下播提醒与群订阅管理"
|
||||
|
||||
@property
|
||||
def author(self) -> str:
|
||||
return "ABOT Team"
|
||||
|
||||
@property
|
||||
def command_prefix(self) -> Optional[str]:
|
||||
return ""
|
||||
|
||||
@property
|
||||
def commands(self) -> List[str]:
|
||||
return self._commands
|
||||
|
||||
@property
|
||||
def feature_key(self) -> Optional[str]:
|
||||
return self.FEATURE_KEY
|
||||
|
||||
@property
|
||||
def feature_description(self) -> Optional[str]:
|
||||
return self.FEATURE_DESCRIPTION
|
||||
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.bot: WechatAPIClient = None
|
||||
self.feature = self.register_feature()
|
||||
self.redis_manager: Optional[DouyuRedisManager] = None
|
||||
self._commands = ["订阅斗鱼", "取消订阅斗鱼", "斗鱼订阅列表"]
|
||||
self._api_template = "https://www.douyu.com/betard/{room_id}"
|
||||
self._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"
|
||||
self._check_interval = 5
|
||||
|
||||
def initialize(self, context: Dict[str, Any]) -> bool:
|
||||
try:
|
||||
dbm = DBConnectionManager.get_instance()
|
||||
self.redis_manager = DouyuRedisManager(dbm)
|
||||
cfg = self._config.get("Douyu", {})
|
||||
self._commands = cfg.get("command", self._commands)
|
||||
self._api_template = cfg.get("api_url_template", self._api_template)
|
||||
self._user_agent = cfg.get("user_agent", self._user_agent)
|
||||
self._check_interval = int(cfg.get("check_interval_minutes", self._check_interval))
|
||||
async_job.every_minutes(self._check_interval)(self._scheduled_check_job)
|
||||
return True
|
||||
except Exception as e:
|
||||
logger.error(f"{self.name} 初始化失败: {e}")
|
||||
return False
|
||||
|
||||
def start(self) -> bool:
|
||||
self.status = PluginStatus.RUNNING
|
||||
return True
|
||||
|
||||
def stop(self) -> bool:
|
||||
self.status = PluginStatus.STOPPED
|
||||
return True
|
||||
|
||||
def can_process(self, message: Dict[str, Any]) -> bool:
|
||||
content = str(message.get("content", "")).strip()
|
||||
if not content:
|
||||
return False
|
||||
for cmd in self._commands:
|
||||
if content.startswith(cmd):
|
||||
return True
|
||||
return False
|
||||
|
||||
@plugin_stats_decorator(plugin_name="斗鱼直播")
|
||||
@plugin_points_cost(1, "斗鱼提醒消耗积分", FEATURE_KEY)
|
||||
async def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
|
||||
content = str(message.get("content", "")).strip()
|
||||
sender = message.get("sender")
|
||||
roomid = message.get("roomid", "")
|
||||
gbm: GroupBotManager = message.get("gbm")
|
||||
self.bot: WechatAPIClient = message.get("bot")
|
||||
if roomid and gbm.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
|
||||
return False, "没有权限"
|
||||
if content.startswith("订阅斗鱼"):
|
||||
parts = content.split()
|
||||
if len(parts) < 2:
|
||||
await self.bot.send_text_message(roomid or sender, "请提供房间号，例如：订阅斗鱼 7718843", sender)
|
||||
return True, "命令格式错误"
|
||||
room_id = parts[1].strip()
|
||||
ok = self.redis_manager.add_group_room(roomid or sender, room_id)
|
||||
await self.bot.send_text_message(roomid or sender, f"✅ 已订阅斗鱼房间 {room_id}", sender)
|
||||
return True, "订阅成功" if ok else "订阅失败"
|
||||
if content.startswith("取消订阅斗鱼"):
|
||||
parts = content.split()
|
||||
if len(parts) < 2:
|
||||
await self.bot.send_text_message(roomid or sender, "请提供房间号，例如：取消订阅斗鱼 7718843", sender)
|
||||
return True, "命令格式错误"
|
||||
room_id = parts[1].strip()
|
||||
ok = self.redis_manager.remove_group_room(roomid or sender, room_id)
|
||||
await self.bot.send_text_message(roomid or sender, f"✅ 已取消订阅斗鱼房间 {room_id}", sender)
|
||||
return True, "取消成功" if ok else "取消失败"
|
||||
if content.startswith("斗鱼订阅列表"):
|
||||
rooms = self.redis_manager.list_group_rooms(roomid or sender)
|
||||
if not rooms:
|
||||
await self.bot.send_text_message(roomid or sender, "暂无订阅", sender)
|
||||
return True, "暂无订阅"
|
||||
text = "当前订阅的斗鱼房间：\n" + "\n".join(rooms)
|
||||
await self.bot.send_text_message(roomid or sender, text, sender)
|
||||
return True, "列表已发送"
|
||||
return False, None
|
||||
|
||||
async def _scheduled_check_job(self):
|
||||
try:
|
||||
rooms = self.redis_manager.all_subscribed_rooms()
|
||||
if not rooms:
|
||||
return
|
||||
async with aiohttp.ClientSession() as session:
|
||||
for room_id in rooms:
|
||||
try:
|
||||
url = self._api_template.format(room_id=room_id)
|
||||
headers = {
|
||||
"User-Agent": self._user_agent,
|
||||
"Referer": f"https://www.douyu.com/{room_id}"
|
||||
}
|
||||
async with session.get(url, headers=headers, timeout=aiohttp.ClientTimeout(total=10)) as resp:
|
||||
data = await resp.json(content_type=None)
|
||||
room_info = data.get("room", {}) if isinstance(data, dict) else {}
|
||||
show_status = room_info.get("show_status")
|
||||
nickname = room_info.get("nickname", "")
|
||||
room_name = room_info.get("room_name", "")
|
||||
prev = self.redis_manager.get_room_status(room_id) or {}
|
||||
prev_live = prev.get("is_live")
|
||||
curr_live = True if show_status == 1 else False
|
||||
status_obj = {
|
||||
"is_live": curr_live,
|
||||
"nickname": nickname,
|
||||
"room_name": room_name
|
||||
}
|
||||
self.redis_manager.set_room_status(room_id, status_obj)
|
||||
if prev_live is None and curr_live is False:
|
||||
continue
|
||||
if prev_live is None and curr_live is True:
|
||||
await self._notify_groups_live(room_id, nickname, room_name)
|
||||
continue
|
||||
if prev_live is False and curr_live is True:
|
||||
await self._notify_groups_live(room_id, nickname, room_name)
|
||||
continue
|
||||
if prev_live is True and curr_live is False:
|
||||
await self._notify_groups_offline(room_id, nickname, room_name)
|
||||
continue
|
||||
await asyncio.sleep(0.1)
|
||||
except Exception as e:
|
||||
logger.error(f"斗鱼检查失败: {e}")
|
||||
continue
|
||||
except Exception as e:
|
||||
logger.error(f"斗鱼定时任务异常: {e}")
|
||||
|
||||
async def _notify_groups_live(self, room_id: str, nickname: str, room_name: str):
|
||||
groups = self.redis_manager.groups_for_room(room_id)
|
||||
text = f"斗鱼直播提醒：{nickname} 开播了\n标题：{room_name}\n地址：https://www.douyu.com/{room_id}"
|
||||
for gid in groups:
|
||||
if GroupBotManager.get_group_permission(gid, self.feature) == PermissionStatus.ENABLED:
|
||||
try:
|
||||
await self.bot.send_text_message(gid, text)
|
||||
except Exception as e:
|
||||
logger.error(f"发送斗鱼开播提醒失败: {e}")
|
||||
continue
|
||||
|
||||
async def _notify_groups_offline(self, room_id: str, nickname: str, room_name: str):
|
||||
groups = self.redis_manager.groups_for_room(room_id)
|
||||
text = f"斗鱼直播提醒：{nickname} 下播了\n标题：{room_name}"
|
||||
for gid in groups:
|
||||
if GroupBotManager.get_group_permission(gid, self.feature) == PermissionStatus.ENABLED:
|
||||
try:
|
||||
await self.bot.send_text_message(gid, text)
|
||||
except Exception as e:
|
||||
logger.error(f"发送斗鱼下播提醒失败: {e}")
|
||||
continue
|
||||
63
test/douyu.py
Normal file
63
test/douyu.py
Normal file
@@ -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)
|
||||
|
||||
|
||||
|
||||
File diff suppressed because one or more lines are too long
@@ -241,3 +241,168 @@ LINK_XML_NEWS = """
|
||||
</appinfo>
|
||||
<commenturl></commenturl>
|
||||
"""
|
||||
|
||||
DOUYU_MESSAGE_XML ="""
|
||||
<appmsg appid="wx6be84d532f192698" sdkver="0">
|
||||
<title>正在直播：{title}，请点击观看</title>
|
||||
<des>主播： {liver} </des>
|
||||
<username />
|
||||
<action>view</action>
|
||||
<type>5</type>
|
||||
<showtype>0</showtype>
|
||||
<content />
|
||||
<url>https://m.douyu.com/{roomid}?share_source=1&dyshci=3&app_type=and&fromuid=251679944&dyshid=f0054c8-618d0d36a18ce969bd03163400507111&medium=wx&dyshti=1769070814&type=lv</url>
|
||||
<lowurl />
|
||||
<forwardflag>0</forwardflag>
|
||||
<dataurl />
|
||||
<lowdataurl />
|
||||
<contentattr>0</contentattr>
|
||||
<streamvideo>
|
||||
<streamvideourl />
|
||||
<streamvideototaltime>0</streamvideototaltime>
|
||||
<streamvideotitle />
|
||||
<streamvideowording />
|
||||
<streamvideoweburl />
|
||||
<streamvideothumburl />
|
||||
<streamvideoaduxinfo />
|
||||
<streamvideopublishid />
|
||||
</streamvideo>
|
||||
<canvasPageItem>
|
||||
<canvasPageXml><![CDATA[]]></canvasPageXml>
|
||||
</canvasPageItem>
|
||||
<appattach></appattach>
|
||||
<extinfo />
|
||||
<androidsource>2</androidsource>
|
||||
<thumburl />
|
||||
<mediatagname />
|
||||
<messageaction><![CDATA[]]></messageaction>
|
||||
<messageext><![CDATA[]]></messageext>
|
||||
<emoticongift>
|
||||
<packageflag>0</packageflag>
|
||||
<packageid />
|
||||
</emoticongift>
|
||||
<emoticonshared>
|
||||
<packageflag>0</packageflag>
|
||||
<packageid />
|
||||
</emoticonshared>
|
||||
<designershared>
|
||||
<designeruin>0</designeruin>
|
||||
<designername>null</designername>
|
||||
<designerrediretcturl><![CDATA[null]]></designerrediretcturl>
|
||||
</designershared>
|
||||
<emotionpageshared>
|
||||
<tid>0</tid>
|
||||
<title>null</title>
|
||||
<desc>null</desc>
|
||||
<iconUrl><![CDATA[null]]></iconUrl>
|
||||
<secondUrl />
|
||||
<pageType>0</pageType>
|
||||
<setKey>null</setKey>
|
||||
</emotionpageshared>
|
||||
<webviewshared>
|
||||
<shareUrlOriginal />
|
||||
<shareUrlOpen />
|
||||
<jsAppId />
|
||||
<publisherId />
|
||||
<publisherReqId />
|
||||
</webviewshared>
|
||||
<template_id />
|
||||
<md5>5a2d851cd5d8b6f13e628583c1ce9a9c</md5>
|
||||
<websearch />
|
||||
<weappinfo>
|
||||
<username />
|
||||
<appid />
|
||||
<appservicetype>0</appservicetype>
|
||||
<secflagforsinglepagemode>0</secflagforsinglepagemode>
|
||||
<videopageinfo>
|
||||
<thumbwidth>136</thumbwidth>
|
||||
<thumbheight>136</thumbheight>
|
||||
<fromopensdk>0</fromopensdk>
|
||||
</videopageinfo>
|
||||
</weappinfo>
|
||||
<statextstr>GhQKEnd4NmJlODRkNTMyZjE5MjY5OA==</statextstr>
|
||||
<musicShareItem>
|
||||
<musicDuration>0</musicDuration>
|
||||
</musicShareItem>
|
||||
<finderLiveProductShare>
|
||||
</finderLiveProductShare>
|
||||
<finderOrder>
|
||||
|
||||
</finderOrder>
|
||||
<finderShopWindowShare>
|
||||
</finderShopWindowShare>
|
||||
<findernamecard>
|
||||
</findernamecard>
|
||||
<finderGuarantee>
|
||||
<scene><![CDATA[0]]></scene>
|
||||
</finderGuarantee>
|
||||
<directshare>0</directshare>
|
||||
<gamecenter>
|
||||
</gamecenter>
|
||||
<patMsg>
|
||||
<chatUser />
|
||||
<records>
|
||||
<recordNum>0</recordNum>
|
||||
</records>
|
||||
</patMsg>
|
||||
<secretmsg>
|
||||
<issecretmsg>0</issecretmsg>
|
||||
</secretmsg>
|
||||
<referfromscene>0</referfromscene>
|
||||
<gameshare>
|
||||
<liteappext>
|
||||
<liteappbizdata />
|
||||
<priority>0</priority>
|
||||
</liteappext>
|
||||
<appbrandext>
|
||||
<litegameinfo />
|
||||
<priority>-1</priority>
|
||||
</appbrandext>
|
||||
<gameshareid />
|
||||
<sharedata />
|
||||
<isvideo>0</isvideo>
|
||||
<duration>-1</duration>
|
||||
<isexposed>0</isexposed>
|
||||
<readtext />
|
||||
</gameshare>
|
||||
<tingChatRoomItem>
|
||||
<type>0</type>
|
||||
<categoryItem>null</categoryItem>
|
||||
<listenItem>null</listenItem>
|
||||
</tingChatRoomItem>
|
||||
<mpsharetrace>
|
||||
<hasfinderelement>0</hasfinderelement>
|
||||
</mpsharetrace>
|
||||
<wxgamecard>
|
||||
<framesetname />
|
||||
<mbcarddata />
|
||||
<minpkgversion />
|
||||
<clientextinfo />
|
||||
<mbcardheight>0</mbcardheight>
|
||||
<isoldversion>0</isoldversion>
|
||||
</wxgamecard>
|
||||
<ecskfcard>
|
||||
<framesetname />
|
||||
<mbcarddata />
|
||||
<minupdateunixtimestamp>0</minupdateunixtimestamp>
|
||||
<needheader>false</needheader>
|
||||
<summary />
|
||||
</ecskfcard>
|
||||
<liteapp>
|
||||
<id>null</id>
|
||||
<path />
|
||||
<query />
|
||||
<istransparent>0</istransparent>
|
||||
<hideicon>0</hideicon>
|
||||
<forbidforward>0</forbidforward>
|
||||
</liteapp>
|
||||
<opensdk_share_is_modified>0</opensdk_share_is_modified>
|
||||
</appmsg>
|
||||
<fromusername>Jyunere</fromusername>
|
||||
<scene>0</scene>
|
||||
<appinfo>
|
||||
<version>30</version>
|
||||
<appname>斗鱼</appname>
|
||||
</appinfo>
|
||||
<commenturl />
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user