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)
        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)
        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 gatewaySubId: 子设备id
        @param response: 响应对象
        @return: response
        """
        sub_device_id = request_dict.get('gatewaySubId', None)
        scene_name = request_dict.get('sceneName', None)
        conditions = request_dict.get('conditions', None)
        tasks = request_dict.get('tasks', None)
        all_day = request_dict.get('allDay', None)

        if not all([sub_device_id, conditions, tasks]):
            return response.json(444)
        now_time = int(time.time())
        try:
            if all_day:
                SmartScene.objects.create(user_id=user_id, scene_name=scene_name, conditions=conditions, tasks=tasks,
                                          created_time=now_time, updated_time=now_time)
            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)
                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
                SmartScene.objects.create(user_id=user_id, scene_name=scene_name, conditions=conditions, tasks=tasks,
                                          effective_time_id=effective_time_id, created_time=now_time, updated_time=now_time)
            return response.json(0)
        except Exception as e:
            return response.json(500, repr(e))