ASJServer/Controller/SensorGateway/SubDeviceController.py

# -*- coding: utf-8 -*-
"""
@Author ： Rocky
@Time ： 2022/5/25 15:17
@File ：SubDeviceController.py
"""
import time
from collections import OrderedDict

from django.db import transaction
from django.db.models import Count, Q
from django.views import View

from Ansjer.Config.gatewaySensorConfig import SMART_SCENE_TOPIC, SENSOR_TYPE, EVENT_TYPE
from Model.models import Device_Info, GatewaySubDevice, FamilyRoomDevice, SensorRecord, SmartScene, SceneLog, FamilyRoom
from Service.CommonService import CommonService


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

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

    def validation(self, request_dict, request, operation):
        token_code, user_id, response = CommonService.verify_token_get_user_id(request_dict, request)
        if token_code != 0:
            return response.json(token_code)
        if operation == 'add':  # 添加子设备
            return self.add_sub_device(request_dict, user_id, response)
        elif operation == 'detail':  # 查询子设备信息
            return self.query_sub_device(request_dict, response)
        elif operation == 'update':  # 更新子设备信息
            return self.sensor_update(request_dict, response)
        elif operation == 'delete':  # 删除子设备
            return self.delete_sub_device(request_dict, response)
        elif operation == 'records/tem-hum':  # 查询温湿度传感器记录
            return self.records_tem_hum(request_dict, response)
        elif operation == 'records':  # 查询其他传感器记录
            return self.records(request_dict, response)
        elif operation == 'records-date':  # 查询传感器记录日期
            return self.records_date(request_dict, response)
        elif operation == 'home':  # 查询传感器主页信息
            return self.sensor_home_info(request_dict, response)
        elif operation == 'update-emergency-status':  # 更新智能按钮紧急开关状态
            return self.update_emergency_status(request_dict, response)
        elif operation == 'list':  # 获取子设备信息
            return self.get_sensor_list(request_dict, response)
        else:
            return response.json(414)

    @staticmethod
    def add_sub_device(request_dict, user_id, response):
        """
        添加子设备
        @param request_dict: 请求参数
        @request_dict serialNumber: 序列号
        @request_dict deviceType: 设备类型
        @request_dict nickName: 设备名
        @request_dict ieeeAddr: 长地址
        @request_dict srcAddr: 短地址
        @request_dict mac: mac地址
        @request_dict deviceModel: 设备型号
        @request_dict manufacturer: 制造商
        @request_dict sensorSerial: 传感器序列号
        @request_dict firmwareVersion: 固件版本
        @request_dict hardwareVersion: 硬件版本
        @request_dict familyId: 家庭id
        @request_dict roomId: 房间id
        @param user_id: 用户id
        @param response: 响应对象
        @return: response 响应对象
        """
        serial_number = request_dict.get('serialNumber', None)
        device_type = int(request_dict.get('deviceType', None))
        nickname = request_dict.get('nickName', None)
        ieee_addr = request_dict.get('ieeeAddr', None)
        src_addr = request_dict.get('srcAddr', None)
        mac = request_dict.get('mac', '')
        device_model = request_dict.get('deviceModel', '')
        manufacturer = request_dict.get('manufacturer', '')
        sensor_serial = request_dict.get('sensorSerial', '')
        firmware_version = request_dict.get('firmwareVersion', '')
        hardware_version = request_dict.get('hardwareVersion', '')

        family_id = request_dict.get('familyId', None)
        room_id = request_dict.get('roomId', 0)

        if not all([serial_number, device_type, nickname, src_addr, family_id]):
            return response.json(444)
        now_time = int(time.time())
        is_tampered = 1 if device_type == SENSOR_TYPE['smart_button'] else 0  # 智能按钮默认开启紧急开关
        try:
            # 查询网关设备数据
            device_info_qs = Device_Info.objects.filter(userID_id=user_id, serial_number=serial_number).values('id')
            if not device_info_qs.exists():
                return response.json(14)
            device_id = device_info_qs[0]['id']
            # 查询是否已添加过该子设备
            sub_device_qs = GatewaySubDevice.objects.filter(device_id=device_id, ieee_addr=ieee_addr)
            if sub_device_qs.exists():
                sub_device_qs.update(src_addr=src_addr)  # 更新短地址
                return response.json(174)
            if device_type == SENSOR_TYPE['smart_button']:
                sub_device_qs = GatewaySubDevice.objects.filter(device_id=device_id, device_type=device_type)
                if sub_device_qs.exists():
                    return response.json(181)

            with transaction.atomic():
                sub_device = GatewaySubDevice.objects.create(device_id=device_id, device_type=device_type,
                                                             nickname=nickname, ieee_addr=ieee_addr, src_addr=src_addr,
                                                             status=1, mac=mac, device_model=device_model,
                                                             manufacturer=manufacturer, sensor_serial=sensor_serial,
                                                             firmware_version=firmware_version, is_tampered=is_tampered,
                                                             hardware_version=hardware_version,
                                                             created_time=now_time, updated_time=now_time)
                if device_type == 216:
                    sub_device.is_tampered = 1
                    sub_device.save()
                family_room_device_qs = FamilyRoomDevice.objects.filter(family_id=family_id, category=0).values(
                    'category_sort')
                if family_room_device_qs.exists():
                    FamilyRoomDevice.objects.create(family_id=family_id, room_id=room_id, device_id=device_id,
                                                    sub_device=sub_device.id, created_time=now_time,
                                                    updated_time=now_time,
                                                    category_sort=family_room_device_qs[0]['category_sort'])
                else:
                    FamilyRoomDevice.objects.create(family_id=family_id, room_id=room_id, device_id=device_id,
                                                    sub_device=sub_device.id, created_time=now_time,
                                                    updated_time=now_time)

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

    @staticmethod
    def query_sub_device(request_dict, response):
        """
        查询子设备信息
        @param request_dict: 请求参数
        @request_dict gatewaySubId: 子设备id
        @param response: 响应对象
        @return: response
        """
        sub_device_id = request_dict.get('gatewaySubId', None)

        if not all([sub_device_id]):
            return response.json(444)
        try:
            gateway_sub_device_qs = GatewaySubDevice.objects.filter(id=sub_device_id).values('manufacturer',
                                                                                             'device_model',
                                                                                             'mac', 'sensor_serial',
                                                                                             'device_type',
                                                                                             'is_tampered')
            if not gateway_sub_device_qs.exists():
                return response.json(173)
            res = {
                'manufacturer': gateway_sub_device_qs[0]['manufacturer'],
                'device_model': gateway_sub_device_qs[0]['device_model'],
                'mac': gateway_sub_device_qs[0]['mac'],
                'sensor_serial': gateway_sub_device_qs[0]['sensor_serial'],
            }
            if gateway_sub_device_qs[0]['device_type'] == SENSOR_TYPE['smart_button']:  # 智能按钮返回紧急开关状态
                res['emergency_status'] = gateway_sub_device_qs[0]['is_tampered']
            return response.json(0, res)
        except Exception as e:
            return response.json(500, repr(e))

    @classmethod
    def sensor_update(cls, request_dict, response):
        """
        传感器修改
        @param request_dict: 请求参数
        @request_dict gateway_sub_id: 子设备id
        @request_dict nickName: 设备名
        @request_dict room_id: 房间id
        @param response: 响应结果
        @return: list
        """
        gateway_sub_id = int(request_dict.get('gatewaySubId', None))
        nickName = request_dict.get('nickName', None)
        room_id = request_dict.get('roomId', None)
        if not all([gateway_sub_id]):
            return response.json(444)
        try:
            gateway_sub_qs = GatewaySubDevice.objects.filter(id=gateway_sub_id)
            if not gateway_sub_qs.exists():
                return response.json(173)
            if nickName:
                gateway_sub_qs.update(nickname=nickName)
            if room_id:
                FamilyRoomDevice.objects.filter(sub_device=gateway_sub_id).update(room_id=room_id)
            return response.json(0)
        except Exception as e:
            print(e.args)
            return response.json(500, repr(e))

    @staticmethod
    def delete_sub_device(request_dict, response):
        """
        更新子设备信息
        @param request_dict: 请求参数
        @request_dict sub_device_id: 子设备id
        @param response: 响应对象
        @return: response
        """
        sub_device_id = request_dict.get('subDeviceId', None)

        if not all([sub_device_id]):
            return response.json(444)
        try:
            with transaction.atomic():
                GatewaySubDevice.objects.filter(id=sub_device_id).delete()
                smart_scene_qs = SmartScene.objects.filter(sub_device_id=sub_device_id)
                if smart_scene_qs.exists():
                    smart_scene_qs.delete()
                scene_log_qs = SceneLog.objects.filter(sub_device_id=sub_device_id)
                if scene_log_qs.exists():
                    scene_log_qs.delete()
            return response.json(0)
        except Exception as e:
            return response.json(500, repr(e))

    @classmethod
    def records_tem_hum(cls, request_dict, response):
        """
        查询温湿度传感器记录
        @param request_dict: 请求参数
        @request_dict gatewaySubId: 子设备id
        @request_dict cycle: 时间周期
        @request_dict eventType: 事件类型, 2200:温度,2201:湿度
        @param response: 响应对象
        @return: response
        """
        sub_device_id = request_dict.get('gatewaySubId', None)
        cycle = request_dict.get('cycle', None)
        event_type = request_dict.get('eventType', None)
        if not all([sub_device_id, cycle, event_type]):
            return response.json(444, {'error param': 'gatewaySubId or cycle or eventType'})

        now_time = int(time.time())
        # 判断event_type
        event_type = int(event_type)
        if event_type != EVENT_TYPE['temperature'] and event_type != EVENT_TYPE['humidity']:
            return response.json(444, {'invalid eventType': event_type})

        try:
            record_dict = cls.get_record_dict(cycle, now_time, sub_device_id, event_type)
            record_list = []
            if record_dict:
                # 组织响应数据列表,value为每 小时/天 的平均值
                for k, v in record_dict.items():
                    record_list.append({'key': k, 'value': round(sum(v) / len(v), 1)})

            res = {
                'records': record_list,
                'time': now_time
            }
            return response.json(0, res)
        except Exception as e:
            return response.json(500, repr(e))

    @staticmethod
    def get_record_dict(cycle, now_time, sub_device_id, event_type):
        """
        获取记录数据
        @param cycle: 时间周期
        @param now_time: 当前时间
        @param sub_device_id: 子设备id
        @param event_type: 事件类型, 2200:温度,2201:湿度
        @return: record_dict: 记录数据
        """
        record_dict = OrderedDict()
        if cycle == 'Hours':
            start_time = now_time - 24 * 60 * 60
            sensor_record_qs = SensorRecord.objects.filter(gateway_sub_device_id=sub_device_id,
                                                           event_type=event_type,
                                                           created_time__range=(start_time, now_time)). \
                values('alarm', 'created_time').order_by('created_time')
            if not sensor_record_qs.exists():
                return record_dict

            for sensor_record in sensor_record_qs:
                created_time = time.strftime('%m/%d %H:%M %w', time.localtime(sensor_record['created_time']))
                hour = int(created_time[-7:-5])
                minute = int(created_time[-4:-2])
                if hour != 23 and minute > 30:  # 不为23时且分钟大于30,hour+1
                    hour += 1
                alarm = float(sensor_record['alarm'])
                # 组织数据,record_dict:{"0": [1.0, 2.0, 3.0], "1": [1.0, 2.0, 3.0]...}
                if str(hour) in record_dict:
                    record_dict[str(hour)].append(alarm)
                else:
                    record_dict[str(hour)] = [alarm]

        elif cycle == 'Week':
            start_time = now_time - 24 * 60 * 60 * 7
            sensor_record_qs = SensorRecord.objects.filter(gateway_sub_device_id=sub_device_id,
                                                           event_type=event_type,
                                                           created_time__range=(start_time, now_time)). \
                values('alarm', 'created_time').order_by('created_time')
            if not sensor_record_qs.exists():
                return record_dict

            for sensor_record in sensor_record_qs:
                created_time = time.strftime('%m/%d %H:%M %w', time.localtime(sensor_record['created_time']))
                week = int(created_time[-1:])
                alarm = float(sensor_record['alarm'])
                # 组织数据,record_dict:{"0": [1.0, 2.0, 3.0], "1": [1.0, 2.0, 3.0]...}
                if str(week) in record_dict:
                    record_dict[str(week)].append(alarm)
                else:
                    record_dict[str(week)] = [alarm]

        elif cycle == 'Month':
            start_time = now_time - 24 * 60 * 60 * 30
            sensor_record_qs = SensorRecord.objects.filter(gateway_sub_device_id=sub_device_id,
                                                           event_type=event_type,
                                                           created_time__range=(start_time, now_time)). \
                values('alarm', 'created_time').order_by('created_time')
            if not sensor_record_qs.exists():
                return record_dict

            for sensor_record in sensor_record_qs:
                created_time = time.strftime('%m/%d %H:%M %w', time.localtime(sensor_record['created_time']))
                month = int(created_time[:2])
                day = int(created_time[3:5])
                date = str(month) + '/' + str(day)
                alarm = float(sensor_record['alarm'])
                # 组织数据,record_dict:{"1/1": [1.0, 2.0, 3.0], "1/2": [1.0, 2.0, 3.0]...}
                if date in record_dict:
                    record_dict[date].append(alarm)
                else:
                    record_dict[date] = [alarm]

        return record_dict

    @staticmethod
    def records(request_dict, response):
        """
        查询其他传感器记录
        @param request_dict: 请求参数
        @request_dict gatewaySubId: 子设备id
        @request_dict page: 页数
        @request_dict size: 条数
        @request_dict startTime: 开始时间
        @request_dict endTime: 结束时间
        @param response: 响应对象
        @return: response
        """
        sub_device_id = request_dict.get('gatewaySubId', None)
        page = request_dict.get('page', None)
        size = request_dict.get('size', None)
        if not all([sub_device_id, page, size]):
            return response.json(444, {'error param': 'gatewaySubId or page or size'})

        start_time = request_dict.get('startTime', None)
        end_time = request_dict.get('endTime', None)

        try:
            page, size = int(page), int(size)
            if start_time and end_time:
                sensor_record_qs = SensorRecord.objects.filter(gateway_sub_device_id=sub_device_id,
                                                               created_time__range=(start_time, end_time)). \
                                       values('alarm', 'event_type', 'created_time').order_by('-created_time')[
                                   (page - 1) * size:page * size]
            else:
                sensor_record_qs = SensorRecord.objects.filter(gateway_sub_device_id=sub_device_id). \
                                       values('alarm', 'event_type', 'created_time').order_by('-created_time')[
                                   (page - 1) * size:page * size]
            if not sensor_record_qs.exists():
                return response.json(0, [])
            return response.json(0, list(sensor_record_qs))
        except Exception as e:
            return response.json(500, repr(e))

    @staticmethod
    def records_date(request_dict, response):
        """
        查询传感器记录日期
        @param request_dict: 请求参数
        @request_dict gatewaySubId: 子设备id
        @request_dict startTime: 开始时间
        @request_dict endTime: 结束时间
        @param response: 响应对象
        @return: response
        """
        sub_device_id = request_dict.get('gatewaySubId', None)
        if not sub_device_id:
            return response.json(444, {'error param': 'gatewaySubId'})

        try:
            sensor_record_qs = SensorRecord.objects.extra(
                select={'date': "FROM_UNIXTIME(created_time,'%%Y-%%m-%%d')"}).values('date'). \
                                   filter(gateway_sub_device_id=sub_device_id). \
                                   annotate(count=Count('created_time')). \
                                   order_by('-date')[:31]
            record_date_list = []
            for sensor_record in sensor_record_qs:
                record_date_list.append({
                    'timestamp': CommonService.str_to_timestamp(sensor_record['date'], '%Y-%m-%d'),
                    'count': sensor_record['count'],
                    'format': sensor_record['date'],
                })
            return response.json(0, record_date_list)
        except Exception as e:
            return response.json(500, repr(e))

    @staticmethod
    def sensor_home_info(request_dict, response):
        """
        查询子设备首页信息
        @param request_dict: 请求参数
        @request_dict gatewaySubId: 子设备id
        @param response: 响应对象
        @return: response
        """
        sub_device_id = request_dict.get('gatewaySubId', None)
        if not sub_device_id:
            return response.json(444, {'error param': 'gatewaySubId'})

        try:
            sub_device_qs = GatewaySubDevice.objects.filter(id=sub_device_id).values('device_type', 'status',
                                                                                     'is_tampered')
            if not sub_device_qs.exists():
                return response.json(173)
            scene_count = SmartScene.objects.filter(sub_device_id=sub_device_id).count()
            res = {
                'scene_count': scene_count,
                'status': sub_device_qs[0]['status']
            }

            device_type = sub_device_qs[0]['device_type']
            # 门磁,烟雾,人体传感器返回拆动状态
            if device_type == SENSOR_TYPE['door_magnet'] or device_type == SENSOR_TYPE['smoke_sensor'] or \
                    device_type == SENSOR_TYPE['body_sensor']:
                res['is_tampered'] = sub_device_qs[0]['is_tampered']
            # 温湿度传感器返回温湿度数据
            elif device_type == SENSOR_TYPE['tem_hum_sensor']:
                tem_record_qs = SensorRecord.objects.filter(gateway_sub_device_id=sub_device_id,
                                                            event_type=EVENT_TYPE['temperature']).order_by(
                    '-created_time').values('alarm')[:1]
                hum_record_qs = SensorRecord.objects.filter(gateway_sub_device_id=sub_device_id,
                                                            event_type=EVENT_TYPE['humidity']).order_by(
                    '-created_time').values('alarm')[:1]
                temperature = tem_record_qs[0]['alarm'] if tem_record_qs.exists() else ''
                humidity = hum_record_qs[0]['alarm'] if tem_record_qs.exists() else ''
                res['temperature'] = temperature
                res['humidity'] = humidity

            return response.json(0, res)
        except Exception as e:
            return response.json(500, repr(e))

    @staticmethod
    def update_emergency_status(request_dict, response):
        """
        更新智能按钮紧急开关状态
        @param request_dict: 请求参数
        @request_dict gatewaySubId: 子设备id
        @request_dict emergencyStatus: 紧急开关状态,0:关,1:开
        @param response: 响应对象
        @return: response
        """
        sub_device_id = request_dict.get('gatewaySubId', None)
        emergency_status = request_dict.get('emergencyStatus', None)
        if not all([sub_device_id, emergency_status]):
            return response.json(444, {'error param': 'gatewaySubId or emergency_status'})

        try:
            emergency_status = int(emergency_status)
            sub_device_qs = GatewaySubDevice.objects.filter(id=sub_device_id)
            if not sub_device_qs.exists():
                response.json(173)
            with transaction.atomic():
                # 更新智能按钮紧急开关状态
                sub_device_qs.update(is_tampered=emergency_status)
                # 获取序列号
                sub_device_qs = sub_device_qs.values('device_id', 'src_addr', 'device__serial_number')
                serial_number = sub_device_qs[0]['device__serial_number']
                topic_name = SMART_SCENE_TOPIC.format(serial_number)
                gateway_sub_device_qs = GatewaySubDevice.objects.filter(device_id=sub_device_qs[0]['device_id'],
                                                                        device_type=SENSOR_TYPE['smart_button']).values(
                    'src_addr')
                sos_count = gateway_sub_device_qs.count()
                sos_num = 1
                for index, gateway_sub_device in enumerate(gateway_sub_device_qs):
                    if sub_device_qs[0]['src_addr'] == gateway_sub_device['src_addr']:
                        sos_num += index
                # 通知设备修改sos状态
                msg = {
                    'sos_count': sos_count,  # 该网关下的智能按钮数量
                    'sos_num': sos_num,  # 第几个按钮
                    'sensor_src': int(sub_device_qs[0]['src_addr'], 16),
                    'sos_select': emergency_status
                }
                success = CommonService.req_publish_mqtt_msg(serial_number, topic_name, msg, 0)
                try:
                    assert success
                except AssertionError:
                    return response.json(10044)
                time.sleep(0.3)
                # 如果打开紧急开关,关闭已开启的场景
                if emergency_status == 1:
                    smart_scene_qs = SmartScene.objects.filter(sub_device_id=sub_device_id, is_enable=True)
                    if not smart_scene_qs.exists():
                        return response.json(0)
                    for smart_scene in smart_scene_qs:
                        # 通知设备关闭场景
                        msg = {
                            'scene_id': smart_scene.id,
                            'scene_status': 0
                        }
                        success = CommonService.req_publish_mqtt_msg(serial_number, topic_name, msg, 0)
                        try:
                            assert success
                        except AssertionError:
                            return response.json(10044)
                        time.sleep(0.3)
                    smart_scene_qs.update(is_enable=False)

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

    @staticmethod
    def get_sensor_list(request_dict, response):
        """
        获取子设备信息
        @param request_dict: 请求参数
        @request_dict familyId: 家庭id
        @param response: 响应对象
        @return: response
        """
        family_id = request_dict.get('familyId', None)
        room_id = request_dict.get('roomId', None)
        if not family_id:
            return response.json(444, {'error param': 'familyId'})

        try:
            family_room_device_qs = FamilyRoomDevice.objects.filter(Q(family_id=family_id), ~Q(sub_device=0)).values(
                'sub_device', 'room_id').order_by('sort', '-device__data_joined')
            if room_id:
                family_room_device_qs = family_room_device_qs.filter(room_id=room_id)
            device_list = []
            for item in family_room_device_qs:
                room_qs = FamilyRoom.objects.filter(id=item['room_id']).values('name')
                room_name = room_qs[0]['name'] if room_qs.exists() else ''
                device_qs = GatewaySubDevice.objects.filter(id=item['sub_device']).values('id', 'nickname',
                                                                                          'device_type', 'src_addr',
                                                                                          'status', 'created_time')
                if device_qs.exists():
                    res = {
                        'gatewaySubId': device_qs[0]['id'],
                        'nickName': device_qs[0]['nickname'],
                        'deviceType': device_qs[0]['device_type'],
                        'srcAddr': device_qs[0]['src_addr'],
                        'status': device_qs[0]['status'],
                        'createdTime': device_qs[0]['created_time'],
                        'roomName': room_name,
                        'familyId': family_id
                    }
                    device_list.append(res)
            return response.json(0, device_list)
        except Exception as e:
            return response.json(500, repr(e))

#
#                   ___====-_  _-====___
#             _--^^^#####//      \\#####^^^--_
#          _-^##########// (    ) \\##########^-_
#         -############//  |\^^/|  \\############-
#       _/############//   (@::@)   \\############\_
#      /#############((     \\//     ))#############\
#     -###############\\    (oo)    //###############-
#    -#################\\  / VV \  //#################-
#   -###################\\/      \//###################-
#  _#/|##########/\######(   /\   )######/\##########|\#_
#  |/ |#/\#/\#/\/  \#/\##\  |  |  /##/\#/  \/\#/\#/\#| \|
#  `  |/  V  V  `   V  \#\| |  | |/#/  V   '  V  V  \|  '
#     `   `  `      `   / | |  | | \   '      '  '   '
#                      (  | |  | |  )
#                     __\ | |  | | /__
#                    (vvv(VVV)(VVV)vvv)
#                         神兽保佑
#                        代码无BUG!
#