# -*- coding: utf-8 -*- """ @Author ： Rocky @Time ： 2022/6/29 9:31 @File ：SmartSceneController.py """ import json import time from django.core.exceptions import ObjectDoesNotExist from django.db import transaction from django.db.models import F, Q from django.views import View 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 not None: 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 is_all_day is None: # 不设置时间 smart_scene_qs = SmartScene.objects.create(**smart_scene_dict) # 设备的time数据 hour, minute = conditions_dict['time']['minutes'] / 60, conditions_dict['time']['minutes'] % 60 start_time = int(hour * 3600 + minute * 60) time_dict = { 'start_time': start_time, 'repeat': conditions_dict['time']['repeat'] } elif is_all_day == 1: # 全天 smart_scene_qs = SmartScene.objects.create(**smart_scene_dict) # 设备的time数据 time_dict = { 'is_all_day': is_all_day } 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) # 设备的time数据 start_hour, start_minute, end_hour, end_minute = \ start_time / 60, start_time % 60, end_time / 60, end_time % 60 start_time, end_time = \ int(start_hour * 3600 + start_minute * 60), \ int(end_hour * 3600 + end_minute * 60) time_dict = { 'is_all_day': is_all_day, 'start_time': start_time, 'end_time': end_time, 'repeat': repeat } else: return response.json(444, {'error param': 'invalid isAllDay'}) msg = { 'time': time_dict } 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) try: assert success except AssertionError: 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)