🎨 添加向群推送bot所用pixiv账号关注画师的功能

Author: GaryDu0123

config.py

@@ -16,4 +16,9 @@
 
 CHAIN_REPLY = True  # 是否启用合并转发回复模式
 
-RANK_LIMIT = 5  # 每次推送排行榜时最多展示的作品数量
+RANK_LIMIT = 5  # 每次推送排行榜时最多展示的作品数量
+
+# 是否启用“推送机器人账号关注的画师”功能
+# 开启后，各群管理员才能通过指令选择是否接收推送
+# 出于隐私和性能考虑，默认关闭
+ENABLE_FOLLOWING_SUBSCRIPTION = False

pixiv.py

@@ -9,7 +9,7 @@
 from hoshino import Service, priv
 from hoshino.typing import CQEvent
 from pixivpy3 import AppPixivAPI
-from .config import PROXY_URL, MAX_DISPLAY_WORKS, IMAGE_QUALITY, CHECK_INTERVAL_HOURS
+from .config import PROXY_URL, MAX_DISPLAY_WORKS, IMAGE_QUALITY, CHECK_INTERVAL_HOURS, ENABLE_FOLLOWING_SUBSCRIPTION
 import aiohttp
 
 # 插件配置
@@ -29,6 +29,8 @@
 [pixiv关闭r18] 屏蔽R18内容
 [pixiv屏蔽tag tag名] 屏蔽包含指定tag的作品
 [pixiv取消屏蔽tag tag名] 取消屏蔽指定tag
