codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 站点User-Agent没有设置的情况下以系统设置的UA进行访问

Browse Source
This commit is contained in:
景大侠 2024-04-05 11:37:26 +08:00
parent 7c93432505
commit a0c3b6b26b
No known key found for this signature in database
GPG Key ID: DC179BBB336AB282
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
app/chain/site.py
View File

@ -64,8 +64,9 @@ class SiteChain(ChainBase):
"""
# 获取token
token = None
user_agent = site.ua or settings.USER_AGENT
res = RequestUtils(
ua=site.ua,
ua=user_agent,
cookies=site.cookie,
proxies=settings.PROXY if site.proxy else None,
timeout=15
@ -81,7 +82,7 @@ class SiteChain(ChainBase):
headers={
'X-CSRF-TOKEN': token,
"Content-Type": "application/json; charset=utf-8",
"User-Agent": f"{site.ua}"
"User-Agent": f"{user_agent}"
},
cookies=site.cookie,
proxies=settings.PROXY if site.proxy else None,
@ -98,9 +99,10 @@ class SiteChain(ChainBase):
"""
判断站点是否已经登陆：m-team
"""
user_agent = site.ua or settings.USER_AGENT
url = f"{site.url}api/member/profile"
res = RequestUtils(
ua=site.ua,
ua=user_agent,
cookies=site.cookie,
proxies=settings.PROXY if site.proxy else None,
timeout=15
@ -110,7 +112,7 @@ class SiteChain(ChainBase):
if user_info and user_info.get("data"):
# 更新最后访问时间
res = RequestUtils(cookies=site.cookie,
ua=site.ua,
ua=user_agent,
timeout=60,
proxies=settings.PROXY if site.proxy else None,
referer=f"{site.url}index"
@ -194,7 +196,7 @@ class SiteChain(ChainBase):
rss_url, errmsg = self.rsshelper.get_rss_link(
url=site_info.url,
cookie=cookie,
ua=settings.USER_AGENT,
ua=site_info.ua or settings.USER_AGENT,
proxy=True if site_info.proxy else False
)
if rss_url:
@ -347,7 +349,7 @@ class SiteChain(ChainBase):
# 通用站点测试
site_url = site_info.url
site_cookie = site_info.cookie
ua = site_info.ua
ua = site_info.ua or settings.USER_AGENT
render = site_info.render
public = site_info.public
proxies = settings.PROXY if site_info.proxy else None