ASJPush/Controller/DetectController.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
@Copyright (C) ansjer cop Video Technology Co.,Ltd.All rights reserved.
@AUTHOR: ASJRD018
@NAME: AnsjerFormal
@software: PyCharm
@DATE: 2019/1/14 15:57
@Version: python3.6
@MODIFY DECORD:ansjer dev
@file: DetectController.py
@Contact: chanjunkai@163.com
"""
import json
import os
import time
import urllib

import apns2
import jpush as jpush
import oss2
import requests
from django.utils.decorators import method_decorator
from django.views.decorators.csrf import csrf_exempt
from django.views.generic.base import View
from AnsjerPush.config import OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET, SERVER_DOMAIN, BASE_DIR, DETECT_PUSH_DOMAIN
from Model.models import Equipment_Info, UidSetModel, UidPushModel
from Object.ResponseObject import ResponseObject
from Object.UidTokenObject import UidTokenObject
from django.http import JsonResponse
from Object.RedisObject import RedisObject
from django.db import transaction


# http://192.168.136.40:8077/notify/push?uidToken=eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ1aWQiOiJUTjdNUEUzMjExVUU3NkFQMTExQSJ9.k501567VdnhFpn_ygzGRDat3Kqlz5CsEA9jAC2dDk_g&obj=12341234&n_time=1234561234
# http://test.dvema.com/notify/push?uidToken=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1aWQiOiJQMldOR0pSRDJFSEE1RVU5MTExQSJ9.xOCI5lerk8JOs5OcAzunrKCfCrtuPIZ3AnkMmnd-bPY&n_time=1526845794&channel=1&event_type=51&is_st=0
# 移动侦测接口
class NotificationView(View):

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

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

    def validation(self, request_dict):
        response = ResponseObject()
        uidToken = request_dict.get('uidToken', None)
        channel = request_dict.get('channel', None)
        n_time = request_dict.get('n_time', None)
        event_type = request_dict.get('event_type', None)
        is_st = request_dict.get('is_st', None)
        if not all([uidToken, channel, n_time]):
            return JsonResponse(status=200, data={
                'code': 444,
                'msg': 'param is wrong'})
            # return response.json(444)
        utko = UidTokenObject(uidToken)
        uid = utko.UID
        uid_set_qs = UidSetModel.objects.filter(uid=uid, detect_status=1)
        if uid_set_qs.exists():
            uid_set_id = uid_set_qs[0].id
            uid_push_qs = UidPushModel.objects.filter(uid_set__id=uid_set_id). \
                values('token_val', 'app_type', 'appBundleId', 'push_type', 'userID_id', 'userID__NickName')
            if uid_set_qs.exists():
                redisObj = RedisObject(db=6)
                pkey = '{uid}_{channel}_ptl'.format(uid=uid, channel=channel)
                if redisObj.get_data(key=pkey):
                    res_data = {'code': 0, 'msg': 'success,!'}
                    return JsonResponse(status=200, data=res_data)
                else:
                    detect_interval = uid_set_qs[0].detect_interval
                    if detect_interval:
                        redisObj.set_data(key=pkey, val=1, expire=detect_interval)
                auth = oss2.Auth(OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET)
                bucket = oss2.Bucket(auth, 'oss-cn-shenzhen.aliyuncs.com', 'apg')
                for up in uid_push_qs:
                    push_type = up['push_type']
                    # ios apns
                    if push_type == 0:
                        self.do_apns(request_dict, up, response, uid)
                    # android gcm
                    elif push_type == 1:
                        self.do_gmc(request_dict, up, response, uid)
                    # android jpush
                    elif push_type == 2:
                        self.do_jpush(request_dict, up, response, uid)
                    # self.do_save_equipment_info(ua, n_time, channel, event_type, is_st)
                # 需求不一样，所以这么做的
                self.do_bulk_create_info(uid_push_qs, n_time, channel, event_type, is_st, uid)
                if is_st == '0' or is_st == '2':
                    return JsonResponse(status=200, data={'code': 0, 'msg': 'success'})
                else:
                    # Endpoint以杭州为例，其它Region请按实际情况填写。
                    obj = '{uid}/{channel}/{filename}.jpeg'.format(uid=uid, channel=channel, filename=n_time)
                    # 设置此签名URL在60秒内有效。
                    url = bucket.sign_url('PUT', obj, 7200)
                    res_data = {'code': 0, 'img_push': url, 'msg': 'success'}
                    return JsonResponse(status=200, data=res_data)
            else:
                return JsonResponse(status=200, data={'code': 404, 'msg': 'data is not exist'})
        else:
            return JsonResponse(status=200, data={'code': 404, 'msg': 'data is not exist'})

    def do_jpush(self, request_dict, uaql, response, uid):
        event_type = request_dict.get('event_type', None)
        jpush_config = {
            'com.ansjer.accloud_ab': {
                'Key': 'f0dc047e5e53fd14199de5b0',
                'Secret': 'aa7f7db33e9f0a7f3871aa1c'},
            'com.ansjer.adcloud_ab': {
                'Key': '76d97b535185114985608234',
                'Secret': 'c9a92b301043cc9c52778692'},
            'com.ansjer.zccloud_ab': {
                'Key': 'd9924f56d3cc7c6017965130',
                'Secret': '869d832d126a232f158b5987'},
            'com.ansjer.loocamccloud_ab': {
                'Key': 'd1cc44797b4642b0e05304fe',
                'Secret': 'c3e8b4ca8c576de61401e56a'},
            'com.ansjer.loocamdcloud_ab': {
                'Key': '76d97b535185114985608234',
                'Secret': 'c9a92b301043cc9c52778692'},
            'com.ansjer.zccloud_a': {
                'Key': '57de2a80d68bf270fd6bdf5a',
                'Secret': '3d354eb6a0b49c2610decf42'},
            'com.ansjer.accloud_a': {
                'Key': 'ff95ee685f49c0dc4013347b',
                'Secret': 'de2c20959f5516fdeeafe78e'},
            'com.ansjer.adcloud_a': {
                'Key': '2e47eb1aee9b164460df3668',
                'Secret': 'b9137d8d684bc248f1809b6d'},
            'com.ansjer.loocamccloud_a': {
                'Key': '23c9213215c7ca0ec945629b',
                'Secret': '81e4b1e859cc8387e2e6c431'},
            'com.ansjer.loocamdcloud_a': {
                'Key': '1dbdd60a16e9892d6f68a073',
                'Secret': '80a97690e7e043109059b403'},
            'com.ansjer.customizedb_a': {
                'Key': '9d79630aa49adfa291fe2568',
                'Secret': '4d8ff52f88136561875a0212'},
        }
        n_time = request_dict.get('n_time', None)
        appBundleId = uaql['appBundleId']
        token_val = uaql['token_val']

        response = ResponseObject()
        app_key = jpush_config[appBundleId]['Key']
        master_secret = jpush_config[appBundleId]['Secret']
        # 此处换成各自的app_key和master_secre
        _jpush = jpush.JPush(app_key, master_secret)
        push = _jpush.create_push()
        # if you set the logging level to "DEBUG",it will show the debug logging.
        _jpush.set_logging("DEBUG")
        # push.audience = jpush.all_
        push.audience = jpush.registration_id(token_val)
        push_data = {"alert": "Motion ", "event_time": n_time, "event_type": event_type, "msg": "",
                     "received_at": n_time, "sound": "sound.aif", "uid": uid, "zpush": "1"}
        push.message = jpush.message('Motion', extras=push_data, title='KPNS', content_type='text')
        push.platform = jpush.all_
        try:
            res = push.send()
            print(res)
        except Exception as e:
            print("Exception")
            print(repr(e))
            return response.json(10, repr(e))
        else:
            return response.json(0)

    def do_gmc(self, request_dict, uaql, response, uid):

        n_time = request_dict.get('n_time')
        appBundleId = uaql['appBundleId']
        token_val = uaql['token_val']
        gcm_config = {
            'com.ansjer.zccloud_a': 'AAAAb9YP3rk:APA91bHu8u-CTpcd0g6lKPo0WNVqCi8jZub1cPPbSAY9AucT1HxlF65ZDUko9iG8q2ch17bwu9YWHpK1xI1gHSRXCslLvZlXEmHZC0AG3JKg15XuUvlFKACIajUFV-pOeGRT8tM6-31I',
            'com.ansjer.loocamccloud_a': 'AAAAb9YP3rk:APA91bFCgd-kbVmpK4EVpfdHH_PJZQCYTkOGnTZdIuBWEz2r7aMRsJYHOH3sB-rwcbaRWgnufTyjX9nGQxb6KxQbWVk4ah_H-M3IqGh6Mb60WQQAuR33V6g_Jes5pGL6ViuIxGHqVMaR',
            'com.ansjer.loocamdcloud_a': 'AAAAb9YP3rk:APA91bGw2I2KMD4i-5T7nZO_wB8kuAOuqgyqe5rxmY-W5qkpYEx9IL2IfmC_qf6B_xOyjIDDSjckvMo-RauN__SEoxvAkis7042GRkoKpw7cjZ_H8lC-d50PC0GclPzccrOGFusyKbFY',
            'com.ansjer.customizedb_a': 'AAAAb9YP3rk:APA91bE7kI4vcm-9h_CJNFlOZfc-xwP4Btn6AnjOrwoKV6fgYN7fdarkO76sYxVZiAbDnxsFfOJyP7vQfwyan6mdjuyD5iHdt_XgO22VqniC0vA1V4GJiCS8Tp7LxIX8JVKZl9I_Powt',
            'com.ansjer.customizeda_a': 'AAAAb9YP3rk:APA91bF0HzizVWDc6dKzobY9fsaKDK4veqkOZehDXshVXs8pEEvNWjR_YWbhP60wsRYCHCal8fWN5cECVOWNMMzDsfU88Ty2AUl8S5FtZsmeDTkoGntQOswBr8Ln7Fm_LAp1VqTf9CpM',
        }
        serverKey = gcm_config[appBundleId]
        event_type = request_dict.get('event_type', None)
        push_data = {"alert": "Motion ", "event_time": n_time, "event_type": event_type, "msg": "",
                     "received_at": n_time, "sound": "sound.aif", "uid": uid, "zpush": "1"}
        json_data = {
            "collapse_key": "WhatYouWant",
            "data": push_data,
            "delay_while_idle": False,
            "time_to_live": 3600,
            "registration_ids": [token_val]
        }

        url = 'https://android.googleapis.com/gcm/send'
        # serverKey = "AAAAb9YP3rk:APA91bHu8u-CTpcd0g6lKPo0WNVqCi8jZub1cPPbSAY9AucT1HxlF65ZDUko9iG8q2ch17bwu9YWHpK1xI1gHSRXCslLvZlXEmHZC0AG3JKg15XuUvlFKACIajUFV-pOeGRT8tM6-31I"

        data = json.dumps(json_data).encode('utf-8')
        headers = {'Content-Type': 'application/json', 'Authorization': 'key=%s' % serverKey}
        req = requests.post(url, data, headers=headers)
        return response.json(0)

    def do_apns(self, request_dict, uaql, response, uid):
        event_type = request_dict.get('event_type', None)
        token_val = uaql['token_val']
        n_time = request_dict.get('n_time')
        appBundleId = uaql['appBundleId']
        apns_config = {
            'com.ansjer.loocamccloud': {
                'pem_path': os.path.join(BASE_DIR, 'Ansjer/file/apns_pem/apns-dev-test.pem'),
                'password': '111111'
            }
        }
        try:
            cli = apns2.APNSClient(mode="dev", client_cert=apns_config[appBundleId]['pem_path'],
                                   password=apns_config[appBundleId]['password'])
            push_data = {"alert": "Motion ", "event_time": n_time, "event_type": event_type, "msg": "",
                         "received_at": n_time, "sound": "sound.aif", "uid": uid, "zpush": "1"}
            body = json.dumps(push_data)
            alert = apns2.PayloadAlert(body=body, title="title!")
            payload = apns2.Payload(alert=alert)
            n = apns2.Notification(payload=payload, priority=apns2.PRIORITY_LOW)
            res = cli.push(n=n, device_token=token_val, topic=appBundleId)
            # assert res.status_code == 200, res.reason
            # assert res.apns_id 
            if res.status_code == 200:
                return response.json(0)
            else:
                return response.json(404, res.reason)
        except Exception as e:
            return response.json(10, repr(e))

    def do_bulk_create_info(self, uaqs, n_time, channel, event_type, is_st, uid):
        #
        qs_list = []
        nowTime = int(time.time())
        # 设备昵称

        for dv in uaqs:
            add_data = {
                'userID_id': dv["userID_id"],
                'eventTime': n_time,
                'eventType': event_type,
                'devUid': uid,
                'devNickName': uid,
                'Channel': channel,
                'alarm': 'Motion \tChannel:{channel}'.format(channel=channel),
                'is_st': int(is_st),
                'receiveTime': n_time,
                'addTime': nowTime
            }
            qs_list.append(Equipment_Info(**add_data))
        if qs_list:
            print(1)
            Equipment_Info.objects.bulk_create(qs_list)
            return True
        else:
            return False