From ab2b33a9fd3a911a905e479a6a437b83de466f93 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=99=AF=E5=A4=A7=E4=BE=A0?= <dengjingren@foxmail.com>
Date: Fri, 31 Jan 2025 16:22:31 +0800
Subject: [PATCH] =?UTF-8?q?fix=20=E7=BD=91=E7=BB=9C=E5=BC=82=E5=B8=B8?=
 =?UTF-8?q?=E6=97=B6=EF=BC=8C=E7=AB=99=E7=82=B9=E6=B5=8B=E8=AF=95=E8=AF=AF?=
 =?UTF-8?q?=E6=8A=A5=E9=89=B4=E6=9D=83=E6=88=96Cookie=E8=BF=87=E6=9C=9F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/chain/site.py | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/app/chain/site.py b/app/chain/site.py
index cedaf657..6705a9b8 100644
--- a/app/chain/site.py
+++ b/app/chain/site.py
@@ -138,7 +138,9 @@ class SiteChain(ChainBase):
             proxies=settings.PROXY if site.proxy else None,
             timeout=site.timeout or 15
         ).get_res(url=site.url)
-        if res and res.status_code == 200:
+        if res is None:
+            return False, "无法打开网站！"
+        if res.status_code == 200:
             csrf_token = re.search(r'<meta name="x-csrf-token" content="(.+?)">', res.text)
             if csrf_token:
                 token = csrf_token.group(1)
@@ -155,7 +157,9 @@ class SiteChain(ChainBase):
             proxies=settings.PROXY if site.proxy else None,
             timeout=site.timeout or 15
         ).get_res(url=f"{site.url}api/user/getInfo")
-        if user_res and user_res.status_code == 200:
+        if user_res is None:
+            return False, "无法打开网站！"
+        if user_res.status_code == 200:
             user_info = user_res.json()
             if user_info and user_info.get("data"):
                 return True, "连接成功"
@@ -182,9 +186,11 @@ class SiteChain(ChainBase):
             proxies=settings.PROXY if site.proxy else None,
             timeout=site.timeout or 15
         ).post_res(url=url)
+        if res is None:
+            return False, "无法打开网站！"
         state = False
         message = "鉴权已过期或无效"
-        if res and res.status_code == 200:
+        if res.status_code == 200:
             user_info = res.json() or {}
             if user_info.get("data"):
                 # 更新最后访问时间
@@ -223,7 +229,9 @@ class SiteChain(ChainBase):
             proxies=settings.PROXY if site.proxy else None,
             timeout=site.timeout or 15
         ).get_res(url=url)
-        if res and res.status_code == 200:
+        if res is None:
+            return False, "无法打开网站！"
+        if res.status_code == 200:
             user_info = res.json()
             if user_info and user_info.get("success"):
                 return True, "连接成功"