From a05724f6645c6ce0db37c2038ef2859096acc1e7 Mon Sep 17 00:00:00 2001
From: jxxghp <jxxghp@gmail.com>
Date: Sun, 17 Mar 2024 12:21:32 +0800
Subject: [PATCH] =?UTF-8?q?fix=20=E8=87=AA=E5=8A=A8=E6=A0=A1=E6=AD=A3?=
 =?UTF-8?q?=E7=AB=99=E7=82=B9=E5=9C=B0=E5=9D=80=E6=A0=BC=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/api/endpoints/site.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/app/api/endpoints/site.py b/app/api/endpoints/site.py
index 02dba326..59360429 100644
--- a/app/api/endpoints/site.py
+++ b/app/api/endpoints/site.py
@@ -50,6 +50,9 @@ def add_site(
         return schemas.Response(success=False, message=f"{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.id = None
     site = Site(**site_in.dict())
@@ -74,6 +77,9 @@ def update_site(
     site = Site.get(db, site_in.id)
     if not site:
         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())
     # 通知缓存站点图标
     EventManager().send_event(EventType.CacheSiteIcon, {