From d4c0d65a54b1cb3f0e9088aea9854baa91d7e5b6 Mon Sep 17 00:00:00 2001 From: jxxghp Date: Mon, 14 Aug 2023 11:10:03 +0800 Subject: [PATCH] fix IYUUAutoSeed bug --- app/plugins/iyuuautoseed/__init__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/app/plugins/iyuuautoseed/__init__.py b/app/plugins/iyuuautoseed/__init__.py index d2683b863..b5a4483d4 100644 --- a/app/plugins/iyuuautoseed/__init__.py +++ b/app/plugins/iyuuautoseed/__init__.py @@ -7,6 +7,7 @@ from apscheduler.schedulers.background import BackgroundScheduler from apscheduler.triggers.cron import CronTrigger from lxml import etree +from ruamel.yaml import CommentedMap from app.core.config import settings from app.helper.sites import SitesHelper @@ -809,7 +810,7 @@ def __get_save_path(torrent: Any, dl_type: str): print(str(e)) return "" - def __get_download_url(self, seed: dict, site: dict, base_url: str): + def __get_download_url(self, seed: dict, site: CommentedMap, base_url: str): """ 拼装种子下载链接 """ @@ -874,7 +875,7 @@ def __get_torrent_url_from_page(self, seed: dict, site: dict): logger.info(f"正在获取种子下载链接：{page_url} ...") res = RequestUtils( cookies=site.get("cookie"), - headers=site.get("ua"), + ua=site.get("ua"), proxies=settings.PROXY if site.get("proxy") else None ).get_res(url=page_url) if res is not None and res.status_code in (200, 500):