ASJServer/Controller/SensorGateway/SmartSceneController.py

# -*- coding: utf-8 -*-
"""
@Author ： Rocky
@Time ： 2022/6/29 9:31
@File ：SmartSceneController.py
"""
import time
import json

from django.core.exceptions import ObjectDoesNotExist
from django.db.models import F, Q
from django.views import View
from django.db import transaction

from Model.models import FamilyRoomDevice, GatewaySubDevice, FamilyRoom, SmartScene, EffectiveTime, Device_Info
from Service.CommonService import CommonService


class SmartSceneView(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 == 'condition-devices':  # 添加条件-查询设备
            return self.condition_devices(request_dict, response)
        elif operation == 'task-devices':  # 添加任务-查询设备
            return self.task_devices(request_dict, response)
        elif operation == 'create':  # 创建智能场景
            return self.create_smart_scene(request_dict, user_id, response)
        elif operation == 'scene-list':  # 查询智能场景列表
            return self.scene_list(request_dict, user_id, response)
        elif operation == 'update-status':  # 更新智能场景状态
            return self.update_status(request_dict, response)
        elif operation == 'detail':  # 查询智能场景详情
            return self.scene_detail(request_dict, response)
        elif operation == 'edit':  # 编辑智能场景
            return self.edit_smart_scene(request_dict, response)
        elif operation == 'delete':  # 删除智能场景
            return self.delete_smart_scene(request_dict, response)
        else:
            return response.json(414)

    @classmethod
    def condition_devices(cls, request_dict, response):
        """
        添加条件-查询设备
        @param request_dict: 请求参数
        @request_dict deviceId: 网关设备id
        @request_dict subDeviceId: 子设备id
        @param response: 响应对象
        @return: response
        """
        device_id = request_dict.get('deviceId', None)
        sub_device_id = request_dict.get('subDeviceId', None)
        if not any([device_id, sub_device_id]):
            return response.json(444, {'error param': 'deviceId or subDeviceId'})
        try:
            if sub_device_id:
                device_id = GatewaySubDevice.objects.get(id=sub_device_id).device_id
            gateway_sub_device_qs = GatewaySubDevice.objects.filter(device_id=device_id)
            if not gateway_sub_device_qs.exists():
                return response.json(173)
            res = cls.get_sub_device_room_name(gateway_sub_device_qs)
            return response.json(0, res)
        except Exception as e:
            return response.json(500, repr(e))

    @classmethod
    def task_devices(cls, request_dict, response):
        """
        添加任务-查询设备
        @param request_dict: 请求参数
        @request_dict deviceId: 网关设备id
        @param response: 响应对象
        @return: response
        """
        sub_device_id = request_dict.get('subDeviceId', None)
        device_id = request_dict.get('deviceId', None)

        if not any([device_id, sub_device_id]):
            return response.json(444, {'error param': 'deviceId or subDeviceId'})
        try:

            if device_id:
                res = [cls.get_gateway_data(device_id)]
            else:
                sub_device_qs = GatewaySubDevice.objects.filter(id=sub_device_id).values('device_id', 'device_type')
                device_id = sub_device_qs[0]['device_id']
                device_type = sub_device_qs[0]['device_type']

                if device_type != 216:  # 非智能按钮只返回网关
                    res = [cls.get_gateway_data(device_id)]
                else:
                    gateway_data = cls.get_gateway_data(device_id)
                    sub_device_qs = GatewaySubDevice.objects.filter(
                        Q(device_id=device_id) & Q(device_type=215) | Q(device_type=219)).values('id', 'nickname',
                                                                                                 'status',
                                                                                                 'device_type')
                    if not sub_device_qs.exists():
                        return response.json(173)
                    res = cls.get_sub_device_room_name(sub_device_qs, gateway_data)

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

    @staticmethod
    def get_sub_device_room_name(sub_device_qs, gateway_data=None):
        """
        获取房间名称
        @param sub_device_qs: 子设备信息
        @param gateway_data: 网关参数
        @return: sub_device_list
        """
        sub_device_list = []
        if gateway_data:
            sub_device_list.append(gateway_data)
        sub_device_qs = sub_device_qs.annotate(gatewaySubId=F('id'),
                                               deviceType=F('device_type'),
                                               deviceNickName=F('nickname')). \
            values('gatewaySubId', 'deviceType', 'deviceNickName', 'status')
        for sub_device in sub_device_qs:
            family_room_device_qs = FamilyRoomDevice.objects.filter(sub_device=sub_device['gatewaySubId']). \
                values('room_id')
            if not family_room_device_qs.exists():
                sub_device['roomName'] = ''
            else:
                room_id = family_room_device_qs[0]['room_id']
                try:
                    sub_device['roomName'] = FamilyRoom.objects.get(id=room_id).name
                except ObjectDoesNotExist:
                    sub_device['roomName'] = ''
            sub_device_list.append(sub_device)
        return sub_device_list

    @staticmethod
    def get_gateway_data(device_id):
        """
        获取网关数据
        @param device_id: 网关设备id
        @return: res
        """
        device_info_qs = Device_Info.objects.filter(id=device_id).values('NickName', 'Type')
        nickname = device_info_qs[0]['NickName']
        device_type = device_info_qs[0]['Type']
        room_id = FamilyRoomDevice.objects.filter(device_id=device_id).values('room_id')[0]['room_id']
        room_id_qs = FamilyRoom.objects.filter(id=room_id).values('name')
        room_name = room_id_qs.first()['name'] if room_id_qs.exists() else ''
        res = {
            'deviceNickName': nickname,
            'deviceType': device_type,
            'roomName': room_name,
            'status': 1,
        }
        return res

    @staticmethod
    def create_smart_scene(request_dict, user_id, response):
        """
        创建智能场景
        @param request_dict: 请求参数
        @param user_id: 用户id
        @request_dict deviceId: 网关设备id
        @request_dict subDeviceId: 子设备id
        @request_dict sceneName: 场景名称
        @request_dict conditions: 条件
        @request_dict tasks: 任务
        @request_dict isAllDay: 是否全天执行
        @request_dict startTime: 开始时间
        @request_dict endTime: 结束时间
        @request_dict repeat: 重复周期
        @param response: 响应对象
        @return: response
        """
        device_id = request_dict.get('deviceId', None)
        sub_device_id = request_dict.get('subDeviceId', None)
        scene_name = request_dict.get('sceneName', None)
        conditions = request_dict.get('conditions', None)
        tasks = request_dict.get('tasks', None)
        is_all_day = request_dict.get('isAllDay', None)

        if not any([device_id, sub_device_id]):
            return response.json(444, {'error param': 'deviceId or subDeviceId'})
        if not all([conditions, tasks]):
            return response.json(444, {'error param': 'conditions and tasks'})

        now_time = int(time.time())
        conditions_dict = eval(conditions)
        tasks_list = eval(tasks)
        try:
            # 判断是否已存在该场景名
            smart_scene_qs = SmartScene.objects.filter(user_id=user_id, scene_name=scene_name)
            if smart_scene_qs.exists():
                return response.json(174)

            smart_scene_dict = {
                'user_id': user_id,
                'scene_name': scene_name,
                'conditions': conditions,
                'tasks': tasks,
                'created_time': now_time,
                'updated_time': now_time,
            }

            # 处理设置时间
            if is_all_day:
                is_all_day = int(is_all_day)
                smart_scene_dict['is_all_day'] = is_all_day

            # 处理传网关设备id和子设备id的情况
            if device_id:
                smart_scene_dict['device_id'] = device_id
                device_info_qs = Device_Info.objects.filter(id=device_id).values('serial_number')
                if not device_info_qs.exists():
                    return response.json(173)
                serial_number = device_info_qs[0]['serial_number']
            else:
                smart_scene_dict['sub_device_id'] = sub_device_id
                sub_device_qs = GatewaySubDevice.objects.filter(id=sub_device_id).values('device__serial_number')
                if not sub_device_qs.exists():
                    return response.json(173)
                serial_number = sub_device_qs[0]['device__serial_number']

            with transaction.atomic():
                if not is_all_day or is_all_day == 1:  # 不设置时间或全天
                    smart_scene_qs = SmartScene.objects.create(**smart_scene_dict)
                elif is_all_day == 2:
                    start_time = int(request_dict.get('startTime', None))
                    end_time = int(request_dict.get('endTime', None))
                    repeat = int(request_dict.get('repeat', None))
                    effective_time_qs = EffectiveTime.objects.filter(start_time=start_time, end_time=end_time,
                                                                     repeat=repeat).values('id')
                    if effective_time_qs.exists():
                        effective_time_id = effective_time_qs[0]['id']
                    else:
                        effective_time_id = EffectiveTime.objects.create(start_time=start_time, end_time=end_time,
                                                                         repeat=repeat).id
                    smart_scene_dict['effective_time_id'] = effective_time_id
                    smart_scene_qs = SmartScene.objects.create(**smart_scene_dict)
                else:
                    return response.json(444, {'error param': 'invalid isAllDay'})

                msg = {}
                if conditions_dict['type'] == 2:  # 条件为选择子设备
                    sub_device_id = conditions_dict['sensor']['subDeviceId']
                    sub_device_qs = GatewaySubDevice.objects.filter(id=sub_device_id).values('src_addr').first()
                    msg['smart_scene_id'] = smart_scene_qs.id
                    msg['scene_status'] = 1
                    msg['sensor_type'] = conditions_dict['sensor']['device_type']
                    msg['sensor_src'] = int(sub_device_qs['src_addr'], 16)
                    msg['sensor_status'] = int(conditions_dict['sensor']['eventValues'][0]['value'])

                task_list = []
                for task in tasks_list:
                    task_temp = {
                        'sensor_type': task['device_type'],
                        'sensor_action': task['event_type']
                    }
                    sub_device_id = task.get('subDeviceId', None)
                    if sub_device_id:
                        sub_device_qs = GatewaySubDevice.objects.filter(id=sub_device_id).values('src_addr').first()
                        task_temp['sensor_src'] = int(sub_device_qs['src_addr'], 16)
                    task_list.append(task_temp)
                msg['task'] = task_list
                smart_scene_qs.device_data = json.dumps(msg)
                smart_scene_qs.save()
                # 发布MQTT消息通知网关设备
                thing_name = serial_number
                topic_name = 'loocam/gateway_sensor/{}/smart_scene'.format(serial_number)

                success = CommonService.req_publish_mqtt_msg(thing_name, topic_name, msg)
                if not success:
                    return response.json(10044)

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

    @staticmethod
    def scene_list(request_dict, user_id, response):
        """
        查询智能场景列表
        @param request_dict: 请求参数
        @param user_id: 用户id
        @request_dict deviceId: 网关设备id
        @request_dict subDeviceId: 子设备id
        @param response: 响应对象
        @return: response
        """
        device_id = request_dict.get('deviceId', None)
        sub_device_id = request_dict.get('subDeviceId', None)

        if not any([device_id, sub_device_id]):
            return response.json(444, {'error param': 'deviceId or subDeviceId'})
        try:
            if device_id:
                smart_scene_qs = SmartScene.objects.filter(user_id=user_id, device_id=device_id)
            else:
                smart_scene_qs = SmartScene.objects.filter(user_id=user_id, sub_device_id=sub_device_id)
            if not smart_scene_qs.exists():
                return response.json(173)
            smart_scene_qs = smart_scene_qs.values('id', 'scene_name', 'is_enable')
            return response.json(0, list(smart_scene_qs))
        except Exception as e:
            return response.json(500, repr(e))

    @staticmethod
    def update_status(request_dict, response):
        """
        更新智能场景状态
        @param request_dict: 请求参数
        @request_dict smartSceneId: 智能场景id
        @request_dict isEnable: 状态,True or False
        @param response: 响应对象
        @return: response
        """
        smart_scene_id = request_dict.get('smartSceneId', None)
        is_enable = request_dict.get('isEnable', None)

        if not all([smart_scene_id, is_enable]):
            return response.json(444, {'error param': 'smartSceneId and status'})
        try:
            SmartScene.objects.filter(id=smart_scene_id).update(is_enable=is_enable)
            return response.json(0)
        except Exception as e:
            return response.json(500, repr(e))

    @staticmethod
    def scene_detail(request_dict, response):
        """
        查询智能场景详情
        @param request_dict: 请求参数
        @request_dict smartSceneId: 智能场景id
        @param response: 响应对象
        @return: response
        """
        smart_scene_id = request_dict.get('smartSceneId', None)

        if not smart_scene_id:
            return response.json(444, {'error param': 'smartSceneId'})
        try:
            smart_scene_qs = SmartScene.objects.filter(id=smart_scene_id).values('id', 'scene_name', 'conditions',
                                                                                 'tasks', 'effective_time_id',
                                                                                 'is_all_day')
            if not smart_scene_qs.exists():
                return response.json(173)
            res = {
                'scene_name': smart_scene_qs[0]['scene_name'],
                'condition': eval(smart_scene_qs[0]['conditions']),
                'task': eval(smart_scene_qs[0]['tasks']),
            }

            # 如果存在关联的时间数据,组织时间数据
            is_all_day = smart_scene_qs[0]['is_all_day']
            effectiveTime = {}
            if is_all_day != 0:
                effectiveTime['isAllDay'] = is_all_day
            if is_all_day == 2:
                try:
                    effective_time_qs = EffectiveTime.objects.get(id=smart_scene_qs[0]['effective_time_id'])
                    effectiveTime['startTime'] = effective_time_qs.start_time
                    effectiveTime['endTime'] = effective_time_qs.end_time
                    effectiveTime['repeat'] = effective_time_qs.repeat
                except ObjectDoesNotExist:
                    return response.json(0, res)
            res['effectiveTime'] = effectiveTime
            return response.json(0, res)

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

    @staticmethod
    def edit_smart_scene(request_dict, response):
        """
        编辑智能场景
        @param request_dict: 请求参数
        @request_dict smartSceneId: 智能场景id
        @param response: 响应对象
        @return: response
        """
        smart_scene_id = request_dict.get('smartSceneId', None)
        scene_name = request_dict.get('sceneName', None)
        conditions = request_dict.get('conditions', None)
        tasks = request_dict.get('tasks', None)
        is_all_day = request_dict.get('isAllDay', None)

        conditions_dict = eval(conditions)
        tasks_list = eval(tasks)
        res = {
            'scene_name': scene_name,
            'conditions': conditions_dict,
            'tasks': tasks_list
        }
        effective_time = {}

        if is_all_day:
            is_all_day = int(is_all_day)
            effective_time['is_all_day'] = is_all_day
        else:
            is_all_day = 0

        if not all([smart_scene_id, scene_name, conditions, tasks]):
            return response.json(444, {'error param': 'smartSceneId,sceneName,conditions or tasks'})
        try:
            smart_scene_qs = SmartScene.objects.filter(id=smart_scene_id)
            if not smart_scene_qs.exists():
                return response.json(173)
            msg = {}
            if conditions_dict['type'] == 2:  # 条件为选择子设备
                sub_device_id = conditions_dict['sensor']['subDeviceId']
                sub_device_qs = GatewaySubDevice.objects.filter(id=sub_device_id).values('src_addr').first()
                msg['smart_scene_id'] = smart_scene_id
                msg['scene_status'] = 1
                msg['sensor_type'] = conditions_dict['sensor']['device_type']
                msg['sensor_src'] = int(sub_device_qs['src_addr'], 16)
                msg['sensor_status'] = int(conditions_dict['sensor']['eventValues'][0]['event_type'])

            task_list = []
            for task in tasks_list:
                task_temp = {
                    'sensor_type': task['device_type'],
                    'sensor_action': task['event_type']
                }
                sub_device_id = task.get('subDeviceId', None)
                if sub_device_id:
                    sub_device_qs = GatewaySubDevice.objects.filter(id=sub_device_id).values('src_addr').first()
                    task_temp['sensor_src'] = int(sub_device_qs['src_addr'], 16)
                task_list.append(task_temp)
            msg['task'] = task_list

            with transaction.atomic():
                smart_scene_qs.update(scene_name=scene_name, conditions=conditions, tasks=tasks,
                                      device_data=json.dumps(msg))
                if is_all_day == 0 or is_all_day == 1:  # 不设置时间或全天
                    smart_scene_qs.update(effective_time_id=0, is_all_day=is_all_day)
                else:
                    start_time = int(request_dict.get('startTime', None))
                    end_time = int(request_dict.get('endTime', None))
                    repeat = int(request_dict.get('repeat', None))
                    effective_time_qs = EffectiveTime.objects.filter(start_time=start_time, end_time=end_time,
                                                                     repeat=repeat).values('id')
                    if effective_time_qs.exists():
                        effective_time_id = effective_time_qs[0]['id']
                    else:
                        effective_time_id = EffectiveTime.objects.create(start_time=start_time, end_time=end_time,
                                                                         repeat=repeat).id
                    smart_scene_qs.update(effective_time_id=effective_time_id, is_all_day=is_all_day)
                    effective_time = {
                        'isAllDay': is_all_day,
                        'startTime': start_time,
                        'endTime': end_time,
                        'repeat': repeat
                    }
            res['effectiveTime'] = effective_time
            return response.json(0, res)

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

    @staticmethod
    def delete_smart_scene(request_dict, response):
        """
        删除智能场景
        @param request_dict: 请求参数
        @request_dict smartSceneIds: 智能场景id
        @param response: 响应对象
        @return: response
        """
        smart_scene_ids = request_dict.get('smartSceneIds', None)

        if not smart_scene_ids:
            return response.json(444, {'error param': 'smartSceneIds'})
        try:
            SmartScene.objects.filter(id__in=smart_scene_ids.split(',')).delete()
        except Exception as e:
            return response.json(500, repr(e))
        else:
            return response.json(0)