mirror of
https://github.com/ngfchl/ptools
synced 2023-07-10 13:41:22 +08:00
98 lines
2.9 KiB
Python
98 lines
2.9 KiB
Python
# Create your views here.
|
|
import json
|
|
import os
|
|
import subprocess
|
|
|
|
import git
|
|
from django.http import JsonResponse
|
|
from django.shortcuts import render
|
|
|
|
from pt_site.views import pt_spider, logger
|
|
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)
|
|
|
|
|
|
def test_subprocess(request):
|
|
"""
|
|
pip install -r requirements.txt
|
|
python manage.py makemigrations
|
|
python manage.py migrate
|
|
:param request:
|
|
:return:
|
|
"""
|
|
update_command = [
|
|
# 'cp db/db.sqlite3 db/db.sqlite3-$(date "+%Y%m%d%H%M%S")',
|
|
'git pull',
|
|
# 'pip install -r requirements.txt',
|
|
# 'python manage.py makemigrations',
|
|
'python manage.py migrate',
|
|
]
|
|
result = []
|
|
for command in update_command:
|
|
p = subprocess.run(command, shell=True)
|
|
print(p.stderr)
|
|
print(p.args)
|
|
print(p.returncode)
|
|
result.append({
|
|
'command': p.args,
|
|
'res': p.returncode
|
|
})
|
|
# print(result)
|
|
return JsonResponse(CommonResponse.success(data=result).to_dict(), safe=False)
|