From e8b4b30aa466f9bacc52dfe392cb70f08dd09399 Mon Sep 17 00:00:00 2001 From: ngfchl Date: Tue, 13 Sep 2022 13:54:41 +0800 Subject: [PATCH] =?UTF-8?q?=E5=AF=BC=E5=85=A5=E6=95=B0=E6=8D=AE=E6=97=B6?= =?UTF-8?q?=E4=BF=9D=E5=AD=98PTPP=E5=A4=87=E4=BB=BD=E4=B8=AD=E7=9A=84?= =?UTF-8?q?=E9=83=A8=E5=88=86=E5=8F=AF=E7=94=A8=E6=95=B0=E6=8D=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pt_site/UtilityTool.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/pt_site/UtilityTool.py b/pt_site/UtilityTool.py index 4d3445a..397fe7a 100644 --- a/pt_site/UtilityTool.py +++ b/pt_site/UtilityTool.py @@ -243,7 +243,7 @@ class PtSpider: # print(domain, cookie_str) cookies.append({ 'url': data.get('url'), - 'uid': info.get('user').get('id'), + 'info': info.get('user'), 'passkey': info.get('passkey'), 'cookies': cookie_str.rstrip(';') }) @@ -262,14 +262,27 @@ class PtSpider: my_site = MySite.objects.filter(site=site).first() # print('查询我的站点：',my_site) # 如果有更新cookie，如果没有继续创建 + my_level = re.sub(u'([^a-zA-Z_ ])', "", cookie.get('info').get('levelName')) + if not my_site: my_site = MySite( site=site, cookie=cookie.get('cookies'), - user_id=cookie.get('uid'), passkey=cookie.get('passkey'), + user_id=cookie.get('info').get('id'), + my_level=my_level, + time_join=cookie.get('info').get('joinTime'), + seed=cookie.get('info').get('seeding'), + mail=cookie.get('info').get('messageCount') ) my_site.save() + # status = SiteStatus( + # my_site=my_site, + # uploaded=cookie.get('uploaded'), + # downloaded=cookie.get('downloaded'), + # my_sp=cookie.get('bonus'), + # seed_vol=cookie.get('seedingSize'), + # ) return CommonResponse.success(msg=site.name + ' 信息导入成功！') else: my_site.cookie = cookie.get('cookies').rstrip(';')