ASJPush/Controller/TransparentTransmissionPushController.py

# @Author    : peng
# @File      : TransparentTransmissionPushController.py
# @Time      : 2024年7月9日17:22:19
import json
import threading
import time
import requests
from django.http import HttpResponse
from django.views import View

from Model.models import UidPushModel, GatewayPush
from Object.ResponseObject import ResponseObject
from Service.HuaweiPushService.HuaweiPushService import HuaweiPushObject
from Service.PushService import PushObject
from AnsjerPush.config import XM_PUSH_CHANNEL_ID, HONORPUSH_CONFIG, LOGGER, APP_MAPPING
import logging

ERROR_INFO_LOGGER = logging.getLogger('error_info')


class TransparentTransmissionPushView(View):

    def get(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        operation = kwargs.get('operation')
        return self.validation(request.GET, operation)

    def post(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        operation = kwargs.get('operation')
        return self.validation(request.POST, operation)

    def validation(self, request_dict, operation):
        response = ResponseObject()
        if operation == 'logout-push':  # 强制退出推送
            return self.logout_push(request_dict, response)
        elif operation == 'del-device-push':  # 删除设备推送
            return self.del_device_push(request_dict, response)

    @staticmethod
    def logout_push(request_dict, response):
        LOGGER.info('进入登出推送接口，参数：{}'.format(request_dict))
        push_token = request_dict.get('push_token', None)
        user_id = request_dict.get('user_id', None)
        app_bundle_id = request_dict.get('app_bundle_id', None)
        if not all([push_token, user_id, app_bundle_id]):
            return response.json(444)
        try:
            app_bundle_list = APP_MAPPING.get(app_bundle_id, None)
            uid_push_qs = GatewayPush.objects.filter(user_id=user_id, app_bundle_id__in=app_bundle_list,
                                                     logout=False).exclude(token_val=push_token).values('token_val',
                                                                                                        'push_type',
                                                                                                        'app_bundle_id',
                                                                                                        'jg_token_val').distinct().order_by(
                'token_val')
            if not uid_push_qs.exists():
                return response.json(173)
            uid_push_list = list(uid_push_qs)
            LOGGER.info('{}推送列表:{}'.format(user_id, uid_push_list))
            push_thread = threading.Thread(target=TransparentTransmissionPushView.thread_push,
                                           args=(uid_push_list, user_id, '强制登出', '强制登出', 705))
            push_thread.start()
            # 删除推送信息
            UidPushModel.objects.filter(userID=user_id, appBundleId__in=app_bundle_list).exclude(
                token_val=push_token).delete()
            GatewayPush.objects.filter(user_id=user_id, app_bundle_id__in=app_bundle_list).exclude(
                token_val=push_token).update(logout=True)
            return response.json(0)
        except Exception as e:
            return HttpResponse(repr(e), status=500)

    @staticmethod
    def del_device_push(request_dict, response):
        uid = request_dict.get('uid', None)
        if not all([uid]):
            return response.json(444)
        try:
            uid_push_qs = UidPushModel.objects.filter(uid_set_uid=uid).values(
                'token_val', 'appBundleId', 'push_type', 'jg_token_val')
            if not uid_push_qs.exists():
                return response.json(173)
            push_thread = threading.Thread(target=TransparentTransmissionPushView.thread_push,
                                           args=(uid_push_qs, uid, '删除设备', '删除设备'))
            push_thread.start()
            return response.json(0)
        except Exception as e:
            return HttpResponse(repr(e), status=500)

    @staticmethod
    def thread_push(push_list, user_id, title, content, event_type):
        try:
            now_time = int(time.time())
            for push_item in push_list:
                kwargs = {
                    'nickname': user_id,
                    'app_bundle_id': push_item['app_bundle_id'],
                    'token_val': push_item['token_val'],
                    'n_time': now_time,
                    'event_type': event_type,
                    'msg_title': title,
                    'msg_text': content,
                }
                LOGGER.info('进入登出推送线程，参数：{}'.format(kwargs))
                if push_item['push_type'] == 0:  # 苹果
                    PushObject.ios_apns_push(**kwargs)
                elif push_item['push_type'] == 1:  # 谷歌
                    PushObject.android_fcm_push_v1(**kwargs)
                elif push_item['push_type'] == 2:  # 极光
                    PushObject.jpush_transparent_transmission(title, content, push_item['app_bundle_id'],
                                                              push_item['token_val'], kwargs)

                elif push_item['push_type'] == 3:  # 华为
                    huawei_push_object = HuaweiPushObject()
                    huawei_push_object.huawei_transparent_transmission(user_id=user_id, **kwargs)
                elif push_item['push_type'] == 4:  # 小米
                    channel_id = XM_PUSH_CHANNEL_ID['single_login']
                    PushObject.android_xmpush(channel_id=channel_id, **kwargs)
                elif push_item['push_type'] in [5, 6]:  # vivo, oppo
                    kwargs['token_val'] = push_item['jg_token_val']
                    PushObject.jpush_transparent_transmission(title, content, push_item['app_bundle_id'],
                                                              push_item['jg_token_val'], kwargs)

                elif push_item['push_type'] == 7:  # 魅族
                    PushObject.android_meizupush(**kwargs)
                elif push_item['push_type'] == 8:  # 荣耀
                    TransparentTransmissionPushView.honor_transparent_transmission(**kwargs)
        except Exception as e:
            ERROR_INFO_LOGGER.info(
                'TransparentTransmissionPushView推送线程异常：errLine:{}, errMsg:{}, 参数:{}'.format(
                    e.__traceback__.tb_lineno, repr(e), push_list))

    @staticmethod
    def honor_transparent_transmission(token_val, n_time, event_type, msg_title, msg_text, app_bundle_id, nickname=''):
        """
        android honor 推送
        @param nickname: 设备昵称
        @param app_bundle_id: app包id
        @param token_val: 推送token
        @param n_time: 当前时间
        @param event_type: 事件类型
        @param msg_title: 推送标题
        @param msg_text: 推送内容
        @return: bool
        """
        try:
            client_id = HONORPUSH_CONFIG[app_bundle_id]['client_id']
            client_secret = HONORPUSH_CONFIG[app_bundle_id]['client_secret']
            app_id = HONORPUSH_CONFIG[app_bundle_id]['app_id']
            get_access_token_url = 'https://iam.developer.hihonor.com/auth/token'
            post_data = {
                'grant_type': 'client_credentials',
                'client_id': client_id,
                'client_secret': client_secret
            }
            headers = {'Content-Type': 'application/x-www-form-urlencoded'}
            access_token_response = requests.post(get_access_token_url, data=post_data, headers=headers)
            access_result = access_token_response.json()
            authorization_token = 'Bearer ' + access_result['access_token']
            # 发送推送
            push_url = 'https://push-api.cloud.hihonor.com/api/v1/{}/sendMessage'.format(app_id)
            headers = {'Content-Type': 'application/json', 'Authorization': authorization_token,
                       'timestamp': str(int(time.time()) * 1000)}
            extra_data = {'alert': msg_text, 'msg': '', 'sound': 'sound.aif', 'zpush': '1',
                          'received_at': n_time, 'event_time': n_time, 'event_type': str(event_type),
                          'nickname': nickname, 'title': msg_title, 'body': msg_text}
            # 透传推送
            push_data = {
                "data": json.dumps(extra_data),
                "token": [token_val]
            }
            response = requests.post(push_url, json=push_data, headers=headers)
            LOGGER.info("用户:{},时间:{}，荣耀透传推送返回值:{}".format(nickname, n_time, response.json()))
            return True
        except Exception as e:
            LOGGER.info("荣耀推送异常:error_line:{},error_msg:{}".format(e.__traceback__.tb_lineno, repr(e)))
            return False