+[pixiv开启关注推送] 订阅机器人账号关注的全部画师
+[pixiv关闭关注推送] 取消订阅机器人账号关注的画师
 [pixiv群设置] 查看当前群���设置
 """.strip()
 
@@ -113,8 +115,12 @@ def ensure_group_settings(self, group_id: str) -> None:
             self.subscriptions[group_id] = {
                 'artists': [],
                 'r18_enabled': False,
-                'blocked_tags': []
+                'blocked_tags': [],
+                'push_following_enabled': False
             }
+        # 兼容旧配置，如果旧配置没有这个键则添加默认值
+        elif 'push_following_enabled' not in self.subscriptions[group_id]:
+            self.subscriptions[group_id]['push_following_enabled'] = False
 
     def add_subscription(self, group_id: str, user_id: str) -> bool:
         """添加订阅"""
@@ -152,6 +158,18 @@ def is_r18_enabled(self, group_id: str) -> bool:
             return self.subscriptions[group_id].get('r18_enabled', False)
         return False
 
+    def set_push_following(self, group_id: str, enabled: bool) -> None:
+        """设置群的 关注画师推送 开关"""
+        self.ensure_group_settings(group_id)
+        self.subscriptions[group_id]['push_following_enabled'] = enabled
+        self.save_subscriptions()
+
+    def is_push_following_enabled(self, group_id: str) -> bool:
+        """检查群是否开启了 关注画师推送"""
+        if group_id in self.subscriptions:
+            return self.subscriptions[group_id].get('push_following_enabled', False)
+        return False
+
     def add_blocked_tag(self, group_id: str, tag: str) -> bool:
         """添加屏蔽tag"""
         self.ensure_group_settings(group_id)
@@ -179,13 +197,9 @@ def get_blocked_tags(self, group_id: str) -> List[str]:
 
     def get_group_settings(self, group_id: str) -> Dict:
         """获取群设置"""
-        if group_id in self.subscriptions:
-            return self.subscriptions[group_id]
-        return {
-            'artists': [],
-            'r18_enabled': False,
-            'blocked_tags': []
-        }
+        self.ensure_group_settings(group_id)
+        return self.subscriptions[group_id]
+
 
     def is_illust_allowed(self, illust: dict, group_id: Union[str, int]) -> bool:
         """检查作品是否允许在指定群推送"""
@@ -332,6 +346,48 @@ async def user_illusts(self, user_id: Union[str, int]):
             sv.logger.error(f"获取Pixiv用户作品列表时发生未知异常 '{user_id}': {e}")
             return {}, {}
 
+    async def get_illust_follow(self, start_time: datetime, interval_hours: float) -> List[Dict]:
+        """
+        获取当前bot关注画师在指定时间窗口内的新作品。
+        API本身返回最近作品，此函数在此基础上进行时间过滤。
+        """
+        try:
+            # 调用API获取原始的关注动态列表
+            result = await self.__exec_and_retry_with_login(
+                self.api.illust_follow
+            )
+
+            # 检查API返回是否有效
+            if not isinstance(result, dict) or 'illusts' not in result or not result.get('illusts'):
+                sv.logger.error(f"获取Pixiv关注作品列表失败或列表为空: {result}")
+                return []  # 失败或无内容时返回空列表
+
+            # 准备时间和用于存放结果的容器
+            check_start = start_time - timedelta(hours=interval_hours)
+            check_end = start_time
+            new_illusts_in_window = []
+
+            # 遍历API返回的所有作品，并根据时间窗口进行过滤
+            for illust in result['illusts']:
+                try:
+                    # 解析作品创建时间字符串
+                    create_date_utc = datetime.fromisoformat(illust['create_date']).astimezone(timezone.utc)
+
+                    # 判断作品是否在检查时间窗口内
+                    if check_start < create_date_utc <= check_end:
+                        new_illusts_in_window.append(illust)
+
+                except (ValueError, TypeError, KeyError) as e:
+                    sv.logger.warning(f"解析或过滤关注作品时跳过一个项目: {e}, 作品ID: {illust.get('id')}")
+                    continue
+            # 返回经过时间过滤后的新作品列表
+            return new_illusts_in_window
+
+        except Exception as e:
+            sv.logger.error(f"获取Pixiv关注作品时发生未知异常: {e}")
+            return []  # 确保任何未知异常都返回一个安全的空列表
+
+
     @staticmethod
     async def download_image_as_base64(url: str) -> str:
         """下载图片并转换为base64编码"""
@@ -524,6 +580,31 @@ async def set_pixiv_token(bot, ev: CQEvent):
     success, msg = manager.login(refresh_token)
     await bot.send(ev, msg)
 
+@sv.on_prefix('pixiv开启关注推送')
+async def enable_push_following(bot, ev: CQEvent):
+    """开启机器人账号关注画师的推送 (仅管理员)"""
+    if not priv.check_priv(ev, priv.ADMIN):
+        await bot.send(ev, "只有群主或管理员才能设置此项")
+        return
+
+    if not ENABLE_FOLLOWING_SUBSCRIPTION:
+        await bot.send(ev, "该功能已被维护组全局关闭")
+        return
+
+    group_id = str(ev.group_id)
+    manager.set_push_following(group_id, True)
+    await bot.send(ev, "本群将会收到账号关注画师的更新")
+
+@sv.on_prefix('pixiv关闭关注推送')
+async def disable_push_following(bot, ev: CQEvent):
+    """关闭机器人账号关注画师的推送 (仅管理员)"""
+    if not priv.check_priv(ev, priv.ADMIN):
+        await bot.send(ev, "只有群主或管理员才能设置此项")
+        return
+
+    group_id = str(ev.group_id)
+    manager.set_push_following(group_id, False)
+    await bot.send(ev, "已关闭关注推送")
 
 @sv.on_prefix('pixiv开启r18')
 async def enable_r18(bot, ev: CQEvent):
@@ -597,17 +678,18 @@ async def show_group_settings(bot, ev: CQEvent):
     msg += f"📋 订阅画师数量: {len(settings['artists'])}\n"
     msg += f"🔞 R18推送: {'开启' if settings['r18_enabled'] else '关闭'}\n"
 
+    if ENABLE_FOLLOWING_SUBSCRIPTION:
+        following_status = '开启' if settings.get('push_following_enabled', False) else '关闭'
+        msg += f"💖 关注画师推送: {following_status}\n"
+
     blocked_tags = settings['blocked_tags']
     if blocked_tags:
         msg += f"🚫 屏蔽tag: {', '.join(blocked_tags)}"
     else:
         msg += "🚫 屏蔽tag: 无"
-
     await bot.send(ev, msg)
 
 
-
-
 @sv.on_prefix('pixiv强制检查')
 async def force_check_updates(bot, ev: CQEvent):
     """强制执行一次更新检查 (仅用于测试)"""
@@ -660,16 +742,64 @@ async def construct_group_message(artist_name: str, filtered_illusts: List[Dict]
     return ''.join(msg_parts)
 
 
+async def process_and_send_updates(bot, user_id: str, artist_name: str, new_illusts: List[Dict], target_group_ids: set):
+    """
+    一个辅助函数, 负责处理单个画师的更新并发送给所有目标群组。
+    发送单个画师的新作, 为每个群组每个独立过滤作品、构造消息并发送。
+
+    :param bot: Bot实例
+    :param user_id: 画师ID
+    :param artist_name: 画师名字
+    :param new_illusts: 该画师的新作品列表
+    :param target_group_ids: 需要被通知的群组ID集合
+    """
+    if not new_illusts:
+        return  # 如果没有新作品，直接返回
+
+    # 向所有订阅了该画师的群组发送消息（根据群设置过滤内容）
+    for group_id in target_group_ids:
+        try:
+            # 根据群设置过滤作品
+            filtered_illusts = [
+                illust for illust in new_illusts if manager.is_illust_allowed(illust, group_id)
+            ]
+
+            # 如果过滤后没有符合条件的作品，则跳过这个群
+            if not filtered_illusts:
+                continue
+
+            await bot.send_group_msg(
+                group_id=int(group_id),
+                message=await construct_group_message(artist_name, filtered_illusts)
+            )
+            # 避免发送消息过快被限制
+            await asyncio.sleep(1)
+
+        except Exception as e:
+            sv.logger.error(f"向群 {group_id} 发送画师 {user_id} ({artist_name}) 更新消息时出错: {e}")
+            continue
+
 @sv.scheduled_job('interval', hours=CHECK_INTERVAL_HOURS)
 async def check_updates():
+    """
+    发送画师订阅的更新作品到对应群组的任务
+
+    实现思路:
+    1. user_follow的获取到的画师更新的作品实际上是和在当前时间窗口内用画师ID获取的作品列表是一样的, 所以需要去重
+    2. 根据避免频繁请求API的原则, 对每个画师只请求一次, 也就是说在user_follow推送之后就不需要用画师ID去请求一次了
+    3. 构建一个画师ID到订阅群列表的映射表
+    4. user_follow获取到时间窗口内的更新之后, 根据群设置过滤内容, 然后根据群是否���阅该画师和是否推送bot关注画师为条件来决定是否发送消息,
+        将发送过的画师ID从映射表中删除
+    5. 剩下的画师ID再用画师ID去请求一次, 这样就避免了重复请求和重复发送消息的问题
+    """
     start_time = datetime.now()
 
     bot = nonebot.get_bot()
 
     # 计算本次检查的时间窗口 - 以当前时间为结束点，向前检查CHECK_INTERVAL_HOURS的小时数
     check_time = datetime.now(timezone.utc)
 
-    # 收集��有需要检查的画师ID，并记录哪些群订阅了哪些画师
+    # 收集所有需要检查的画师ID，并记录画师被哪些群订阅
     artist_to_groups = {}  # {artist_id: [group_id1, group_id2, ...]}
 
     for group_id, group_data in manager.subscriptions.items():
@@ -679,54 +809,60 @@ async def check_updates():
                 artist_to_groups[user_id] = []
             artist_to_groups[user_id].append(group_id)
 
-    if not artist_to_groups:  # 没有订阅任何画师
-        return
+    # 处理关注推送 (如果开启)
+    if ENABLE_FOLLOWING_SUBSCRIPTION:
+        groups_enabling_following = {
+            group_id for group_id, setting in manager.subscriptions.items()
+            if setting.get('push_following_enabled', False)
+        }
+
+        # 获取关注画师在时间窗口内的新作品
+        followed_illusts = await manager.get_illust_follow(
+            start_time=check_time,
+            interval_hours=CHECK_INTERVAL_HOURS
+        )
 
-    # 对每个画师只请求一次
+        # 按画师ID分组作品
+        bot_followed_illusts = {}
+        for illust in followed_illusts:
+            user_id = str(illust['user']['id'])
+            if user_id not in bot_followed_illusts:
+                bot_followed_illusts[user_id] = {'user': illust['user'], 'illusts': []}
+            bot_followed_illusts[user_id]['illusts'].append(illust)
+
+        # 处理并发送关注画师的更新
+        for user_id, data in bot_followed_illusts.items():
+            artist_name = data['user']['name']
+            new_illusts = data['illusts']
+
+            # 计算需要通知的所有群组：订阅了该画师的 + 开启了全局关注推送的
+            target_group_ids = set(artist_to_groups.get(user_id, [])) | groups_enabling_following
+
+            await process_and_send_updates(bot, user_id, artist_name, new_illusts, target_group_ids)
+
+            # 从待检查列表中移除，避免重复请求
+            if user_id in artist_to_groups:
+                del artist_to_groups[user_id]
+
+    # 处理剩下的、未被关注推送覆盖的画师
     for user_id, group_ids in artist_to_groups.items():
         try:
-            # 使用精确的时间窗口获取新作品
             user_info, new_illusts = await manager.get_new_illusts_with_user_info(
                 user_id,
                 start_time=check_time,
                 interval_hours=CHECK_INTERVAL_HOURS
             )
 
-            artist_name = user_info['name'] if user_info else f"画师ID:{user_id}"
-
-            # 如果没有新作品，跳过
             if not new_illusts:
-                sv.logger.info(f"{artist_name} 没有新作品，跳过")
-                await asyncio.sleep(3) # 避免频繁请求API
+                sv.logger.info(f"画师 {user_id} 没有新作品，跳过")
+                await asyncio.sleep(3)
                 continue
 
-            # 向所有订阅了该画师的群组发送消息（根据群设置过滤内容）
-            for group_id in group_ids:
-                try:
-                    # 根据群设置过滤作品
-                    filtered_illusts = []
-                    for illust in new_illusts:
-                        is_allowed = manager.is_illust_allowed(illust, group_id)
-                        if is_allowed:
-                            filtered_illusts.append(illust)
-
-                    # 如果过滤后没有作品，跳过这个群
-                    if not filtered_illusts:
-                        continue
-
-                    await bot.send_group_msg(
-                        group_id=int(group_id),
-                        message=await construct_group_message(artist_name, filtered_illusts)
-                    )
-                    # 避免发送消息过快被限制
-                    await asyncio.sleep(1)
-
-                except Exception as e:
-                    sv.logger.error(f"向群 {group_id} 发送画师 {user_id} 更新消息时出错: {e}")
-                    continue
+            artist_name = user_info.get('name', f"画师ID:{user_id}")
+
+            await process_and_send_updates(bot, user_id, artist_name, new_illusts, set(group_ids))
 
-            # 避免频繁请求API
-            sv.logger.info(f"画师 {user_id} 处理完成，等待5秒...")
+            sv.logger.info(f"画师 {user_id} 处理完成，等待3秒...")
             await asyncio.sleep(3)
         except Exception as e:
             sv.logger.error(f"获取画师 {user_id} 更新时出错: {e}")

pixiv_tools.py

@@ -28,12 +28,12 @@
 HELP = '''
 [预览画师 画师ID/画师URL] 预览画师最新作品
 [获取插画|pget 作品ID/作品URL] 通过作品ID或URL获取指定作品
-[插画日榜] 获取Pixiv插画日榜
-[插画男性向排行] 获取Pixiv插画男性向排行榜
-[插画女性向排行] 获取Pixiv插画女性向排行榜
-[插画周榜] 获取Pixiv插画周榜
-[插画月榜] 获取Pixiv插画月榜
-[插画原画榜] 获取Pixiv插画原画榜
+[插画日榜] 获取插画日榜
+[插画男性向排行] 获取插画男性向排行榜
+[插画女性向排行] 获取插画女性向排行榜
+[插画周榜] 获取插画周榜
+[插画月榜] 获取插画月榜
+[插画原画榜] 获取插画原画榜
 '''.strip()
 
 sv = Service(