From 7cba054e7f86a473a08a045336c6e41b41d086f8 Mon Sep 17 00:00:00 2001 From: InfinityPacer <160988576+InfinityPacer@users.noreply.github.com> Date: Fri, 22 Mar 2024 20:45:52 +0800 Subject: [PATCH] =?UTF-8?q?=E9=80=9A=E8=BF=87tracker=E8=A1=A5=E5=85=A8?= =?UTF-8?q?=E7=AB=99=E7=82=B9=E6=97=B6=EF=BC=8C=E9=83=A8=E5=88=86=E7=AB=99?= =?UTF-8?q?=E7=82=B9tracker=E9=80=9A=E8=BF=87=E5=9F=9F=E5=90=8D=E6=98=A0?= =?UTF-8?q?=E5=B0=84=E8=BF=9B=E8=A1=8C=E5=89=8D=E7=BD=AE=E5=A4=84=E7=90=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- package.json | 2 +- plugins/downloadsitetag/__init__.py | 13 ++++++++++++- 2 files changed, 13 insertions(+), 2 deletions(-) diff --git a/package.json b/package.json index bb99105..ea0bd89 100644 --- a/package.json +++ b/package.json @@ -338,7 +338,7 @@ "DownloadSiteTag": { "name": "下载任务分类与标签", "description": "自动给下载任务分类与打站点标签、剧集名称标签", - "version": "1.9", + "version": "2.0", "icon": "Youtube-dl_B.png", "author": "叮叮当", "level": 1 diff --git a/plugins/downloadsitetag/__init__.py b/plugins/downloadsitetag/__init__.py index f366e8d..3326d98 100644 --- a/plugins/downloadsitetag/__init__.py +++ b/plugins/downloadsitetag/__init__.py @@ -190,6 +190,11 @@ class DownloadSiteTag(_PluginBase): # JackettIndexers索引器支持多个站点, 如果不存在历史记录, 则通过tracker会再次附加其他站点名称 indexers.append("JackettIndexers") indexers = set(indexers) + tracker_mappings = { + "chdbits.xyz": "ptchdbits.co", + "agsvpt.trackers.work": "agsvpt.com", + "tracker.cinefiles.info": "audiences.me", + } for DOWNLOADER in ["qbittorrent", "transmission"]: logger.info(f"{self.LOG_TAG}开始扫描下载器 {DOWNLOADER} ...") # 获取下载器中的种子 @@ -241,7 +246,13 @@ class DownloadSiteTag(_PluginBase): elif not history.torrent_site: trackers = self._get_trackers(torrent=torrent, dl_type=DOWNLOADER) for tracker in trackers: - domain = StringUtils.get_url_domain(tracker) + # 检查tracker是否包含特定的关键字，并进行相应的映射 + for key, mapped_domain in tracker_mappings.items(): + if key in tracker: + domain = mapped_domain + break + else: + domain = StringUtils.get_url_domain(tracker) site_info = self.sites_helper.get_indexer(domain) if site_info: history.torrent_site = site_info.get("name")