From 594d916c67b2f8168a8817756bd7ac749ac5de3a Mon Sep 17 00:00:00 2001
From: wangzhibo <wangzhibo@shsening.com>
Date: 星期日, 27 七月 2025 22:27:57 +0800
Subject: [PATCH] change proxy check

---
 data/auto_media_publisher.sqlite3                 |    0 
 auto_media_publisher/utils/utils_network.py       |    6 +++++-
 auto_media_publisher/tasks/task_platform_login.py |    3 ++-
 3 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/auto_media_publisher/tasks/task_platform_login.py b/auto_media_publisher/tasks/task_platform_login.py
index 7436f58..65ba3db 100644
--- a/auto_media_publisher/tasks/task_platform_login.py
+++ b/auto_media_publisher/tasks/task_platform_login.py
@@ -56,7 +56,8 @@
                     max_session_hour = platform["max_session_hour"] 
 
                     if self._check_need_login(user,platform_name,max_session_hour):
-                        if check_proxy_valid(proxy):
+                        proxy_valid,ip,city = check_proxy_valid(proxy)
+                        if proxy_valid:
                             all_need_logins.append({
                                 "user":user,
                                 "proxy":proxy,
diff --git a/auto_media_publisher/utils/utils_network.py b/auto_media_publisher/utils/utils_network.py
index 84fb0e7..220956d 100644
--- a/auto_media_publisher/utils/utils_network.py
+++ b/auto_media_publisher/utils/utils_network.py
@@ -7,7 +7,10 @@
 
 
 def check_proxy_valid(proxy=None):
-    ip, city = get_public_ip_info(proxy)
+    if proxy:
+        ip, city = get_public_ip_info(proxy)
+    else:
+        ip, city = get_public_ip_info()
     return ip is not None,ip,city
 
 def get_public_ip_info(proxy=None):
@@ -15,6 +18,7 @@
         proxies = {"http": proxy, "https": proxy}
     else:
         proxies = None
+    print(f"checking proxy... {proxy}")
     try:
         url  = random.choice(PUBLIC_IP_CITY_URL)
         response = requests.get(url, proxies=proxies, timeout=5)
diff --git a/data/auto_media_publisher.sqlite3 b/data/auto_media_publisher.sqlite3
index 572bc8b..2a3b573 100644
--- a/data/auto_media_publisher.sqlite3
+++ b/data/auto_media_publisher.sqlite3
Binary files differ

--
Gitblit v1.9.1