From 2e364852351527f55cfd59002017db706419354b Mon Sep 17 00:00:00 2001 From: ngfchl Date: Mon, 15 May 2023 09:20:12 +0800 Subject: [PATCH] =?UTF-8?q?=E5=8D=87=E7=BA=A7=E5=B9=BF=E5=91=8A=E6=8E=A5?= =?UTF-8?q?=E5=8F=A3=EF=BC=8C=E5=A2=9E=E5=8A=A0=E5=8E=BB=E5=B9=BF=E5=91=8A?= =?UTF-8?q?=E5=8A=9F=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pt_site/UtilityTool.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/pt_site/UtilityTool.py b/pt_site/UtilityTool.py index d172647..57df0a7 100644 --- a/pt_site/UtilityTool.py +++ b/pt_site/UtilityTool.py @@ -8,7 +8,6 @@ import ssl import threading import time import traceback -import urllib.parse from datetime import datetime from urllib.request import urlopen @@ -26,12 +25,12 @@ from lxml import etree from pypushdeer import PushDeer from requests import Response, ReadTimeout from urllib3.exceptions import NewConnectionError, ConnectTimeoutError, MaxRetryError -from util.wxpusher import WxPusher from auto_pt.models import Notify, OCR from pt_site.models import MySite, SignIn, TorrentInfo, SiteStatus, Site from ptools.base import TorrentBaseInfo, PushConfig, CommonResponse, StatusCodeEnum, DownloaderCategory from ptools.settings import BASE_DIR +from util.wxpusher import WxPusher from .wechat_push import WechatPush urllib3.util.ssl_.DEFAULT_CIPHERS = 'ALL' @@ -126,7 +125,12 @@ class PtSpider: }, delay=delay) def select_ad(self): - res = requests.get('http://api.ptools.fun/ad') + ad_token = self.parse_token('ad') + if ad_token.code == 0: + token = ad_token.data + else: + token = None + res = requests.get('http://api.ptools.fun/ad', params=token) if res.status_code == 200 and res.json().get('code') == 0: return res.json().get('data') else: