You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
flyapps/fir_ser/api/views/receiveudids.py

112 lines
4.7 KiB

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# project: 3月
# author: liuyu
# date: 2020/3/6
from api.utils.app.randomstrings import make_random_uuid
from api.utils.app.supersignutils import udid_bytes_to_dict, get_redirect_server_domain, make_sign_udid_mobile_config, \
get_post_udid_url, get_http_server_domain
from api.models import Apps
from django.views import View
from django.http import HttpResponsePermanentRedirect, FileResponse
from rest_framework.response import Response
from api.tasks import run_sign_task
from api.utils.response import BaseResponse
from api.utils.storage.caches import check_app_permission
from api.utils.throttle import ReceiveUdidThrottle1, ReceiveUdidThrottle2
from fir_ser.celery import app
import logging
from rest_framework.views import APIView
from api.utils.baseutils import get_app_domain_name
from celery.exceptions import TimeoutError
logger = logging.getLogger(__name__)
class IosUDIDView(View):
throttle_classes = [ReceiveUdidThrottle1, ReceiveUdidThrottle2]
def post(self, request, short):
stream_f = str(request.body)
format_udid_info = udid_bytes_to_dict(stream_f)
logger.info(f"short {short} receive new udid {format_udid_info}")
server_domain = get_redirect_server_domain(request)
try:
app_info = Apps.objects.filter(short=short).first()
if app_info:
server_domain = get_redirect_server_domain(request, app_info.user_id, get_app_domain_name(app_info))
if app_info.issupersign and app_info.user_id.supersign_active:
res = check_app_permission(app_info, BaseResponse())
if res.code != 1000:
msg = "&msg=%s" % res.msg
else:
c_task = run_sign_task.apply_async((format_udid_info, short))
task_id = c_task.id
logger.info(f"sign app {app_info} task_id:{task_id}")
try:
result = c_task.get(propagate=False, timeout=2)
except TimeoutError:
result = ''
if c_task.successful():
c_task.forget()
msg = "&msg=%s" % result
else:
msg = "&task_id=%s" % task_id
else:
return HttpResponsePermanentRedirect(
"%s/%s" % (server_domain, short))
else:
return HttpResponsePermanentRedirect(
"%s/%s" % (server_domain, short))
except Exception as e:
msg = "&msg=系统内部错误"
logger.error(f"short {short} receive udid Exception:{e}")
return HttpResponsePermanentRedirect(
"%s/%s?udid=%s%s" % (server_domain, short, format_udid_info.get("udid"), msg))
class TaskView(APIView):
def get(self, request, short):
res = BaseResponse()
task_id = request.query_params.get('task_id', None)
if task_id:
app_info = Apps.objects.filter(short=short).first()
if app_info:
result = app.AsyncResult(task_id)
logger.info(f"app {app_info} sign task state {result.state} AA {result.successful()}")
if result.successful():
res.msg = result.get(propagate=False)
return Response(res.dict)
elif result.state in ['PENDING']:
res.msg = '签名队列中'
elif result.state in ['STARTED']:
res.msg = '正在签名中'
else:
res.msg = ''
res.code = 1001
return Response(res.dict)
res.code = 1002
return Response(res.dict)
class ShowUdidView(View):
def get(self, request):
server_domain = get_http_server_domain(request)
path_info_lists = [server_domain, "look_udid"]
udid_url = "/".join(path_info_lists)
ios_udid_mobile_config = make_sign_udid_mobile_config(udid_url, 'show_udid_info', 'flyapps.cn', '查询设备udid')
response = FileResponse(ios_udid_mobile_config)
response['Content-Type'] = "application/x-apple-aspen-config"
response['Content-Disposition'] = 'attachment; filename=' + make_random_uuid() + '.mobileconfig'
return response
def post(self, request):
stream_f = str(request.body)
format_udid_info = udid_bytes_to_dict(stream_f)
logger.info(f"look_udid receive new udid {format_udid_info}")
server_domain = get_redirect_server_domain(request)
return HttpResponsePermanentRedirect(
"%sudid=%s" % (server_domain, format_udid_info.get("udid")))