ASJServer/Controller/IotCoreController.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import hashlib
import logging
import time
import uuid
from collections import OrderedDict

import requests
from django.views import View

from Ansjer.config import AWS_IOT_GETS3_PULL_CHINA_ID, AWS_IOT_GETS3_PULL_CHINA_SECRET, \
    AWS_IOT_GETS3_PULL_FOREIGN_ID, AWS_IOT_GETS3_PULL_FOREIGN_SECRET, AWS_ARN, AWS_IOT_SES_ACCESS_CHINA_REGION, \
    AWS_IOT_SES_ACCESS_FOREIGN_REGION_ASIA, AWS_IOT_SES_ACCESS_FOREIGN_REGION_EUROPE, \
    AWS_IOT_SES_ACCESS_FOREIGN_REGION_AMERICA
from Controller.DeviceConfirmRegion import Device_Region
from Model.models import Device_Info, iotdeviceInfoModel, SerialNumberModel, UidSetModel
from Object.IOTCore.IotObject import IOTClient
from Object.ResponseObject import ResponseObject
from Object.TokenObject import TokenObject
from Service.CommonService import CommonService


class IotCoreView(View):

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

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

    def validate(self, operation, request_dict, request):
        response = ResponseObject()
        lang = request_dict.get('lang', 'en')
        response.lang = lang
        if operation == 'createKeysAndCertificate':
            return self.create_keys_and_certificate(request_dict, response, request)
        elif operation == 'requestPublishMessage':
            return self.request_publish_message(request_dict, response, request)
        elif operation == 'getS3PullKey':
            return self.get_s3_pull_key(request_dict, response, request)
        elif operation == 'thingRegroup':
            return self.thing_regroup(request_dict, response)
        elif operation == 'pcGetIotInfo':
            return self.pcGetIotInfo(request_dict, response)
        else:
            token = TokenObject(request_dict.get('token', None))
            if token.code != 0:
                return response.json(token.code)
            response.lang = token.lang

            if operation == 'clearIotCerm':
                return self.clear_Iot_Cerm(request_dict, response)
            elif operation == 'getIotInfo':
                return self.getIotInfo(request_dict, response)
            else:
                return response.json(404)

    # 设备注册到aws iot core
    @staticmethod
    def create_keys_and_certificate(request_dict, response, request):
        logger = logging.getLogger('info')
        logger.info('设备注册到aws iot core请求参数:{}'.format(request_dict))
        token = request_dict.get('token', None)
        language = request_dict.get('language', None)
        time_stamp = request_dict.get('time_stamp', None)
        device_version = request_dict.get('device_version', None).replace('.', '_')  # 物品组命名不能包含'.'

        if not all([token, language, time_stamp, device_version]):
            return response.json(444, {'param': 'token, language, time_stamp, device_version'})

        try:
            # 时间戳token校验
            no_rtc = request_dict.get('no_rtc', None)
            if no_rtc:
                if not CommonService.check_time_stamp_token_without_distance(token, time_stamp):
                    return response.json(13)
            else:
                if not CommonService.check_time_stamp_token(token, time_stamp):
                    return response.json(13)

            uid = request_dict.get('uid', '')
            uid_code = request_dict.get('uid_code', None)
            company_mark = '11A'
            if not uid:  # 传序列号
                serial_number = request_dict.get('serial_number', None)
                serial_number_code = request_dict.get('serial_number_code', None)
                if not all([serial_number, serial_number_code]):
                    return response.json(444, {'param': 'serial_number, serial_number_code'})

                # 序列号编码解码校验
                serial_number_code = CommonService.decode_data(serial_number_code)
                if serial_number != serial_number_code:
                    return response.json(404)

                serial = serial_number[0:6]
                company_mark = serial_number[-3:]
                try:
                    SerialNumberModel.objects.get(serial_number=serial)
                except:
                    return response.json(444)

                thing_name_suffix = serial_number  # 物品名后缀
                iot_device_info_qs = iotdeviceInfoModel.objects.filter(serial_number=serial)
            else:  # 传uid
                # uid编码解码校验
                uid_code = CommonService.decode_data(uid_code)
                if uid != uid_code:
                    return response.json(404)

                serial = ''  # iot_deviceInfo表写入serial_number为''
                thing_name_suffix = uid  # 物品名后缀
                iot_device_info_qs = iotdeviceInfoModel.objects.filter(uid=uid)
            # 判断设备是否已注册过
            if iot_device_info_qs.exists():
                iot = iot_device_info_qs[0]
                res = {
                    'certificateId': iot.certificate_id,
                    'certificatePem': iot.certificate_pem,
                    'publicKey': iot.public_key,
                    'privateKey': iot.private_key,
                    'endpoint': iot.endpoint
                }
                return response.json(0, {'res': res})
            else:
                # 根据地区id确定使用的boto3 client
                region_id = request_dict.get('region_id', None)
                if region_id:
                    region_id = int(region_id)
                else:
                    ip = CommonService.get_ip_address(request)
                    region_id = Device_Region().get_device_region(ip)
                iotClient = IOTClient(region_id)

                # 拼接物品名
                thingName = CommonService.get_thing_name(company_mark, thing_name_suffix)
                thingGroup = device_version + '_' + language
                res = iotClient.register_to_iot_core(thingName, thingGroup, response)

                token_iot_number = hashlib.md5((str(uuid.uuid1()) + str(int(time.time()))).encode('utf-8')).hexdigest()
                iotdeviceInfoModel.objects.create(uid=uid,
                                                  serial_number=serial,
                                                  endpoint=res[0]['endpoint'],
                                                  certificate_id=res[0]['certificateId'],
                                                  certificate_pem=res[0]['certificatePem'],
                                                  public_key=res[0]['publicKey'],
                                                  private_key=res[0]['privateKey'],
                                                  thing_name=res[1]['ThingName'],
                                                  thing_groups=res[1]['thingGroupName'],
                                                  token_iot_number=token_iot_number
                                                  )
                res = {
                    'certificateId': res[0]['certificateId'],
                    'certificatePem': res[0]['certificatePem'],
                    'publicKey': res[0]['publicKey'],
                    'privateKey': res[0]['privateKey'],
                    'endpoint': res[0]['endpoint']
                }
                return response.json(0, {'res': res})
        except Exception as e:
            print(e)
            return response.json(500, repr(e))

    @staticmethod
    def thing_regroup(request_dict, response):
        # 物品重新分组
        uid = request_dict.get('uid', '')
        token = request_dict.get('token', None)
        language = request_dict.get('language', None)
        region_id = request_dict.get('region_id', None)
        time_stamp = request_dict.get('time_stamp', None)
        device_version = request_dict.get('device_version', None)

        if not all([token, language, region_id, time_stamp, device_version]):
            return response.json(444)

        # 时间戳token校验
        if not CommonService.check_time_stamp_token(token, time_stamp):
            return response.json(13)

        company_mark = '11A'
        thing_name_suffix = uid
        if not uid:
            # 使用序列号
            serial_number = request_dict.get('serial_number', None)
            if not serial_number:
                return response.json(444)
            company_mark = serial_number[-3:]
            thing_name_suffix = serial_number
            uid = CommonService.query_uid_with_serial(serial_number)

        uid_set_qs = UidSetModel.objects.filter(uid=uid)
        thingName = CommonService.get_thing_name(company_mark, thing_name_suffix)
        new_thingGroupName = (device_version + '_' + language).replace('.', '_')  # 物品组命名不能包含'.'

        try:
            iotClient = IOTClient(int(region_id))
            # 获取旧物品组
            list_groups_res = iotClient.client.list_thing_groups_for_thing(thingName=thingName, maxResults=1)
            old_thingGroupName = list_groups_res['thingGroups'][0]['groupName']
            # 没有新物品组则创建
            list_thing_groups_res = iotClient.client.list_thing_groups(namePrefixFilter=new_thingGroupName
                                                                       , maxResults=1, recursive=False)
            if not list_thing_groups_res['thingGroups']:
                attributes = {
                    "update_time": "0"
                }
                thingGroupProperties = {
                    "thingGroupDescription": "OTA",
                    "attributePayload": {
                        "attributes": attributes,
                        "merge": False  # 更新时覆盖掉而不是合并
                    }
                }
                iotClient.client.create_thing_group(thingGroupName=new_thingGroupName
                                                    , thingGroupProperties=thingGroupProperties)
            iotClient.client.update_thing_groups_for_thing(thingName=thingName
                                                           , thingGroupsToAdd=[new_thingGroupName]
                                                           , thingGroupsToRemove=[old_thingGroupName])

            # 更新设备版本信息
            uid_set_qs.update(version=device_version)
            return response.json(0)
        except Exception as e:
            print(e)
            return response.json(500, repr(e))

    def clear_Iot_Cerm(self, request_dict, response):
        serial_number = request_dict.get('serial_number', None)

        if serial_number:

            iot = iotdeviceInfoModel.objects.filter(thing_name="Ansjer_Device_" + serial_number)
            if iot.exists():
                iot.delete()
            return response.json(0)

        else:
            return response.json(444)

    def request_publish_message(self, request_dict, response, request):
        # Alexa请求IoT Core下发MQTT消息通知设备开始或停止推流,或唤醒设备
        UID = request_dict.get('UID', None)
        rtsp = request_dict.get('rtsp', None)
        enable = request_dict.get('enable', '1')
        if not all([UID, rtsp]):
            return response.json(444)

        try:
            # 获取设备的物品名后缀
            device_info_qs = Device_Info.objects.filter(UID=UID).values('UID', 'serial_number')
            if not device_info_qs.exists():
                return response.json(10043)
            uid = device_info_qs[0]['UID']
            serial_number = device_info_qs[0]['serial_number']
            # 如果device_info表的serial_number不为空,物品名为'Ansjer_Device_序列号'
            thing_name_suffix = serial_number if serial_number != '' else uid
            # 获取数据组织将要请求的url
            iot = iotdeviceInfoModel.objects.filter(thing_name__contains=thing_name_suffix).values('thing_name',
                                                                                                   'endpoint',
                                                                                                   'token_iot_number')
            if not iot.exists():
                return response.json(10043)
            thing_name = iot[0]['thing_name'][14:]  # IoT core上的物品名: Ansjer_Device_ + 序列号+企业编码/uid
            endpoint = iot[0]['endpoint']
            Token = iot[0]['token_iot_number']
            # Token = '297a601b3925e04daab5a60280650e09'
            topic_name = 'ansjer/generic/' + thing_name  # MQTT主题名

            # api doc: https://docs.aws.amazon.com/zh_cn/iot/latest/developerguide/http.html
            # url: https://IoT_data_endpoint/topics/url_encoded_topic_name?qos=1
            # post请求url来发布MQTT消息
            url = 'https://{}/topics/{}'.format(endpoint, topic_name)
            authorizer_name = 'Ansjer_Iot_Auth'
            signature = CommonService.rsa_sign(Token)  # Token签名
            headers = {'x-amz-customauthorizer-name': authorizer_name, 'Token': Token,
                       'x-amz-customauthorizer-signature': signature}
            params = OrderedDict(
                [
                    ('alexaRtspCommand', rtsp),
                    ('enable', enable),
                ]
            )
            r = requests.post(url=url, headers=headers, json=params, timeout=2)
            if r.status_code == 200:
                res = r.json()
                if res['message'] == 'OK':
                    return response.json(0)
                return response.json(10044)
            else:
                # print('发布失败')
                return response.json(10044)
        except Exception as e:
            # print(e)
            return response.json(500, repr(e))

    def get_s3_pull_key(self, request_dict, response, request):
        # 通用发布主题通知
        UID = request_dict.get('UID', None)
        if not all([UID]):
            return response.json(444)

        try:
            # 获取检查uid的序列号,如果没有序列号,不使用MQTT下发消息
            device_info_qs = Device_Info.objects.filter(UID=UID).values('UID', 'serial_number')
            if not device_info_qs.exists():
                return response.json(10043)
            uid = device_info_qs[0]['UID']
            serial_number = device_info_qs[0]['serial_number']
            # 如果device_info表的serial_number不为空,物品名为'Ansjer_Device_序列号'
            thing_name_suffix = serial_number if serial_number != '' else uid
            # 获取数据组织将要请求的url
            iot = iotdeviceInfoModel.objects.filter(thing_name__contains=thing_name_suffix).values('thing_name',
                                                                                                   'endpoint',
                                                                                                   'token_iot_number')
            if not iot.exists():
                return response.json(10043)
            endpoint = iot[0]['endpoint']
            MSG = self.get_s3_key_return_msg(endpoint)

            return response.json(0, MSG)
        except Exception as e:
            # print(e)
            return response.json(500, repr(e))

    def get_s3_key_return_msg(self, endpoint):
        MSG = {}
        if 'cn-northwest-1' in endpoint:
            key = AWS_IOT_GETS3_PULL_CHINA_ID
            secret = AWS_IOT_GETS3_PULL_CHINA_SECRET
            arn = AWS_ARN[0]
            region_name = AWS_IOT_SES_ACCESS_CHINA_REGION
        else:
            key = AWS_IOT_GETS3_PULL_FOREIGN_ID
            secret = AWS_IOT_GETS3_PULL_FOREIGN_SECRET
            arn = AWS_ARN[1]
            if 'ap-southeast-1' in endpoint:
                region_name = AWS_IOT_SES_ACCESS_FOREIGN_REGION_ASIA
            if 'eu-west-1' in endpoint:
                region_name = AWS_IOT_SES_ACCESS_FOREIGN_REGION_EUROPE
            if 'us-east-1' in endpoint:
                region_name = AWS_IOT_SES_ACCESS_FOREIGN_REGION_AMERICA

        MSG['AccessKeyId'] = key
        MSG['AccessKeySecret'] = secret
        MSG['bucket_name'] = 'asj-log'
        MSG['arn'] = arn
        MSG['region_name'] = region_name
        return MSG

    def getIotInfo(self, request_dict, response):
        # 获取IoT数据
        serial_number = request_dict.get('serial_number', None)
        uid = request_dict.get('uid', None)
        if not uid and not serial_number:
            return response.json(444)

        try:
            if serial_number:
                serial_number = serial_number[0:6]
                iot_info_qs = iotdeviceInfoModel.objects.filter(serial_number=serial_number). \
                    values('endpoint', 'token_iot_number')
            else:
                iot_info_qs = iotdeviceInfoModel.objects.filter(uid=uid). \
                    values('endpoint', 'token_iot_number')

            if not iot_info_qs.exists():
                return response.json(173)

            endpoint = iot_info_qs[0]['endpoint']
            token_iot_number = iot_info_qs[0]['token_iot_number']
            res = {'endpoint': endpoint, 'token_iot_number': token_iot_number}
            return response.json(0, res)
        except Exception as e:
            return response.json(500, repr(e))

    def pcGetIotInfo(self, request_dict, response):
        # PC工具获取IoT数据
        serial_number = request_dict.get('serial_number', None)
        uid = request_dict.get('uid', None)
        if not uid and not serial_number:
            return response.json(444)

        try:
            if serial_number:
                serial_number = serial_number[0:6]
                iot_info_qs = iotdeviceInfoModel.objects.filter(serial_number=serial_number). \
                    values('endpoint', 'token_iot_number')
            else:
                iot_info_qs = iotdeviceInfoModel.objects.filter(uid=uid). \
                    values('endpoint', 'token_iot_number')

            if not iot_info_qs.exists():
                return response.json(173)

            endpoint = iot_info_qs[0]['endpoint']
            token_iot_number = iot_info_qs[0]['token_iot_number']
            res = {'endpoint': endpoint, 'token_iot_number': token_iot_number}
            return response.json(0, res)
        except Exception as e:
            return response.json(500, repr(e))