mirror of
https://github.com/xingsu1021/pthelper.git
synced 2026-06-16 15:07:07 +08:00
374 lines
11 KiB
Python
374 lines
11 KiB
Python
#coding=utf-8
|
||
from django.http import JsonResponse, StreamingHttpResponse,HttpResponse
|
||
from django.db.models import Q
|
||
from django.contrib.auth.decorators import login_required
|
||
|
||
from .models import SiteConfig, SiteRank, SiteInfo
|
||
import simplejson as json
|
||
import io
|
||
from urllib import parse
|
||
import tempfile
|
||
|
||
#直接使用检查是否管理员
|
||
#from django.contrib.auth.decorators import user_passes_test
|
||
#@user_passes_test(lambda u: u.is_superuser)
|
||
|
||
@login_required
|
||
def siteconfig(request):
|
||
"""
|
||
提供相应的数据
|
||
"""
|
||
|
||
#得到偏移量
|
||
pageIndex= int(request.GET.get('page',1))
|
||
#得到查询个数
|
||
pageSize = int(request.GET.get('limit',10))
|
||
|
||
#搜索字段
|
||
searchKey = request.GET.get('searchKey','')
|
||
searchValue = request.GET.get('searchValue','')
|
||
#去掉空格
|
||
searchValue = searchValue.strip()
|
||
|
||
#得到排序字段
|
||
sort = request.GET.get('sort','id')
|
||
#得到排序规则
|
||
order_by_type = request.GET.get('order','desc')
|
||
|
||
if order_by_type == 'desc':
|
||
order_by = sort
|
||
else:
|
||
order_by = '-' + sort
|
||
|
||
data = {}
|
||
data['code'] = 0
|
||
data['msg'] = ""
|
||
data['data'] = []
|
||
|
||
if searchValue == '':
|
||
|
||
data['count'] = SiteConfig.objects.count()
|
||
if pageIndex == 1:
|
||
ormdata = SiteConfig.objects.order_by(order_by)[:pageSize] #offset:limit
|
||
else:
|
||
ormdata = SiteConfig.objects.order_by(order_by)[pageSize*(pageIndex-1):pageIndex * pageSize]
|
||
else:
|
||
|
||
data['count'] = SiteConfig.objects.filter(Q(name__icontains=searchValue)|Q(name_cn__icontains=searchValue)).count()
|
||
if pageIndex == 1:
|
||
ormdata = SiteConfig.objects.filter(Q(name__icontains=searchValue)|Q(name_cn__icontains=searchValue)).order_by(order_by)[:pageSize] #offset:limit
|
||
else:
|
||
ormdata = SiteConfig.objects.filter(Q(name__icontains=searchValue)|Q(name_cn__icontains=searchValue)).order_by(order_by)[pageSize*(pageIndex-1):pageIndex * pageSize]
|
||
|
||
for i in ormdata:
|
||
|
||
data['data'].append({"id":i.id,
|
||
"name":i.name,
|
||
"name_cn":i.name_cn,
|
||
"index_url":i.index_url,
|
||
})
|
||
#返回json串
|
||
#return HttpResponse(json.dumps(data,ensure_ascii = False), "application/json")
|
||
return JsonResponse(data)
|
||
|
||
@login_required
|
||
def siterankconfig(request):
|
||
"""
|
||
提供相应的数据
|
||
"""
|
||
|
||
#得到偏移量
|
||
pageIndex= int(request.GET.get('page',1))
|
||
#得到查询个数
|
||
pageSize = int(request.GET.get('limit',10))
|
||
|
||
#搜索字段
|
||
searchKey = request.GET.get('searchKey','')
|
||
searchValue = request.GET.get('searchValue','')
|
||
#去掉空格
|
||
searchValue = searchValue.strip()
|
||
|
||
#得到排序字段
|
||
sort = request.GET.get('sort','id')
|
||
#得到排序规则
|
||
order_by_type = request.GET.get('order','desc')
|
||
|
||
if order_by_type == 'desc':
|
||
order_by = sort
|
||
else:
|
||
order_by = '-' + sort
|
||
|
||
data = {}
|
||
data['code'] = 0
|
||
data['msg'] = ""
|
||
data['data'] = []
|
||
|
||
if searchValue == '':
|
||
|
||
data['count'] = SiteRank.objects.count()
|
||
if pageIndex == 1:
|
||
ormdata = SiteRank.objects.order_by(order_by)[:pageSize] #offset:limit
|
||
else:
|
||
ormdata = SiteRank.objects.order_by(order_by)[pageSize*(pageIndex-1):pageIndex * pageSize]
|
||
else:
|
||
|
||
data['count'] = SiteRank.objects.filter(Q(name__icontains=searchValue)|Q(siteconfig_name__icontains=searchValue)).count()
|
||
if pageIndex == 1:
|
||
ormdata = SiteRank.objects.filter(Q(name__icontains=searchValue)|Q(siteconfig_name__icontains=searchValue)).order_by(order_by)[:pageSize] #offset:limit
|
||
else:
|
||
ormdata = SiteRank.objects.filter(Q(name__icontains=searchValue)|Q(siteconfig_name__icontains=searchValue)).order_by(order_by)[pageSize*(pageIndex-1):pageIndex * pageSize]
|
||
|
||
for i in ormdata:
|
||
|
||
data['data'].append({"id":i.id,
|
||
"name":i.name,
|
||
"siteconfig_name":i.siteconfig_name,
|
||
"download":i.download,
|
||
"up_time":i.up_time,
|
||
"upload":i.upload,
|
||
"ratio":i.ratio,
|
||
"privilege":i.privilege,
|
||
"serial_number":i.serial_number,
|
||
})
|
||
#返回json串
|
||
#return HttpResponse(json.dumps(data,ensure_ascii = False), "application/json")
|
||
return JsonResponse(data)
|
||
|
||
|
||
|
||
@login_required
|
||
def siteinfo(request):
|
||
"""
|
||
提供相应的数据
|
||
"""
|
||
|
||
#得到偏移量
|
||
pageIndex= int(request.GET.get('page',1))
|
||
#得到查询个数
|
||
pageSize = int(request.GET.get('limit',10))
|
||
|
||
#搜索字段
|
||
searchKey = request.GET.get('searchKey','')
|
||
searchValue = request.GET.get('searchValue','')
|
||
#去掉空格
|
||
searchValue = searchValue.strip()
|
||
|
||
#得到排序字段
|
||
sort = request.GET.get('sort','id')
|
||
#得到排序规则
|
||
order_by_type = request.GET.get('order','desc')
|
||
|
||
if order_by_type == 'desc':
|
||
order_by = sort
|
||
else:
|
||
order_by = '-' + sort
|
||
|
||
data = {}
|
||
data['code'] = 0
|
||
data['msg'] = ""
|
||
data['data'] = []
|
||
|
||
if searchValue == '':
|
||
|
||
data['count'] = SiteInfo.objects.count()
|
||
if pageIndex == 1:
|
||
ormdata = SiteInfo.objects.order_by(order_by)[:pageSize] #offset:limit
|
||
else:
|
||
ormdata = SiteInfo.objects.order_by(order_by)[pageSize*(pageIndex-1):pageIndex * pageSize]
|
||
else:
|
||
|
||
data['count'] = SiteInfo.objects.filter(Q(siteconfig_name__icontains=searchValue)).count()
|
||
if pageIndex == 1:
|
||
ormdata = SiteInfo.objects.filter(Q(siteconfig_name__icontains=searchValue)).order_by(order_by)[:pageSize] #offset:limit
|
||
else:
|
||
ormdata = SiteInfo.objects.filter(Q(siteconfig_name__icontains=searchValue)).order_by(order_by)[pageSize*(pageIndex-1):pageIndex * pageSize]
|
||
|
||
for i in ormdata:
|
||
|
||
siteconfig_ormdata = SiteConfig.objects.get(name=i.siteconfig_name)
|
||
|
||
data['data'].append({"id":i.id,
|
||
"siteconfig_name":i.siteconfig_name,
|
||
'siteconfig_name_cn':siteconfig_ormdata.name_cn,
|
||
"cookie":i.cookie,
|
||
"passkey":i.passkey,
|
||
})
|
||
#返回json串
|
||
#return HttpResponse(json.dumps(data,ensure_ascii = False), "application/json")
|
||
return JsonResponse(data)
|
||
|
||
def get_stream(data):
|
||
# 开始这里我用ByteIO流总是出错，但是后来参考廖雪峰网站用StringIO就没问题
|
||
file = io.StringIO()
|
||
data=json.dumps(data)
|
||
file.write(data)
|
||
res=file.getvalue()
|
||
file.close()
|
||
return res
|
||
|
||
#==================
|
||
@login_required
|
||
def export(request):
|
||
"""
|
||
文件导出
|
||
"""
|
||
|
||
#得到批量或者单个要删除的id
|
||
#ids = request.POST.getlist("ids[]")
|
||
#print("ids====>",ids)
|
||
ids = []
|
||
#fetch方法
|
||
if request.method == "PUT":
|
||
body = request.body
|
||
json_body = json.loads(body)
|
||
|
||
ids = json_body['ids']
|
||
|
||
data = []
|
||
|
||
if len(ids) == 0:
|
||
ormdata = SiteInfo.objects.all()
|
||
else:
|
||
ormdata = SiteInfo.objects.filter(id__in=ids)
|
||
|
||
for i in ormdata:
|
||
|
||
data.append({"name":i.siteconfig_name,
|
||
"cookie":i.cookie,
|
||
"passkey":i.passkey,
|
||
})
|
||
#print(data)
|
||
json_stream=get_stream(data)
|
||
response = HttpResponse(content_type='application/json')
|
||
|
||
response['Content-Disposition'] = 'attachment;filename=' + parse.quote('pthelper站点配置信息') + '.json'
|
||
response.write(json_stream)
|
||
|
||
return response
|
||
|
||
#==================
|
||
@login_required
|
||
def importJson(request):
|
||
"""
|
||
文件上传
|
||
"""
|
||
data = {}
|
||
data['code'] = 0
|
||
data['msg'] = ""
|
||
data['data'] = []
|
||
|
||
if request.method == 'POST':
|
||
file_obj = request.FILES.get('file')
|
||
#data = get_stream(file_obj)
|
||
|
||
file_temp = tempfile.NamedTemporaryFile()
|
||
file_temp.write(file_obj.read())
|
||
#res=file_temp.getvalue()
|
||
file_temp.seek(0)
|
||
data = file_temp.read()
|
||
file_temp.close()
|
||
|
||
#print(data)
|
||
try:
|
||
json_data = json.loads(data)
|
||
for i in json_data:
|
||
siteconfig_name = i['name']
|
||
cookie = i['cookie']
|
||
passkey = i['passkey']
|
||
|
||
get_site = SiteInfo.objects.filter(siteconfig_name=siteconfig_name).count()
|
||
if get_site == 0:
|
||
SiteInfo.objects.create(siteconfig_name = siteconfig_name,
|
||
cookie = cookie,
|
||
passkey = passkey,
|
||
)
|
||
else:
|
||
#更新站点配置
|
||
SiteInfo.objects.filter(siteconfig_name=siteconfig_name).update(cookie = cookie,
|
||
passkey=passkey,
|
||
)
|
||
response_data={"code":1,"msg":"更新成功"}
|
||
|
||
except:
|
||
response_data={"code":0,"msg":"非法文件"}
|
||
|
||
return JsonResponse(response_data)
|
||
|
||
#==================
|
||
@login_required
|
||
def siteconfigname(request):
|
||
"""
|
||
网站集合
|
||
xmSelect
|
||
"""
|
||
|
||
data = {}
|
||
data['code'] = 0
|
||
data['msg'] = ""
|
||
data['data'] = []
|
||
|
||
#得到排序字段
|
||
sort = request.GET.get('sort','name')
|
||
#得到排序规则
|
||
order_by_type = request.GET.get('order','desc')
|
||
|
||
if order_by_type == 'desc':
|
||
order_by = sort
|
||
else:
|
||
order_by = '-' + sort
|
||
|
||
#获取所有记录
|
||
data['count'] = SiteConfig.objects.count()
|
||
|
||
ormdata = SiteConfig.objects.order_by(order_by).all()
|
||
|
||
|
||
for i in ormdata:
|
||
|
||
data['data'].append({
|
||
"name":i.name,
|
||
"name_en_cn":i.name + "(" + i.name_cn + ")"
|
||
})
|
||
|
||
return JsonResponse(data)
|
||
|
||
#==================
|
||
@login_required
|
||
def siteconfigname2siteinfo(request):
|
||
"""
|
||
网站集合,去掉已经创建的站点名称
|
||
xmSelect
|
||
"""
|
||
|
||
data = {}
|
||
data['code'] = 0
|
||
data['msg'] = ""
|
||
data['data'] = []
|
||
|
||
#得到排序字段
|
||
sort = request.GET.get('sort','name')
|
||
#得到排序规则
|
||
order_by_type = request.GET.get('order','desc')
|
||
|
||
if order_by_type == 'desc':
|
||
order_by = sort
|
||
else:
|
||
order_by = '-' + sort
|
||
|
||
#获取所有记录
|
||
data['count'] = SiteConfig.objects.count()
|
||
|
||
ormdata = SiteConfig.objects.order_by(order_by).all()
|
||
|
||
#获取siteinfo中配置的站点名
|
||
ormdata_siteinfo = list(SiteInfo.objects.values_list('siteconfig_name', flat=True))
|
||
|
||
for i in ormdata:
|
||
#忽略已经配置的站点
|
||
if i.name not in ormdata_siteinfo:
|
||
data['data'].append({
|
||
"name":i.name,
|
||
"name_en_cn":i.name + "(" + i.name_cn + ")"
|
||
})
|
||
|
||
return JsonResponse(data) |