diff --git a/pt_site/UtilityTool.py b/pt_site/UtilityTool.py
index 51e3e9f..97c6720 100644
--- a/pt_site/UtilityTool.py
+++ b/pt_site/UtilityTool.py
@@ -454,6 +454,31 @@ class PtSpider:
return False
return img_data
+ def sign_in_hdupt(self, my_site: MySite):
+ site = my_site.site
+ url = site.url + site.page_control_panel.lstrip('/')
+ result = self.send_request(
+ my_site=my_site,
+ url=url,
+ )
+ sign_str = self.parse(result, '//span[@id="yiqiandao"]')
+ logger.info(sign_str)
+ if len(sign_str) == 1:
+ return CommonResponse.success(msg=site.name + '已签到,请勿重复操作!!')
+ sign_res = self.send_request(
+ my_site=my_site,
+ url=site.url + site.page_sign_in.lstrip('/'),
+ method=site.sign_in_method
+ ).content.decode('utf8')
+ if isinstance(sign_res, int):
+ msg = '你还需要继续努力哦!此次签到,你获得了魔力奖励:{}'.format(sign_res)
+ else:
+ msg = sign_res
+ logger.info(msg)
+ return CommonResponse.success(
+ msg=msg
+ )
+
def sign_in_hd4fans(self, my_site: MySite):
site = my_site.site
url = site.url + site.page_control_panel.lstrip('/')
@@ -687,7 +712,7 @@ class PtSpider:
return '0' if len(res_list) == 0 else res_list[0]
def do_sign_in(self, pool, queryset: QuerySet[MySite]):
- message_list = '# 自动签到通知 \n\n ### 未显示的站点已经签到过了哟! \n\n'
+ message_list = '# 自动签到通知 \n\n### 未显示的站点已经签到过了哟! \n\n'
if datetime.now().hour < 9:
# U2每天九点前不签到
queryset = [my_site for my_site in queryset if 'u2.dmhy.org' not in my_site.site.url and
@@ -744,6 +769,13 @@ class PtSpider:
signin_today.sign_in_info = result.msg
signin_today.save()
return result
+ if 'hdupt.com' in site.url:
+ result = self.sign_in_hdupt(my_site)
+ if result.code == StatusCodeEnum.OK.code:
+ signin_today.sign_in_today = True
+ signin_today.sign_in_info = result.msg
+ signin_today.save()
+ return result
if 'hdchina' in site.url:
result = self.sign_in_hdc(my_site)
if result.code == StatusCodeEnum.OK.code:
@@ -834,9 +866,8 @@ class PtSpider:
)
res = self.send_request(my_site=my_site, method=site.sign_in_method, url=url,
data=eval(site.sign_in_params))
-
+ logger.info(res.text)
if 'hares.top' in site.url:
- logger.info(res.text)
code = res.json().get('code')
# logger.info('白兔返回码:'+ type(code))
if int(code) == 0:
@@ -901,7 +932,6 @@ class PtSpider:
return CommonResponse.success(msg='签到成功!')
else:
return CommonResponse.error(msg='签到失败!')
- logger.info(res.text)
if res.status_code == 200:
status = converter.convert(res.content.decode('utf8'))
# status = ''.join(self.parse(res, '//a[contains(@href,{})]/text()'.format(site.page_sign_in)))
diff --git a/pt_site/views.py b/pt_site/views.py
index e2e20a1..06a5ea2 100644
--- a/pt_site/views.py
+++ b/pt_site/views.py
@@ -55,7 +55,7 @@ def auto_get_status():
更新个人数据
"""
start = time.time()
- message_list = '# 更新个人数据 \n\n '
+ message_list = '# 更新个人数据 \n\n'
queryset = MySite.objects.all()
site_list = [my_site for my_site in queryset if my_site.site.get_userinfo_support]
results = pool.map(pt_spider.send_status_request, site_list)
@@ -109,7 +109,7 @@ def auto_update_torrents():
拉取最新种子
"""
start = time.time()
- message_list = '# 拉取免费种子 \n\n '
+ message_list = '# 拉取免费种子 \n\n'
queryset = MySite.objects.all()
site_list = [my_site for my_site in queryset if my_site.site.get_torrent_support]
results = pool.map(pt_spider.send_torrent_info_request, site_list)