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 == 'get-device-information':
            return self.get_device_information(request_dict, 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, {'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 gatewaySubId: 子设备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('gatewaySubId', 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 get_device_information(request_dict, response):

        device_id = request_dict.get('deviceId', None)
        if not device_id:
            return response.json(444)
        device_qs = FamilyRoomDevice.objects.filter(device_id=device_id, sub_device=0)
        if not device_qs.exists():
            return response.json(173)
        gate_way = GatewaySubDevice.objects.filter(device_id=device_id)
        if not gate_way.exists():
            return response.json(173)
        try:
            gate_way = gate_way.values('id', 'device_type', 'nickname', 'status')
            gate_way_list = []
            for gate_int in gate_way:
                gate_way.exists()
                gate_way_list.append(gate_int)
            device_qs = device_qs.values('family_id', 'device_id', 'room_id', 'device__Type', 'device__NickName',
                                         'device__UID')
            device_qs_list = []
            for device_int in device_qs:
                device_qs.exists()
                device_qs_list.append(device_int)
            device_qs = device_qs.first()
            family_id = device_qs['family_id']
            room_id = device_qs['room_id']
            room_qs = FamilyRoom.objects.filter(id=room_id).values('name')
            room_qs_list = []
            for room_int in room_qs:
                room_qs.exists()
                room_qs_list.append(room_int)

            repe = {'device_qs_list': device_qs_list, 'gate_way_list': gate_way_list, 'room_qs_list': room_qs_list}

            return response.json(0, repe)

        except Exception as e:
            print(e.args)
            return response.json(500)