ASJServer/Controller/SensorGateway/SmartSceneController.py

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

from django.core.exceptions import ObjectDoesNotExist
from django.views import View

from Model.models import FamilyRoomDevice, GatewaySubDevice, FamilyRoom, SmartScene, EffectiveTime
from Object.ResponseObject import ResponseObject
from Object.TokenObject import TokenObject


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_obj = TokenObject(request.META.get('HTTP_AUTHORIZATION'))
        lang = request_dict.get('lang', None)
        response = ResponseObject(lang if lang else token_obj.lang)

        if token_obj.code != 0:
            return response.json(token_obj.code)
        user_id = token_obj.userID
        if operation == 'condition-devices':  # 添加条件-查询设备
            return self.condition_devices(request_dict, response)
        if operation == 'task-devices':  # 添加任务-查询设备
            return self.task_devices(request_dict, user_id, 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)
        else:
            return response.json(414)

    @staticmethod
    def condition_devices(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 device_id:
                gateway_sub_device_qs = GatewaySubDevice.objects.filter(device_id=device_id)
            else:
                gateway_sub_device_qs = GatewaySubDevice.objects.filter(id=sub_device_id)
            if not gateway_sub_device_qs.exists():
                return response.json(173)
            gateway_sub_device_qs = gateway_sub_device_qs.values('id', 'device_type', 'nickname', 'status')
            sub_device_list = []
            for gateway_sub_device in gateway_sub_device_qs:
                room_id = FamilyRoomDevice.objects.filter(sub_device=gateway_sub_device['id']).values('room_id')[0][
                    'room_id']
                try:
                    gateway_sub_device['room_name'] = FamilyRoom.objects.get(id=room_id).name
                except ObjectDoesNotExist:
                    gateway_sub_device['room_name'] = ''
                gateway_sub_device.pop('id')
                sub_device_list.append(gateway_sub_device)
            return response.json(0, sub_device_list)
        except Exception as e:
            return response.json(500, repr(e))

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

        if not all([device_id]):
            return response.json(444)
        try:
            device_info_qs = GatewaySubDevice.objects.filter(device__userID_id=user_id)
            return response.json(0)
        except Exception as e:
            return response.json(500, repr(e))

    @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())
        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 device_id:
                smart_scene_dict['device_id'] = device_id
            else:
                smart_scene_dict['sub_device_id'] = sub_device_id
            if is_all_day:
                SmartScene.objects.create(**smart_scene_dict)
            else:
                start_time = request_dict.get('startTime', None)
                end_time = request_dict.get('endTime', None)
                repeat = request_dict.get('repeat', None)
                if not all([start_time, end_time, repeat]):
                    return response.json(444, {'error param': 'startTime and endTime and repeat'})
                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
                SmartScene.objects.create(**smart_scene_dict)
            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))