nas/ptools
1
0
Fork 0
mirror of https://github.com/ngfchl/ptools synced 2023-07-10 13:41:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

上一版使用的魔改版延长有效期的备份文件，已修正

Browse Source
This commit is contained in:
ngfchl
2022-09-10 15:44:03 +08:00
parent 804963760c
commit 51518d1010
2 changed files with 73 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
auto_pt/views.py
View File

@@ -78,6 +78,22 @@ def test_notify(request):
def do_sql(request):
with open('./cookies.json', 'r') as f:
# print(f.readlines())
datas = json.load(f)
cookies = []
for data in datas:
domain = data.get('url')
cookie_list = data.get('cookies')
cookie_str = ''
for cookie in cookie_list:
cookie_str += cookie.get('name') + '=' + cookie.get('value') + ';'
print(domain, cookie_str)
cookies.append({
'domain': domain,
'cookies': cookie_str.rstrip(';')
})
print(len(cookies))
return JsonResponse('ok', safe=False)
@@ -85,44 +101,18 @@ def import_from_ptpp(request):
if request.method == 'GET':
return render(request, 'auto_pt/import_ptpp.html')
else:
# print(request.body)
data_list = json.loads(request.body).get('ptpp')
datas = json.loads(data_list)
print('content', len(datas))
cookies = pt_spider.parse_ptpp_cookies()
# success_messages = []
# error_messages = []
message_list = []
# print(datas)
cookies = []
cookie = {
'domain': '',
'cookies': ''
}
for index, data in enumerate(datas):
domain = data.get('domain').lstrip('.').lstrip('www.')
# print('domain', domain)
value = data.get('name') + '=' + data.get('value') + ';'
# print('value', value)
if not cookie.get('domain'):
cookie['domain'] = domain
cookie['cookies'] = value
elif domain in cookie.get('domain'):
# new_value = cookie['cookies'] + value
# cookie['cookies'] = new_value
cookie['cookies'] += value
# print('new_value', cookie['cookies'])
else:
cookie['cookies'] = cookie['cookies'].lstrip(';')
cookies.append(cookie)
# print(len(cookies))
# cookie = {}
# cookie['domain'] = domain
# cookie['cookies'] = value
cookie = {'domain': domain, 'cookies': value}
if index == len(datas) - 1:
cookies.append(cookie)
# print('cookie:', cookie)
print('cookies,', len(cookies))
for data in cookies:
try:
print(data)

53
pt_site/UtilityTool.py
View File

@@ -228,6 +228,59 @@ class PtSpider:
msg=StatusCodeEnum.OCR_ACCESS_ERR.errmsg + str(e)
)
def parse_ptpp_cookies(self, datas):
cookies = []
for data in datas:
domain = data.get('url')
cookie_list = data.get('cookies')
cookie_str = ''
for cookie in cookie_list:
cookie_str += cookie.get('name') + '=' + cookie.get('value') + ';'
print(domain, cookie_str)
cookies.append({
'domain': domain,
'cookies': cookie_str.rstrip(';')
})
return cookies
def parse_cookie_expire(self, datas):
"""
使用魔改版的一键延长COOKIE有效期备份文件
:param datas:
:return:
"""
cookies = []
cookie = {
'domain': '',
'cookies': ''
}
for index, data in enumerate(datas):
domain = data.get('domain').lstrip('.').lstrip('www.')
# print('domain', domain)
value = data.get('name') + '=' + data.get('value') + ';'
# print('value', value)
if not cookie.get('domain'):
cookie['domain'] = domain
cookie['cookies'] = value
elif domain in cookie.get('domain'):
# new_value = cookie['cookies'] + value
# cookie['cookies'] = new_value
cookie['cookies'] += value
# print('new_value', cookie['cookies'])
else:
cookie['cookies'] = cookie['cookies'].rstrip(';')
cookies.append(cookie)
# print(len(cookies))
# cookie = {}
# cookie['domain'] = domain
# cookie['cookies'] = value
cookie = {'domain': domain, 'cookies': value}
if index == len(datas) - 1:
cookies.append(cookie)
# print('cookie:', cookie)
print('cookies,', len(cookies))
return cookies
def get_uid_and_passkey(self, cookie: dict):
site = Site.objects.filter(url__contains=cookie.get('domain')).first()
# print('查询站点信息：',site)