diff --git a/package.v2.json b/package.v2.json index eb4a28b..9ab6200 100644 --- a/package.v2.json +++ b/package.v2.json @@ -149,5 +149,18 @@ "v1.4": "人物图片调整为优先从TMDB获取，避免douban图片CDN加载过慢的问题", "v1.3": "修复v1.8.5版本后刮削报错问题" } + }, + "SpeedLimiter": { + "name": "播放限速", + "description": "外网播放媒体库视频时，自动对下载器进行限速。", + "labels": "网络", + "version": "2.0", + "icon": "Librespeed_A.png", + "author": "Shurelol", + "level": 1, + "history": { + "v2.0": "兼容MoviePilot V2 版本", + "v1.2": "增加不限速路径配置，以应对网盘直链播放的情况" + } } } \ No newline at end of file diff --git a/plugins.v2/speedlimiter/__init__.py b/plugins.v2/speedlimiter/__init__.py index 517f683..5dcfc1b 100644 --- a/plugins.v2/speedlimiter/__init__.py +++ b/plugins.v2/speedlimiter/__init__.py @@ -1,16 +1,12 @@ import ipaddress -from typing import List, Tuple, Dict, Any +from typing import List, Tuple, Dict, Any, Optional -from app.core.config import settings from app.core.event import eventmanager, Event +from app.helper.downloader import DownloaderHelper +from app.helper.mediaserver import MediaServerHelper from app.log import logger -from app.modules.emby import Emby -from app.modules.jellyfin import Jellyfin -from app.modules.plex import Plex -from app.modules.qbittorrent import Qbittorrent -from app.modules.transmission import Transmission from app.plugins import _PluginBase -from app.schemas import NotificationType, WebhookEventInfo +from app.schemas import NotificationType, WebhookEventInfo, ServiceInfo from app.schemas.types import EventType from app.utils.ip import IpUtils @@ -23,7 +19,7 @@ class SpeedLimiter(_PluginBase): # 插件图标 plugin_icon = "Librespeed_A.png" # 插件版本 - plugin_version = "1.2" + plugin_version = "2.0" # 插件作者 plugin_author = "Shurelol" # 作者主页 @@ -36,9 +32,9 @@ class SpeedLimiter(_PluginBase): auth_level = 1 # 私有属性 + downloader_helper = None + mediaserver_helper = None _scheduler = None - _qb = None - _tr = None _enabled: bool = False _notify: bool = False _interval: int = 60 @@ -58,6 +54,8 @@ class SpeedLimiter(_PluginBase): _exclude_path = "" def init_plugin(self, config: dict = None): + self.downloader_helper = DownloaderHelper() + self.mediaserver_helper = MediaServerHelper() # 读取配置 if config: self._enabled = config.get("enabled") @@ -91,11 +89,6 @@ class SpeedLimiter(_PluginBase): self._unlimited_ips["ipv6"] = config.get("ipv6") or "" self._downloader = config.get("downloader") or [] - if self._downloader: - if 'qbittorrent' in self._downloader: - self._qb = Qbittorrent() - if 'transmission' in self._downloader: - self._tr = Transmission() def get_state(self) -> bool: return self._enabled @@ -177,18 +170,20 @@ class SpeedLimiter(_PluginBase): 'content': [ { 'component': 'VCol', + 'props': { + 'cols': 12 + }, 'content': [ { 'component': 'VSelect', 'props': { - 'chips': True, 'multiple': True, - 'model': 'downloader', + 'chips': True, + 'clearable': True, + 'model': 'downloaders', 'label': '下载器', - 'items': [ - {'title': 'Qbittorrent', 'value': 'qbittorrent'}, - {'title': 'Transmission', 'value': 'transmission'}, - ] + 'items': [{"title": config.name, "value": config.name} + for config in self.downloader_helper.get_configs().values()] } } ] @@ -398,12 +393,39 @@ class SpeedLimiter(_PluginBase): def get_page(self) -> List[dict]: pass + @property + def service_infos(self) -> Optional[Dict[str, ServiceInfo]]: + """ + 服务信息 + """ + if not self._downloader: + logger.warning("尚未配置下载器，请检查配置") + return None + + services = self.downloader_helper.get_services(name_filters=self._downloader) + if not services: + logger.warning("获取下载器实例失败，请检查配置") + return None + + active_services = {} + for service_name, service_info in services.items(): + if service_info.instance.is_inactive(): + logger.warning(f"下载器 {service_name} 未连接，请检查配置") + else: + active_services[service_name] = service_info + + if not active_services: + logger.warning("没有已连接的下载器，请检查配置") + return None + + return active_services + @eventmanager.register(EventType.WebhookMessage) def check_playing_sessions(self, event: Event = None): """ 检查播放会话 """ - if not self._qb and not self._tr: + if not self.service_infos: return if not self._enabled: return @@ -420,18 +442,17 @@ class SpeedLimiter(_PluginBase): return # 当前播放的总比特率 total_bit_rate = 0 - # 媒体服务器类型，多个以,分隔 - if not settings.MEDIASERVER: + media_servers = self.mediaserver_helper.get_services() + if not media_servers: return - media_servers = settings.MEDIASERVER.split(',') # 查询所有媒体服务器状态 - for media_server in media_servers: + for server, service in media_servers.items(): # 查询播放中会话 playing_sessions = [] - if media_server == "emby": + if service.type == "emby": req_url = "[HOST]emby/Sessions?api_key=[APIKEY]" try: - res = Emby().get_data(req_url) + res = service.instance.get_data(req_url) if res and res.status_code == 200: sessions = res.json() for session in sessions: @@ -453,10 +474,10 @@ class SpeedLimiter(_PluginBase): elif not IpUtils.is_private_ip(session.get("RemoteEndPoint")) \ and session.get("NowPlayingItem", {}).get("MediaType") == "Video": total_bit_rate += int(session.get("NowPlayingItem", {}).get("Bitrate") or 0) - elif media_server == "jellyfin": + elif service.type == "jellyfin": req_url = "[HOST]Sessions?api_key=[APIKEY]" try: - res = Jellyfin().get_data(req_url) + res = service.instance.get_data(req_url) if res and res.status_code == 200: sessions = res.json() for session in sessions: @@ -481,8 +502,8 @@ class SpeedLimiter(_PluginBase): media_streams = session.get("NowPlayingItem", {}).get("MediaStreams") or [] for media_stream in media_streams: total_bit_rate += int(media_stream.get("BitRate") or 0) - elif media_server == "plex": - _plex = Plex().get_plex() + elif service.type == "plex": + _plex = service.instance.get_plex() if _plex: sessions = _plex.sessions() for session in sessions: @@ -543,7 +564,7 @@ class SpeedLimiter(_PluginBase): """ 设置限速 """ - if not self._qb and not self._tr: + if not self.service_infos: return state = f"U:{upload_limit},D:{download_limit}" if self._current_state == state: @@ -555,6 +576,7 @@ class SpeedLimiter(_PluginBase): try: cnt = 0 for download in self._downloader: + service = self.service_infos.get(download) if self._auto_limit and limit_type == "播放": # 开启了播放智能限速 if len(self._downloader) == 1: @@ -578,44 +600,42 @@ class SpeedLimiter(_PluginBase): text = f"{text}\n下载：{download_limit} KB/s" else: text = f"{text}\n下载：未限速" - if str(download) == 'qbittorrent': - if self._qb: - self._qb.set_speed_limit(download_limit=download_limit, upload_limit=upload_limit) - # 发送通知 - if self._notify: - title = "【播放限速】" - if upload_limit or download_limit: - subtitle = f"Qbittorrent 开始{limit_type}限速" - self.post_message( - mtype=NotificationType.MediaServer, - title=title, - text=f"{subtitle}\n{text}" - ) - else: - self.post_message( - mtype=NotificationType.MediaServer, - title=title, - text=f"Qbittorrent 已取消限速" - ) + if service.type == 'qbittorrent': + service.instance.set_speed_limit(download_limit=download_limit, upload_limit=upload_limit) + # 发送通知 + if self._notify: + title = "【播放限速】" + if upload_limit or download_limit: + subtitle = f"Qbittorrent 开始{limit_type}限速" + self.post_message( + mtype=NotificationType.MediaServer, + title=title, + text=f"{subtitle}\n{text}" + ) + else: + self.post_message( + mtype=NotificationType.MediaServer, + title=title, + text=f"Qbittorrent 已取消限速" + ) else: - if self._tr: - self._tr.set_speed_limit(download_limit=download_limit, upload_limit=upload_limit) - # 发送通知 - if self._notify: - title = "【播放限速】" - if upload_limit or download_limit: - subtitle = f"Transmission 开始{limit_type}限速" - self.post_message( - mtype=NotificationType.MediaServer, - title=title, - text=f"{subtitle}\n{text}" - ) - else: - self.post_message( - mtype=NotificationType.MediaServer, - title=title, - text=f"Transmission 已取消限速" - ) + service.instance.set_speed_limit(download_limit=download_limit, upload_limit=upload_limit) + # 发送通知 + if self._notify: + title = "【播放限速】" + if upload_limit or download_limit: + subtitle = f"Transmission 开始{limit_type}限速" + self.post_message( + mtype=NotificationType.MediaServer, + title=title, + text=f"{subtitle}\n{text}" + ) + else: + self.post_message( + mtype=NotificationType.MediaServer, + title=title, + text=f"Transmission 已取消限速" + ) except Exception as e: logger.error(f"设置限速失败：{str(e)}")