mirror of
https://github.com/ngfchl/ptools
synced 2023-07-10 13:41:22 +08:00
66 lines
2.0 KiB
Python
66 lines
2.0 KiB
Python
# Create your views here.
|
|
import json
|
|
|
|
from django.http import JsonResponse
|
|
from django.shortcuts import render
|
|
|
|
from pt_site.views import pt_spider
|
|
from ptools.base import CommonResponse, StatusCodeEnum
|
|
|
|
|
|
def test_import(request):
|
|
if request.method == 'GET':
|
|
return render(request, 'pt_test/test_import.html')
|
|
else:
|
|
data_list = json.loads(request.body).get('user')
|
|
res = pt_spider.parse_ptpp_cookies(data_list)
|
|
if res.code == StatusCodeEnum.OK.code:
|
|
cookies = res.data
|
|
print(cookies)
|
|
else:
|
|
return JsonResponse(res.to_dict(), safe=False)
|
|
message_list = []
|
|
for data in cookies:
|
|
try:
|
|
print(data)
|
|
res = pt_spider.get_uid_and_passkey(data)
|
|
msg = res.msg
|
|
print(msg)
|
|
if res.code == StatusCodeEnum.OK.code:
|
|
message_list.append({
|
|
'msg': msg,
|
|
'tag': 'success'
|
|
})
|
|
else:
|
|
# error_messages.append(msg)
|
|
message_list.append({
|
|
'msg': msg,
|
|
'tag': 'error'
|
|
})
|
|
except Exception as e:
|
|
message = '{} 站点导入失败！{} \n'.format(data.get('domain'), str(e))
|
|
message_list.append({
|
|
'msg': message,
|
|
'tag': 'warning'
|
|
})
|
|
return JsonResponse(CommonResponse.success(data={
|
|
'messages': message_list
|
|
}).to_dict(), safe=False)
|
|
|
|
|
|
def handle_json(request):
|
|
if request.method == 'GET':
|
|
return render(request, 'pt_test/test_json.html')
|
|
|
|
|
|
def test_post(request):
|
|
if request.method == 'GET':
|
|
return render(request, 'pt_test/test_post.html')
|
|
else:
|
|
r = request.POST
|
|
for i, j in r.items():
|
|
print(i, type(i))
|
|
print(j, type(j))
|
|
print(r)
|
|
return JsonResponse(CommonResponse.success(data=r).to_dict(), safe=False)
|