#!/usr/bin/env python # -*- coding:utf-8 -*- # project: 3月 # author: liuyu # date: 2020/3/4 import datetime import json import logging from concurrent.futures import ThreadPoolExecutor from django.db.models import Count, Q, Sum from django.http.response import FileResponse from rest_framework.response import Response from rest_framework.views import APIView from api.models import Apps, UserInfo from api.utils.modelutils import PageNumber from api.utils.response import BaseResponse from common.base.baseutils import get_choices_dict, get_choices_name_from_key, AppleDeveloperUid, get_real_ip_address from common.cache.state import CleanSignDataState, MigrateStorageState from common.constants import SignStatus, AppleDeveloperStatus from common.core.auth import ExpiringTokenAuthentication, SuperSignPermission from common.core.sysconfig import Config from common.utils.download import get_app_download_url from xsign.models import AppIOSDeveloperInfo, APPSuperSignUsedInfo, AppUDID, IosDeveloperPublicPoolBill, \ UDIDsyncDeveloper, AppleDeveloperToAppUse, DeveloperAppID, APPToDeveloper, DeveloperDevicesID, \ IosDeveloperBill, AppleSignMessage from xsign.tasks import run_resign_task_do, run_resign_task from xsign.utils.modelutils import get_user_public_used_sign_num, get_user_public_sign_num, check_uid_has_relevant, \ get_developer_devices from xsign.utils.serializer import DeveloperSerializer, SuperSignUsedSerializer, DeviceUDIDSerializer, \ BillInfoSerializer, \ DeveloperDeviceSerializer, AppleDeveloperToAppUseSerializer, AppleDeveloperToAppUseAppsSerializer, \ BillTransferSerializer, AppleSignMessageSerializer from xsign.utils.supersignutils import IosUtils logger = logging.getLogger(__name__) def delete_developer_and_clean(developer_obj, user_obj): if developer_obj and user_obj: logger.error(f"user {user_obj} delete developer {developer_obj}") if developer_obj.certid: IosUtils.clean_developer(developer_obj, user_obj) IosUtils.revoke_developer_cert(developer_obj, user_obj) developer_obj.delete() class DeveloperView(APIView): authentication_classes = [ExpiringTokenAuthentication, ] permission_classes = [SuperSignPermission, ] def get(self, request): res = BaseResponse() issuer_id = request.query_params.get("issuer_id", '') developer_choice = request.query_params.get("developer_choice", None) developer_status_choice = request.query_params.get("developer_status_choice", None) developer_obj = AppIOSDeveloperInfo.objects.filter(user_id=request.user) if developer_choice and developer_choice in ['private', 'public']: if developer_choice == 'public': developer_obj = developer_obj.exclude(appledevelopertoappuse__developerid__isnull=False) else: developer_obj = developer_obj.filter(appledevelopertoappuse__developerid__isnull=False) try: status_choice = json.loads(developer_status_choice) except Exception as e: status_choice = None if status_choice is not None and isinstance(status_choice, list) and status_choice: developer_obj = developer_obj.filter(status__in=status_choice) developer_obj = developer_obj.distinct() res.use_num = get_developer_devices(developer_obj) if issuer_id: developer_obj = developer_obj.filter( Q(developerappid__app_id__bundle_id=issuer_id, developerappid__app_id__user_id=request.user) | Q( issuer_id=issuer_id) | Q(description__contains=issuer_id)) page_obj = PageNumber() app_page_serializer = page_obj.paginate_queryset(queryset=developer_obj.order_by("-updated_time"), request=request, view=self) developer_serializer = DeveloperSerializer(app_page_serializer, many=True) res.data = developer_serializer.data res.count = developer_obj.count() res.status_choices = get_choices_dict(AppIOSDeveloperInfo.status_choices, Config.DEVELOPER_DISABLED_STATUS) res.status_choices.extend([ {'id': 'open_auto_check', 'name': '开启自动检测', 'disabled': False}, {'id': 'close_auto_check', 'name': '关闭自动检测', 'disabled': False}, ]) res.apple_auth_list = get_choices_dict(AppIOSDeveloperInfo.auth_type_choices) return Response(res.dict) def put(self, request): data = request.data res = BaseResponse() issuer_id = data.get("issuer_id", "").strip() if issuer_id: developer_obj = AppIOSDeveloperInfo.objects.filter(user_id=request.user, issuer_id=issuer_id).first() else: act = data.get("act", '').strip() pools = ThreadPoolExecutor(10) if act == "syncalldevice": res = BaseResponse() result_list = [] def run_task(developer_obj): status, result = IosUtils.get_device_from_developer(developer_obj) if not status: result_list.append({'issuer_id': developer_obj.issuer_id, 'msg': result.get("return_info")}) run_queryset = AppIOSDeveloperInfo.objects.filter(user_id=request.user, status__in=Config.DEVELOPER_USE_STATUS).all() devicestatus = data.get("devicestatus", '').strip() udidsearch = data.get("udidsearch", '').strip() if udidsearch: run_queryset = run_queryset.filter(udidsyncdeveloper__udid=udidsearch) if devicestatus: run_queryset = run_queryset.filter(udidsyncdeveloper__status=devicestatus) for developer_s_obj in run_queryset: pools.submit(run_task, developer_s_obj) pools.shutdown() if len(result_list): logger.warning(result_list) res.data = result_list return Response(res.dict) elif act == "checkauth": issuer_ids = data.get("issuer_ids", []) if issuer_ids: result_list = [] def run_task(developer_obj): status, result = IosUtils.active_developer(developer_obj, False) if status: status, result = IosUtils.get_device_from_developer(developer_obj, True) if not status: result_list.append( {'issuer_id': developer_obj.issuer_id, 'msg': result.get("return_info")}) else: result_list.append({'issuer_id': developer_obj.issuer_id, 'msg': result.get("return_info")}) for developer_s_obj in AppIOSDeveloperInfo.objects.filter(user_id=request.user, issuer_id__in=issuer_ids).all(): pools.submit(run_task, developer_s_obj) pools.shutdown() if len(result_list): logger.warning(result_list) res.data = result_list elif act == "setstatus": issuer_ids = data.get("issuer_ids", []) status = data.get("status", None) if status and status in ['open_auto_check', 'close_auto_check']: auto_check = False if status == 'open_auto_check': auto_check = True AppIOSDeveloperInfo.objects.filter(user_id=request.user, issuer_id__in=issuer_ids).update( auto_check=auto_check) return Response(res.dict) if issuer_ids and status is not None and status not in Config.DEVELOPER_DISABLED_STATUS: status_text = get_choices_name_from_key(AppIOSDeveloperInfo.status_choices, status) if status_text: AppIOSDeveloperInfo.objects.filter(user_id=request.user, issuer_id__in=issuer_ids).update( status=status) return Response(res.dict) if developer_obj: act = data.get("act", "").strip() if act: logger.info(f"user {request.user} ios developer {developer_obj} act {act}") if act == "checkauth": status, result = IosUtils.active_developer(developer_obj, False) if status: status, msg = IosUtils.get_device_from_developer(developer_obj) if not status: res.msg = result.get("return_info", "未知错误") res.code = 1008 return Response(res.dict) else: res.code = 1008 res.msg = result.get("return_info", "未知错误") return Response(res.dict) elif act == "ioscert": if not developer_obj.certid: status, result = IosUtils.create_developer_cert(developer_obj, request.user) if status: status, msg = IosUtils.get_device_from_developer(developer_obj) if not status: res.msg = result.get("return_info", "未知错误") res.code = 1008 else: res.code = 1008 res.msg = result.get("return_info") return Response(res.dict) elif act == "checkcert": if developer_obj.certid: status, result = IosUtils.get_developer_cert_info(developer_obj) if status: res.data = result else: res.code = 1008 res.msg = result.get("return_info") return Response(res.dict) elif act in ["renewcert", "cleancert"]: if developer_obj.certid: # clean developer somethings. remove profile and revoke cert if act == 'cleancert': IosUtils.clean_developer(developer_obj, request.user) status, result = IosUtils.revoke_developer_cert(developer_obj, request.user) if status: if act == 'renewcert': AppUDID.objects.filter(udid__developerid=developer_obj).update( sign_status=SignStatus.APP_REGISTRATION_COMPLETE) status, result = IosUtils.create_developer_cert(developer_obj, request.user) # if status: # IosUtils.get_device_from_developer(developer_obj, request.user) # else: # res.code = 1008 # res.msg = result.get("err_info") # return Response(res.dict) else: res.code = 1008 res.msg = result.get("err_info", '') return Response(res.dict) elif act in ["syncdevice", "syncalldevice"]: status, result = IosUtils.get_device_from_developer(developer_obj) if not status: res.code = 1008 res.msg = result.get("return_info") return Response(res.dict) elif act == "cleandevice": with CleanSignDataState(request.user.uid) as state: if state: status, result = IosUtils.clean_developer(developer_obj, request.user, False) if not status: res.code = 1008 res.msg = result.get("err_info") else: res.code = 1008 res.msg = "数据清理中,请耐心等待" return Response(res.dict) elif act == 'disable': developer_obj.status = AppleDeveloperStatus.INACTIVATED developer_obj.save(update_fields=['status']) return Response(res.dict) else: update_fields = [] logger.info(f"user {request.user} ios developer {developer_obj} update input data {data}") logger.info( f"user {request.user} ios developer {developer_obj} update old data {developer_obj.__dict__}") try: usable_number = int(data.get("usable_number", developer_obj.usable_number)) app_limit_number = int(data.get("app_limit_number", developer_obj.app_limit_number)) if 0 <= usable_number <= 100: developer_obj.usable_number = usable_number update_fields.append("usable_number") if 0 <= app_limit_number <= 160: developer_obj.app_limit_number = app_limit_number update_fields.append("app_limit_number") except Exception as e: logger.error( f"developer {developer_obj} usable_number {data.get('usable_number', developer_obj.usable_number)} get failed Exception:{e}") developer_obj.description = data.get("description", developer_obj.description) update_fields.append("description") developer_obj.clean_status = data.get("clean_status", developer_obj.clean_status) update_fields.append("clean_status") developer_obj.auto_check = data.get("auto_check", developer_obj.auto_check) update_fields.append("auto_check") private_key_id = data.get("private_key_id", developer_obj.private_key_id) p8key = data.get("p8key", developer_obj.p8key) if private_key_id != "" and private_key_id != developer_obj.private_key_id: developer_obj.private_key_id = private_key_id developer_obj.status = AppleDeveloperStatus.INACTIVATED update_fields.append("private_key_id") if p8key != "" and p8key != developer_obj.p8key: developer_obj.p8key = p8key developer_obj.status = AppleDeveloperStatus.INACTIVATED update_fields.append("p8key") read_only_mode = data.get("read_only_mode", '') if developer_obj.status == AppleDeveloperStatus.ACTIVATED and read_only_mode == 'on': developer_obj.status = AppleDeveloperStatus.MAINTENANCE update_fields.append("status") try: # update_fields.append("status") developer_obj.save(update_fields=update_fields) logger.info( f"user {request.user} ios developer {developer_obj} update now data {developer_obj.__dict__}") except Exception as e: logger.error( f"user {request.user} ios developer {developer_obj} update error data {data} Exception {e}") return Response(res.dict) def post(self, request): res = BaseResponse() data = request.data data_info = {} if data.get("auth_type") == 0: data_info = { "usable_number": data.get("usable_number", ""), "description": data.get("description", ""), "issuer_id": data.get("issuer_id", ""), "private_key_id": data.get("private_key_id", ""), "p8key": data.get("p8key", ""), "auth_type": 0 } try: logger.error(f"user {request.user} add new developer {data.get('issuer_id', '')} data {data_info}") developer_obj, created = AppIOSDeveloperInfo.objects.update_or_create(user_id=request.user, issuer_id=data_info.get('issuer_id'), defaults=data_info) IosUtils.create_developer_space(developer_obj, request.user) status, result = IosUtils.active_developer(developer_obj, False) if not status: res.code = 1008 res.msg = result.get("return_info", "未知错误") if created: delete_developer_and_clean(developer_obj, request.user) return Response(res.dict) else: IosUtils.get_device_from_developer(developer_obj) except Exception as e: logger.error(f"user {request.user} create developer {data_info} failed Exception:{e}") res.code = 1005 res.msg = f"添加失败 {e}" return Response(res.dict) return self.get(request) def delete(self, request): issuer_id = request.query_params.get("issuer_id", None) if issuer_id: developer_obj = AppIOSDeveloperInfo.objects.filter(user_id=request.user, issuer_id=issuer_id).first() else: return self.get(request) delete_developer_and_clean(developer_obj, request.user) return self.get(request) def base_super_sign_used_info(request): udid = request.query_params.get("udid", None) bundle_id = request.query_params.get("bundleid", None) issuer_id = request.query_params.get("issuer_id", None) mine = True super_sign_used_objs = APPSuperSignUsedInfo.objects.filter(developerid__user_id=request.user, ) if not super_sign_used_objs: super_sign_used_objs = APPSuperSignUsedInfo.objects.filter(user_id=request.user, ) mine = False if issuer_id: super_sign_used_objs = super_sign_used_objs.filter(developerid__issuer_id=issuer_id) if udid: super_sign_used_objs = super_sign_used_objs.filter(udid__udid__udid=udid) if bundle_id: super_sign_used_objs = super_sign_used_objs.filter(app_id__bundle_id=bundle_id) return super_sign_used_objs, mine class SuperSignUsedView(APIView): authentication_classes = [ExpiringTokenAuthentication, ] permission_classes = [SuperSignPermission, ] def get(self, request): res = BaseResponse() super_sign_used_objs, mine = base_super_sign_used_info(request) page_obj = PageNumber() app_page_serializer = page_obj.paginate_queryset(queryset=super_sign_used_objs.order_by("-created_time"), request=request, view=self) app_serializer = SuperSignUsedSerializer(app_page_serializer, many=True, context={'mine': mine, 'user_obj': request.user}) res.data = app_serializer.data res.count = super_sign_used_objs.count() return Response(res.dict) def post(self, request): res = BaseResponse() data = request.data device_udid = data.get('device_udid', '') developer_id = data.get('developer_id', '') bundle_id = data.get('bundle_id', '') other_uid_info = data.get('other_uid', '') if device_udid and developer_id and bundle_id: app_obj = Apps.objects.filter(user_id=request.user, bundle_id=bundle_id).first() if app_obj is None and not developer_id.startswith(Config.DEVELOPER_UID_KEY): if other_uid_info: other_uid = other_uid_info.get('uid') if other_uid and check_uid_has_relevant(request.user.uid, other_uid): app_obj = Apps.objects.filter(bundle_id=bundle_id).first() if developer_id.startswith(Config.DEVELOPER_UID_KEY): developer_id = AppleDeveloperUid().get_decrypt_uid(developer_id.lstrip('T:')) app_to_dev_obj = APPToDeveloper.objects.filter(app_id=app_obj, developerid__issuer_id=developer_id).first() if app_obj and app_to_dev_obj: developer_app_id_obj = DeveloperAppID.objects.filter(app_id=app_obj, developerid__issuer_id=developer_id).first() appudid_obj = AppUDID.objects.filter(app_id=app_obj, udid__udid=device_udid, udid__developerid__issuer_id=developer_id).last() need_download_profile = True if appudid_obj.sign_status in [SignStatus.PROFILE_DOWNLOAD_COMPLETE, SignStatus.SIGNATURE_PACKAGE_COMPLETE]: need_download_profile = False c_task = run_resign_task_do.apply_async((app_obj.pk, app_to_dev_obj.developerid.pk, developer_app_id_obj.aid, need_download_profile, False)) msg = c_task.get(propagate=False) logger.info(f"app {app_obj} run_resign_task msg:{msg}") if c_task.successful(): c_task.forget() app_to_dev_obj = APPToDeveloper.objects.filter(app_id=app_obj, developerid__issuer_id=developer_id).first() res = get_app_download_url(request, res, app_obj.app_id, app_obj.short, app_obj.password, app_to_dev_obj.binary_file, True, device_udid) return Response(res.dict) class AppUDIDUsedView(APIView): authentication_classes = [ExpiringTokenAuthentication, ] permission_classes = [SuperSignPermission, ] def get(self, request): res = BaseResponse() super_sign_used_objs, mine = base_super_sign_used_info(request) page_obj = PageNumber() app_udid_objs = AppUDID.objects.filter(appsupersignusedinfo__in=super_sign_used_objs) app_page_serializer = page_obj.paginate_queryset(queryset=app_udid_objs.order_by("-created_time"), request=request, view=self) app_serializer = DeviceUDIDSerializer(app_page_serializer, many=True, context={'mine': mine, 'user_obj': request.user}) res.data = app_serializer.data res.count = app_udid_objs.count() return Response(res.dict) def delete(self, request): res = BaseResponse() pk = request.query_params.get("id", None) other_uid = request.query_params.get("uid", None) app_id = request.query_params.get("aid", None) disabled = request.query_params.get("disabled", None) if disabled is not None and disabled == '1': disabled = True else: disabled = False app_udid_obj = AppUDID.objects.filter(pk=pk, app_id__user_id=request.user) if not app_udid_obj and check_uid_has_relevant(request.user.uid, other_uid): app_udid_obj = AppUDID.objects.filter(pk=pk, app_id__user_id__uid=other_uid) if app_udid_obj: super_sign_used_obj = APPSuperSignUsedInfo.objects.filter(udid=app_udid_obj.first()).first() if super_sign_used_obj: if super_sign_used_obj.developerid.user_id.pk == request.user.pk: logger.error(f"user {request.user} delete devices {app_udid_obj}") IosUtils.disable_udid(app_udid_obj.first(), app_id, disabled) else: IosUtils.disable_udid(app_udid_obj.first(), app_id) # res.code = 10002 # res.msg = '公共账号池不允许禁用删除' app_udid_obj.delete() else: res.code = 10003 res.msg = '数据异常，删除失败' return Response(res.dict) def post(self, request): res = BaseResponse() pk = request.data.get("id", None) other_uid = request.data.get("uid", None) app_id = request.data.get("aid", None) app_udid_obj = AppUDID.objects.filter(pk=pk, app_id__user_id=request.user) if not app_udid_obj and check_uid_has_relevant(request.user.uid, other_uid): app_udid_obj = AppUDID.objects.filter(pk=pk, app_id__user_id__uid=other_uid) if app_udid_obj: super_sign_used_obj = APPSuperSignUsedInfo.objects.filter(udid=app_udid_obj.first()).first() if super_sign_used_obj: run_resign_task(app_id, need_download_profile=True, force=True, developers_filter=[super_sign_used_obj.developerid]) else: res.code = 10003 res.msg = '数据异常，重签名失败' return Response(res.dict) class DeveloperDeviceView(APIView): authentication_classes = [ExpiringTokenAuthentication, ] permission_classes = [SuperSignPermission, ] def get(self, request): res = BaseResponse() udid = request.query_params.get("udid", None) issuer_id = request.query_params.get("issuer_id", None) device_status = request.query_params.get("devicestatus", None) super_sign_used_objs = UDIDsyncDeveloper.objects.filter(developerid__user_id=request.user, ) if device_status: super_sign_used_objs = super_sign_used_objs.filter(status=device_status) if issuer_id: super_sign_used_objs = super_sign_used_objs.filter(developerid__issuer_id=issuer_id) if udid: super_sign_used_objs = super_sign_used_objs.filter(udid=udid) page_obj = PageNumber() app_page_serializer = page_obj.paginate_queryset(queryset=super_sign_used_objs.order_by('-id'), request=request, view=self) app_serializer = DeveloperDeviceSerializer(app_page_serializer, many=True) res.data = app_serializer.data res.status_choices = get_choices_dict(UDIDsyncDeveloper.status_choices) res.count = super_sign_used_objs.count() return Response(res.dict) def put(self, request): res = BaseResponse() developer_id = request.data.get("developer_id", None) udid = request.data.get("udid", None) disabled = request.data.get("disabled", None) if developer_id and udid and disabled is not None and disabled in [0, 1]: developer_obj = AppIOSDeveloperInfo.objects.filter(user_id=request.user, issuer_id=developer_id).first() udid_sync_obj_queryset = UDIDsyncDeveloper.objects.filter(udid=udid, developerid=developer_obj).all() if disabled == 1: for udid_sync_obj in udid_sync_obj_queryset: IosUtils.do_enable_device_by_sync(developer_obj, udid_sync_obj) else: app_udid_obj_list = AppUDID.objects.filter(udid__udid=udid, app_id__user_id=request.user, udid__developerid=developer_obj).all() for app_udid_obj in app_udid_obj_list: q_infos = DeveloperDevicesID.objects.filter(udid=app_udid_obj.udid, developerid=developer_obj).values( 'app_id').all().distinct() for app_id_info in q_infos: logger.error(f"user {request.user} delete devices {app_udid_obj}") IosUtils.disable_udid(app_udid_obj, app_id_info.get('app_id'), True) AppUDID.objects.filter(pk=app_udid_obj.pk).delete() for udid_sync_obj in udid_sync_obj_queryset: IosUtils.do_disable_device_by_sync(developer_obj, udid_sync_obj) return Response(res.dict) class SuperSignCertView(APIView): authentication_classes = [ExpiringTokenAuthentication, ] permission_classes = [SuperSignPermission, ] def get(self, request): issuer_id = request.query_params.get('issuer_id', None) if issuer_id: developer_obj = AppIOSDeveloperInfo.objects.filter(user_id=request.user, issuer_id=issuer_id).first() resign_app_obj = IosUtils.get_resign_obj(request.user, developer_obj) if not resign_app_obj.check_p12_exists(): resign_app_obj.make_p12_from_cert(developer_obj.certid) if developer_obj: zip_file_path = IosUtils.zip_cert(request.user, developer_obj) response = FileResponse(open(zip_file_path, 'rb')) response['Content-Type'] = "application/octet-stream" response["Access-Control-Expose-Headers"] = "Content-Disposition" response['Content-Disposition'] = 'attachment; filename=' + developer_obj.certid + '.zip' return response res = BaseResponse() return Response(res.dict) def post(self, request): res = BaseResponse() issuer_id = request.data.get('issuer_id', None) if issuer_id: developer_obj = AppIOSDeveloperInfo.objects.filter(user_id=request.user, issuer_id=issuer_id).first() if developer_obj: resign_app_obj = IosUtils.get_resign_obj(request.user, developer_obj) status, result = resign_app_obj.make_cert_from_p12(request.data.get('cert_pwd', ''), request.data.get('cert_content', None)) if status: status, _ = IosUtils.auto_get_cert_id_by_p12(developer_obj, request.user) if status: resign_app_obj.write_cert() else: res.code = 1003 res.msg = '证书未在开发者账户找到，请检查证书是否属于该开发者' else: res.code = 1002 res.msg = str(result['err_info']) return Response(res.dict) return Response(res.dict) class DeviceUsedBillView(APIView): authentication_classes = [ExpiringTokenAuthentication, ] permission_classes = [SuperSignPermission, ] def get(self, request): res = BaseResponse() udid = request.query_params.get("udid", None) act = request.query_params.get("act", None) receive_user_id_list = IosDeveloperBill.objects.filter(user_id=request.user, to_user_id__isnull=False, status=2).values( 'to_user_id').distinct() user_used_list = IosDeveloperPublicPoolBill.objects.filter( Q(user_id=request.user) | Q(user_id_id__in=receive_user_id_list)) page_obj = PageNumber() if udid: user_used_list = user_used_list.filter(udid=udid) else: user_used_list = user_used_list.filter(udid__isnull=False) if act and act == 'info': app_page_serializer = page_obj.paginate_queryset(queryset=user_used_list.order_by("-created_time"), request=request, view=self) app_serializer = BillInfoSerializer(app_page_serializer, many=True, context={'user_obj': request.user}) res.data = app_serializer.data res.count = user_used_list.count() else: if udid: user_used_list = user_used_list.filter(udid=udid) user_used_list3 = user_used_list.values('udid', 'product', 'version', 'udid_sync_info_id').annotate( counts=Count('udid')) res.count = user_used_list3.count() user_used_list3 = page_obj.paginate_queryset(queryset=user_used_list3.order_by('-udid_sync_info_id'), request=request, view=self) for user_used in user_used_list3: user_used['counts'] = user_used_list.filter(udid=user_used.get('udid')).count() res.data = user_used_list3 res.balance_info = { 'used_balance': get_user_public_used_sign_num(request.user), 'all_balance': get_user_public_sign_num(request.user) } return Response(res.dict) class DeviceTransferBillView(APIView): authentication_classes = [ExpiringTokenAuthentication, ] permission_classes = [SuperSignPermission, ] def get(self, request): res = BaseResponse() uidsearch = request.query_params.get("uidsearch", None) user_used_list = IosDeveloperBill.objects.filter( Q(user_id=request.user) | Q(to_user_id=request.user)).distinct() page_obj = PageNumber() if uidsearch: user_used_list = user_used_list.filter(Q(user_id__uid=uidsearch) | Q(to_user_id__uid=uidsearch)) app_page_serializer = page_obj.paginate_queryset(queryset=user_used_list.order_by("-created_time"), request=request, view=self) app_serializer = BillTransferSerializer(app_page_serializer, many=True, context={'user_obj': request.user}) res.data = app_serializer.data res.count = user_used_list.count() res.balance_info = { 'used_balance': get_user_public_used_sign_num(request.user), 'all_balance': get_user_public_sign_num(request.user) } return Response(res.dict) def post(self, request): res = BaseResponse() uid = request.data.get('uid') number = request.data.get('number') if uid and number: to_user_obj = UserInfo.objects.filter(uid=uid, is_active=True, supersign_active=True).first() if to_user_obj: if isinstance(number, int) and 0 < number < 99999: user_obj = request.user if user_obj.pk != to_user_obj.pk: try: use_num = get_developer_devices(AppIOSDeveloperInfo.objects.filter(user_id=user_obj)) all_balance = use_num.get('max_total', 0) if all_balance > 0 and number <= all_balance: bill_obj = IosDeveloperBill.objects.filter(user_id=user_obj, to_user_id=to_user_obj, status=2).first() if bill_obj: bill_obj.number = number + bill_obj.number if bill_obj.number >= all_balance: bill_obj.number = all_balance bill_obj.remote_addr = get_real_ip_address(request) bill_obj.description = f'{user_obj.first_name} 共享给 {to_user_obj.first_name} {bill_obj.number} 设备数' bill_obj.save(update_fields=['number', 'remote_addr', 'description']) else: IosDeveloperBill.objects.create(user_id=user_obj, to_user_id=to_user_obj, status=2, number=number, remote_addr=get_real_ip_address(request), description=f'{user_obj.first_name} 共享给 {to_user_obj.first_name} {number} 设备数') return Response(res.dict) else: res.msg = f'设备余额不足,当前设备余额最大为 {all_balance}' except Exception as e: res.msg = str(e) else: res.msg = '用户不合法' else: res.msg = '划转数量异常' else: res.msg = '用户信息不存在' else: res.msg = '参数有误' res.code = 1003 return Response(res.dict) def put(self, request): res = BaseResponse() uid = request.data.get("uid", None) status = request.data.get("status", None) number = request.data.get("number", None) if uid and status and number: if check_uid_has_relevant(uid, request.user.uid): if MigrateStorageState(uid).get_state(): res.code = 1008 res.msg = "数据迁移中，无法处理该操作" return Response(res.dict) bill_obj = IosDeveloperBill.objects.filter(user_id__uid=uid, to_user_id=request.user, status=status, number=abs(int(number))).first() if bill_obj: for app_obj in Apps.objects.filter(user_id=request.user, type=1): count = APPToDeveloper.objects.filter(app_id=app_obj).count() if app_obj.issupersign or count > 0: logger.info(f"app_id:{app_obj} is super_sign ,clean IOS developer") IosUtils.clean_app_by_user_obj(app_obj) return Response(res.dict) res.code = 1003 res.msg = '转移记录不存在' return Response(res.dict) def delete(self, request): res = BaseResponse() uid = request.query_params.get("uid", None) status = request.query_params.get("status", None) number = request.query_params.get("number", None) if MigrateStorageState(request.user.uid).get_state(): res.code = 1008 res.msg = "数据迁移中，无法处理该操作" return Response(res.dict) if uid and status and number: bill_obj = IosDeveloperBill.objects.filter(user_id=request.user, to_user_id__uid=uid, status=status, number=abs(int(number))).first() if bill_obj: target_user_obj = UserInfo.objects.filter(uid=uid).first() if target_user_obj: for app_obj in Apps.objects.filter(user_id=target_user_obj, type=1): app_obj.issupersign = False app_obj.save(update_fields=['issupersign']) count = APPToDeveloper.objects.filter(app_id=app_obj).count() if app_obj.issupersign or count > 0: logger.info(f"app_id:{app_obj} is super_sign ,clean IOS developer") IosUtils.clean_app_by_user_obj(app_obj) bill_obj.status = 1 bill_obj.save(update_fields=['status']) else: res.code = 1003 res.msg = '转移记录不存在' return Response(res.dict) class DeviceUsedRankInfoView(APIView): authentication_classes = [ExpiringTokenAuthentication, ] permission_classes = [SuperSignPermission, ] def get(self, request): res = BaseResponse() page_obj = PageNumber() search_key = request.query_params.get("appnamesearch") start_time = request.query_params.get("start_time") end_time = request.query_params.get("end_time") receive_user_id_list = IosDeveloperBill.objects.filter(user_id=request.user, to_user_id__isnull=False, status=2).values( 'to_user_id').distinct() app_used_sign_objs = APPSuperSignUsedInfo.objects.filter( Q(user_id=request.user) | Q(user_id_id__in=receive_user_id_list)) if search_key: app_used_sign_objs = app_used_sign_objs.filter( Q(app_id__name=search_key) | Q(app_id__bundle_id=search_key) | Q( developerid__issuer_id=search_key)) if end_time and start_time: try: start_time = datetime.date.fromtimestamp(int(start_time) / 1000) end_time = datetime.date.fromtimestamp(int(end_time) / 1000) + datetime.timedelta(days=1) app_used_sign_objs = app_used_sign_objs.filter(created_time__range=[start_time, end_time]) except Exception as e: logger.error(f"get time range failed {e}") app_used_sign_objs = app_used_sign_objs.values('app_id__app_id', 'app_id__name', 'app_id__bundle_id').annotate( count=Count('app_id__app_id')).order_by('-count') res.count = app_used_sign_objs.count() res.number = app_used_sign_objs.aggregate(Sum('count')).get('count__sum') app_used_sign_infos = page_obj.paginate_queryset(queryset=app_used_sign_objs, request=request, view=self) new_app_used_sign_infos = [] for app_used_info in app_used_sign_infos: new_app_used_info = {} for key, value in app_used_info.items(): new_key = key.replace('app_id__', '') new_app_used_info[new_key] = value new_app_used_sign_infos.append(new_app_used_info) res.data = new_app_used_sign_infos return Response(res.dict) class AppleDeveloperBindAppsView(APIView): authentication_classes = [ExpiringTokenAuthentication, ] permission_classes = [SuperSignPermission, ] def get(self, request): res = BaseResponse() act = request.query_params.get("act", '') issuer_id = request.query_params.get("issuer_id", '') app_id = request.query_params.get("app_id", '') if act == 'apps': res.app_limit_number = 0 apps_obj_list = Apps.objects.filter(user_id=request.user, type=1).all() app_serializer = AppleDeveloperToAppUseAppsSerializer(apps_obj_list, many=True, context={'issuer_id': issuer_id}) if issuer_id: developer_obj = AppIOSDeveloperInfo.objects.filter(user_id=request.user, issuer_id=issuer_id).first() if developer_obj: res.app_limit_number = developer_obj.app_limit_number elif act == 'developer': developer_obj = AppIOSDeveloperInfo.objects.filter(user_id=request.user).all() app_serializer = DeveloperSerializer(developer_obj, many=True, context={'app_id': app_id}) else: apple_to_app_obj = AppleDeveloperToAppUse.objects.filter(app_id__user_id=request.user) if issuer_id: apple_to_app_obj = apple_to_app_obj.filter(developerid__issuer_id=issuer_id) if app_id: apple_to_app_obj = apple_to_app_obj.filter(app_id__app_id=app_id) app_serializer = AppleDeveloperToAppUseSerializer(apple_to_app_obj.all(), many=True) res.data = app_serializer.data return Response(res.dict) def post(self, request): res = BaseResponse() issuer_id = request.data.get('issuer_id', '') app_id = request.data.get('app_id', '') choices_data = request.data.get('choices_data', []) logger.info(f"change apple to app data:{request.data}") many_obj = [] if app_id: app_obj = Apps.objects.filter(app_id=app_id, type=1, user_id=request.user).first() issuer_id_list = [apple_to_app['developerid__issuer_id'] for apple_to_app in AppleDeveloperToAppUse.objects.filter(app_id=app_obj).values( 'developerid__issuer_id').all().distinct()] add_issuer_ids = list(set(choices_data) - set(issuer_id_list)) remove_issuer_ids = list(set(issuer_id_list) - set(choices_data)) for item in add_issuer_ids: developer_obj = AppIOSDeveloperInfo.objects.filter(issuer_id=item, user_id=request.user, status__in=Config.DEVELOPER_USE_STATUS, certid__isnull=False).first() developer_aid_obj = DeveloperAppID.objects.filter(developerid=developer_obj) is_exist = developer_aid_obj.filter(app_id=app_obj).count() if developer_obj: app_flag = False if is_exist: app_flag = True else: if developer_aid_obj.count() < developer_obj.app_limit_number: app_flag = True if app_flag: many_obj.append(AppleDeveloperToAppUse(developerid=developer_obj, app_id=app_obj)) AppleDeveloperToAppUse.objects.filter(developerid__issuer_id__in=remove_issuer_ids, app_id=app_obj).delete() if issuer_id: developer_obj = AppIOSDeveloperInfo.objects.filter(issuer_id=issuer_id, user_id=request.user).first() app_id_list = [apple_to_app['app_id__app_id'] for apple_to_app in AppleDeveloperToAppUse.objects.filter(developerid=developer_obj).values( 'app_id__app_id').all().distinct()] add_app_ids = list(set(choices_data) - set(app_id_list)) remove_app_ids = list(set(app_id_list) - set(choices_data)) exist_app_id_queryset = DeveloperAppID.objects.filter(developerid=developer_obj).values( 'app_id__app_id').all() exist_app_ids = [exist_app_id_obj.get('app_id__app_id') for exist_app_id_obj in exist_app_id_queryset] can_add_number = developer_obj.app_limit_number - len(exist_app_ids) for item in add_app_ids: app_obj = Apps.objects.filter(app_id=item, type=1, user_id=request.user).first() if app_obj: app_flag = False if item in exist_app_ids: app_flag = True else: if can_add_number > 0: can_add_number -= 1 app_flag = True if app_flag: many_obj.append(AppleDeveloperToAppUse(developerid=developer_obj, app_id=app_obj)) AppleDeveloperToAppUse.objects.filter(developerid=developer_obj, app_id__app_id__in=remove_app_ids, app_id__user_id=request.user).delete() if many_obj: AppleDeveloperToAppUse.objects.bulk_create(many_obj) return Response(res.dict) def put(self, request): res = BaseResponse() issuer_id = request.data.get('issuer_id', '') app_id = request.data.get('app_id', '') infos = request.data.get('infos', {}) logger.info(f"change apple to app data:{request.data}") if issuer_id and not app_id: app_id = infos.get('app_id', '') if app_id and not issuer_id: issuer_id = infos.get('issuer_id', '') if app_id and issuer_id: try: app_usable_number = int(infos.get('app_usable_number', 100)) if 0 < app_usable_number < 101: AppleDeveloperToAppUse.objects.filter(developerid__issuer_id=issuer_id, app_id__app_id=app_id, app_id__user_id=request.user).update( usable_number=int(app_usable_number)) except Exception as e: logger.error(f'update app developer used failed infos:{infos} Exception:{e}') return Response(res.dict) class SignOperateMessageView(APIView): authentication_classes = [ExpiringTokenAuthentication, ] permission_classes = [SuperSignPermission, ] def get(self, request): res = BaseResponse() issuer_id = request.query_params.get("issuer_id", None) operate_status = request.query_params.get("operate_status", None) sign_message_queryset = AppleSignMessage.objects.filter(user_id=request.user) if issuer_id: sign_message_queryset = sign_message_queryset.filter(developerid__issuer_id=issuer_id) if operate_status: sign_message_queryset = sign_message_queryset.filter(operate_status=operate_status) page_obj = PageNumber() page_serializer = page_obj.paginate_queryset(queryset=sign_message_queryset.order_by("-operate_time"), request=request, view=self) message_serializer = AppleSignMessageSerializer(page_serializer, many=True, ) res.data = message_serializer.data res.count = sign_message_queryset.count() res.status_choices = get_choices_dict(AppleSignMessage.status_choices) return Response(res.dict)