#!/usr/bin/env python # -*- coding:utf-8 -*- # project: 3月 # author: liuyu # date: 2020/3/6 import logging from celery.exceptions import TimeoutError from django.http import HttpResponsePermanentRedirect, FileResponse, HttpResponse from django.views import View from rest_framework.response import Response from rest_framework.views import APIView from api.models import Apps from api.tasks import run_sign_task from api.utils.app.supersignutils import udid_bytes_to_dict, make_sign_udid_mobile_config from api.utils.modelutils import get_redirect_server_domain, add_remote_info_from_request, \ get_app_download_uri from api.utils.response import BaseResponse from common.base.baseutils import get_real_ip_address, make_random_uuid, get_server_domain_from_request from common.core.sysconfig import Config from common.core.throttle import ReceiveUdidThrottle1, ReceiveUdidThrottle2 from common.utils.caches import check_app_permission from fir_ser.celery import app logger = logging.getLogger(__name__) class IosUDIDView(APIView): 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_obj = Apps.objects.filter(short=short).first() if app_obj: server_domain = get_app_download_uri(request, app_obj.user_id, app_obj, preview=False) if app_obj.issupersign and app_obj.user_id.supersign_active: res = check_app_permission(app_obj, BaseResponse()) if res.code != 1000: msg = "&msg=%s" % res.msg else: client_ip = get_real_ip_address(request) logger.info(f"client_ip {client_ip} short {short} app_info {app_obj}") # from api.utils.app.supersignutils import IosUtils # ios_obj = IosUtils(format_udid_info, app_obj.user_id, app_obj) # ios_obj.sign_ipa(client_ip) # return Response('ok') c_task = run_sign_task.apply_async((format_udid_info, short, client_ip)) add_remote_info_from_request(request, f'{app_obj}-{format_udid_info}') task_id = c_task.id logger.info(f"sign app {app_obj} task_id:{task_id}") try: result = c_task.get(propagate=False, timeout=3) except TimeoutError: logger.error(f"get task task_id:{task_id} result timeout") 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): udid = request.GET.get("udid") if udid: return HttpResponse("udid: %s" % udid) server_domain = get_server_domain_from_request(request, Config.POST_UDID_DOMAIN) path_info_lists = [server_domain, "show_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"show_udid receive new udid {format_udid_info}") server_domain = get_server_domain_from_request(request, Config.POST_UDID_DOMAIN) return HttpResponsePermanentRedirect( "%s/show_udid?udid=%s" % (server_domain, format_udid_info.get("udid")))