From 80f253d8c6208ab3d2e8be8c63e0800d5e26c2ae Mon Sep 17 00:00:00 2001
From: ngfchl <ngfchl@126.com>
Date: Sun, 30 Apr 2023 23:21:23 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8D=87=E7=BA=A7=E5=B9=BF=E5=91=8A=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E6=8E=A5=E5=8F=A3=E4=B8=BA=E8=BF=9C=E7=A8=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pt_site/UtilityTool.py | 15 +++++----------
 1 file changed, 5 insertions(+), 10 deletions(-)

diff --git a/pt_site/UtilityTool.py b/pt_site/UtilityTool.py
index 5c4b395..6efefc7 100644
--- a/pt_site/UtilityTool.py
+++ b/pt_site/UtilityTool.py
@@ -126,16 +126,11 @@ class PtSpider:
         }, delay=delay)
 
     def select_ad(self):
-        # res = requests.get('http://api.ptools.fun/select_ad')
-        ads = [
-            'PTools目前开发遇到困难，开发电脑性能不足，时常崩溃，欢迎赞助！'
-            '<img src="http://img.ptools.fun/pay.png" width="100%" align="center">',
-            '为补贴开发，努力卖点农副产品，自家生纯净榨核桃油，如有需要欢迎联系我！'
-            '<img src="http://img.ptools.fun/wechat.png" width="100%" align="center">',
-            '广告位招租哟，欢迎投放广告！'
-            '<img src="http://img.ptools.fun/wechat.png" width="100%" align="center">',
-        ]
-        return random.choice(ads)
+        res = requests.get('http://api.ptools.fun/ad')
+        if res.status_code == 200 and res.json().get('code') == 0:
+            return res.json().get('data')
+        else:
+            return ''
 
     def send_text(self, message: str, title: str = '', url: str = None):
         """通知分流"""