mirror of
https://github.com/ngfchl/ptools
synced 2023-07-10 13:41:22 +08:00
137 lines
4.3 KiB
Python
137 lines
4.3 KiB
Python
import json
|
||
import os
|
||
import subprocess
|
||
from datetime import datetime
|
||
|
||
import markdown
|
||
from django.http import JsonResponse
|
||
from django.shortcuts import render
|
||
|
||
from pt_site import views as tasks
|
||
from pt_site.models import SiteStatus, MySite
|
||
from pt_site.views import scheduler, pt_spider
|
||
from ptools.base import CommonResponse
|
||
|
||
|
||
def add_task(request):
|
||
if request.method == 'POST':
|
||
content = json.loads(request.body.decode()) # 接收参数
|
||
try:
|
||
start_time = content['start_time'] # 用户输入的任务开始时间, '10:00:00'
|
||
start_time = start_time.split(':')
|
||
hour = int(start_time[0])
|
||
minute = int(start_time[1])
|
||
second = int(start_time[2])
|
||
s = content['s'] # 接收执行任务的各种参数
|
||
# 创建任务
|
||
scheduler.add_job(tasks.scheduler, 'cron', hour=hour, minute=minute, second=second, args=[s])
|
||
code = '200'
|
||
message = 'success'
|
||
except Exception as e:
|
||
code = '400'
|
||
message = e
|
||
|
||
data = {
|
||
'code': code,
|
||
'message': message
|
||
}
|
||
return JsonResponse(json.dumps(data, ensure_ascii=False), safe=False)
|
||
|
||
|
||
def get_tasks(request):
|
||
# print(dir(tasks))
|
||
data = [key for key in dir(tasks) if key.startswith('auto')]
|
||
print(data)
|
||
# print(tasks.__getattr__)
|
||
# print(tasks.auto_get_status.__doc__)
|
||
# inspect.getmembers(tasks, inspect.isfunction)
|
||
# inspect.getmodule(tasks)
|
||
# print(sys.modules[__name__])
|
||
# print(sys.modules.values())
|
||
# print(sys.modules.keys())
|
||
# print(sys.modules.items())
|
||
return JsonResponse('ok', safe=False)
|
||
|
||
|
||
def exec_task(request):
|
||
# res = AutoPt.auto_sign_in()
|
||
# print(res)
|
||
tasks.auto_sign_in.delay()
|
||
return JsonResponse('ok!', safe=False)
|
||
|
||
|
||
def test_field(request):
|
||
my_site = MySite.objects.get(pk=1)
|
||
list1 = SiteStatus.objects.filter(site=my_site, created_at__date__gte=datetime.today())
|
||
print(list1)
|
||
return JsonResponse('ok!', safe=False)
|
||
|
||
|
||
def test_notify(request):
|
||
"""
|
||
app_id：28987
|
||
uid: UID_jkMs0DaVVwOcBuFPQGzymjCwYVgH
|
||
应用名称：pt_helper
|
||
appToken：AT_ShUnRu2CJRcsqbbW540voVkjMZ1PKjGy
|
||
"""
|
||
# res = NotifyDispatch().send_text(text='66666')
|
||
|
||
res = pt_spider.send_text('666')
|
||
print(res)
|
||
return JsonResponse(res, safe=False)
|
||
|
||
|
||
def do_restart(request):
|
||
try:
|
||
print('重启')
|
||
# print(os.system('pwd'))
|
||
if os.environ.get('CONTAINER_NAME'):
|
||
subprocess.Popen('chmod +x ./restart.sh', shell=True)
|
||
subprocess.Popen('./restart.sh', shell=True)
|
||
return JsonResponse(data=CommonResponse.success(
|
||
msg='重启指令发送成功！!'
|
||
).to_dict(), safe=False)
|
||
return JsonResponse(data=CommonResponse.error(
|
||
msg='未配置CONTAINER_NAME（容器名称）环境变量，请自行重启容器！!'
|
||
).to_dict(), safe=False)
|
||
except Exception as e:
|
||
return JsonResponse(data=CommonResponse.error(
|
||
msg='重启指令发送失败!' + str(e)
|
||
).to_dict(), safe=False)
|
||
|
||
|
||
def do_update(request):
|
||
try:
|
||
print('更新')
|
||
# print(os.system('cat ./update.sh'))
|
||
subprocess.Popen('chmod +x ./update.sh', shell=True)
|
||
|
||
p = subprocess.Popen('./update.sh', shell=True, stdout=subprocess.PIPE, bufsize=1)
|
||
p.wait()
|
||
out = p.stdout.readlines()
|
||
result = []
|
||
for i in out:
|
||
result.append(i.decode('utf8'))
|
||
print(result)
|
||
return JsonResponse(data=CommonResponse.success(
|
||
msg='更新成功！!',
|
||
data=result
|
||
# data={
|
||
# # 'p': str(p.stdout.readlines()).strip("'").strip('[').strip(']').strip()
|
||
# }
|
||
).to_dict(), safe=False)
|
||
except Exception as e:
|
||
return JsonResponse(data=CommonResponse.error(
|
||
msg='更新指令发送失败!' + str(e)
|
||
).to_dict(), safe=False)
|
||
|
||
|
||
def restart_container(request):
|
||
scraper = pt_spider.get_scraper()
|
||
res = scraper.get('https://gitee.com/ngfchl/ptools/raw/master/update.md')
|
||
update_notes = markdown.markdown(res.text, extensions=['tables'])
|
||
print(update_notes)
|
||
return render(request, 'auto_pt/restart.html',
|
||
context={'update_notes': update_notes}
|
||
)
|