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

fix 自动校正站点地址格式

Browse Source
This commit is contained in:
jxxghp 2024-03-17 12:21:32 +08:00
parent 97d0fc046a
commit a05724f664
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app/api/endpoints/site.py
View File

@ -50,6 +50,9 @@ def add_site(
return schemas.Response(success=False, message=f"{domain} 站点己存在") return schemas.Response(success=False, message=f"{domain} 站点己存在")
# 保存站点信息 # 保存站点信息
site_in.domain = domain site_in.domain = domain
# 校正地址格式
_scheme, _netloc = StringUtils.get_url_netloc(site_in.url)
site_in.url = f"{_scheme}://{_netloc}/"
site_in.name = site_info.get("name") site_in.name = site_info.get("name")
site_in.id = None site_in.id = None
site = Site(**site_in.dict()) site = Site(**site_in.dict())
@ -74,6 +77,9 @@ def update_site(
site = Site.get(db, site_in.id) site = Site.get(db, site_in.id)
if not site: if not site:
return schemas.Response(success=False, message="站点不存在") return schemas.Response(success=False, message="站点不存在")
# 校正地址格式
_scheme, _netloc = StringUtils.get_url_netloc(site_in.url)
site_in.url = f"{_scheme}://{_netloc}/"
site.update(db, site_in.dict()) site.update(db, site_in.dict())
# 通知缓存站点图标 # 通知缓存站点图标
EventManager().send_event(EventType.CacheSiteIcon, { EventManager().send_event(EventType.CacheSiteIcon, {