# -*- encoding: utf-8 -*- """ @File : GatewayFamilyRoomController.py @Time : 2022/5/24 19:43 @Author : stephen @Email : zhangdongming@asj6.wecom.work @Software: PyCharm """ from django.db import transaction from django.db.models import Q, Count, F from django.views.generic.base import View from Controller.SensorGateway.EquipmentFamilyController import EquipmentFamilyView from Model.models import FamilyRoomDevice, FamilyRoom, GatewaySubDevice, Device_Info from Object.ResponseObject import ResponseObject from Object.TokenObject import TokenObject # 家庭房间管理 class GatewayFamilyRoomView(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 = request.META.get('HTTP_AUTHORIZATION') token = TokenObject(token) lang = request_dict.get('lang', token.lang) response = ResponseObject(lang) if token.code != 0: return response.json(token.code) app_user_id = token.userID # 添加设备关联房间 if operation == 'device-changes': return self.room_device_save(app_user_id, request_dict, response) # 房间排序 elif operation == 'sort': return self.room_sort_save(request_dict, response) # 房间删除 elif operation == 'del': return self.room_del(app_user_id, request_dict, response) # 房间详情 elif operation == 'details': return self.get_room_details(app_user_id, request_dict, response) elif operation == 'all-devices': # 家庭所有设备 return self.all_devices(request_dict, response) elif operation == 'devices-sort': # 家庭设备排序 return self.devices_sort(request_dict, response) elif operation == 'device-category': # 获取设备类别 return self.device_category(request_dict, response) elif operation == 'category-sort': # 设备类别排序 return self.category_sort(request_dict, response) else: return response.json(414) @classmethod def room_device_save(cls, app_user_id, request_dict, response): """ 房间加入设备or移除设备 @param app_user_id: @param request_dict: @param response: @return: """ family_id = request_dict.get('familyId', None) device_ids = request_dict.get('deviceIds', None) room_id = request_dict.get('roomId', None) if not all([family_id, room_id]): return response.json(444) family_id = int(family_id) room_id = int(room_id) is_owner = EquipmentFamilyView.get_family_owner(app_user_id, family_id) if not is_owner: return response.json(404) try: with transaction.atomic(): room_qs = FamilyRoom.objects.filter(family_id=family_id, id=room_id) if not room_qs.exists(): return response.json(173) qs = FamilyRoomDevice.objects.filter(family_id=family_id, room_id=room_id) if qs.exists(): qs.update(room_id=0, sort=0) if device_ids: device_ids = eval(device_ids) for i, item in enumerate(device_ids): device_id = item['id'] device_type = item['type'] device_qs = Device_Info.objects.filter(id=device_id, Type=device_type) if device_qs.exists(): room_device_qs = FamilyRoomDevice.objects.filter(device_id=device_id) else: room_device_qs = FamilyRoomDevice.objects.filter(sub_device=device_id) if room_device_qs.exists(): room_device_qs.update(room_id=room_id, sort=i) return response.json(0) except Exception as e: print(e) return response.json(177, repr(e)) @classmethod def room_del(cls, user_id, request_dict, response): """ 房间多选删除 @param user_id: 当前登录用户id @param request_dict: 请求参数 @param response: 响应参数 @return: """ ids = request_dict.get('roomIds', None) if not ids: return response.json(444) ids = ids.split(',') room_id = ids[0] room_info = FamilyRoom.objects.filter(id=room_id) if not room_info.exists(): return response.json(173) is_owner = EquipmentFamilyView.get_family_owner(user_id, room_info.first().family_id) if not is_owner: return response.json(404) try: with transaction.atomic(): for item in ids: room_id = int(item) room_device = FamilyRoomDevice.objects.filter(room_id=room_id) if room_device.exists(): room_device.update(room_id=0) FamilyRoom.objects.filter(id=room_id).delete() return response.json(0) except Exception as e: print(e) return response.json(177, repr(e)) @classmethod def room_sort_save(cls, request_dict, response): """ 房间排序 @param request_dict: 请求参数 @param response: 响应参数 @return: """ ids = request_dict.get('ids', None) if not ids: return response.json(444) items = ids.split(',') for item in items: item = item.split('-') room_id, sort = item[0], item[1] family_room = FamilyRoom.objects.filter(id=int(room_id)) if family_room.exists(): family_room.update(sort=int(sort)) return response.json(0) @classmethod def get_room_details(cls, app_user_id, request_dict, response): """ 房间设备详情(所在当前房间下，和所在家庭不在当前房间下的主设备) @param app_user_id: @param request_dict: @param response: @return: """ family_id = request_dict.get('familyId', None) room_id = request_dict.get('roomId', None) if not all([family_id, room_id]): return response.json(444) is_owner = EquipmentFamilyView.get_family_owner(app_user_id, family_id) if not is_owner: return response.json(404) family_id = int(family_id) room_id = int(room_id) room_device_qs = FamilyRoomDevice.objects.filter(family_id=family_id, room_id=room_id).order_by('sort').values( 'device_id', 'device__Type', 'device__NickName', 'sub_device') device_room = [] if room_device_qs.exists(): room_name = FamilyRoom.objects.filter(id=room_id) for item in room_device_qs: if not item['sub_device']: device_dict = { 'deviceId': item['device_id'], 'deviceType': item['device__Type'], 'nickName': item['device__NickName'], 'roomName': room_name.first().name if room_name.exists() else '', } else: sub_device_qs = GatewaySubDevice.objects.filter(id=item['sub_device']).values('id', 'device_type', 'nickname') device_dict = { 'deviceId': sub_device_qs[0]['id'], 'deviceType': sub_device_qs[0]['device_type'], 'nickName': sub_device_qs[0]['nickname'], 'roomName': room_name.first().name if room_name.exists() else '', } device_room.append(device_dict) device_not_room = [] device_not_room_qs = FamilyRoomDevice.objects.filter(family_id=family_id) device_not_room_qs = device_not_room_qs.filter(~Q(room_id=room_id)).values('device_id').annotate( count=Count('device_id')).values('room_id', 'device_id', 'device__Type', 'device__NickName') if device_not_room_qs.exists(): for item in device_not_room_qs: name = '' if room_device_qs.exists(): family_room_qs = FamilyRoom.objects.filter(id=item['room_id']) if family_room_qs.exists(): name = family_room_qs.first().name device_not_room.append({ 'deviceId': item['device_id'], 'deviceType': item['device__Type'], 'nickName': item['device__NickName'], 'roomName': name }) return response.json(0, {'deviceRooms': device_room, 'deviceNotRooms': device_not_room}) @staticmethod def all_devices(request_dict, response): """ 家庭所有设备(网关和摄像头设备) @param request_dict: 请求参数 @request_dict familyId: 家庭id @param response: 响应参数 @return: """ family_id = request_dict.get('familyId', None) if not family_id: return response.json(444) try: family_room_device_qs = FamilyRoomDevice.objects.filter(family_id=family_id).values('device_id', 'device__Type', 'device__NickName', 'room_id', 'sub_device', 'category', ).order_by( 'sort') if not family_room_device_qs.exists(): return response.json(0, []) sort = FamilyRoomDevice.objects.filter(family_id=family_id).values('category').annotate( count=Count('category')).values('category', 'category_sort').order_by('category_sort') for item in sort: if item['category'] == 0: item['category_name'] = '网关子设备类' else: item['category_name'] = '摄像头类' gateways = [] cameras = [] # 查询房间名称 for device in family_room_device_qs: room_id = device['room_id'] device_type = device['device__Type'] category = device['category'] room_name = FamilyRoom.objects.filter(id=room_id) if category == 0: # 网关设备 if not device['sub_device']: device_dict = { 'deviceId': device['device_id'], 'deviceType': device_type, 'nickName': device['device__NickName'], 'roomName': room_name.first().name if room_name.exists() else '', } else: sub_device_qs = GatewaySubDevice.objects.filter(id=device['sub_device']).values('device_type', 'nickname') device_dict = { 'deviceId': device['sub_device'], 'deviceType': sub_device_qs[0]['device_type'], 'nickName': sub_device_qs[0]['nickname'], 'roomName': room_name.first().name if room_name.exists() else '', } gateways.append(device_dict) else: # 摄像头设备 cameras.append({ 'deviceId': device['device_id'], 'deviceType': device_type, 'nickName': device['device__NickName'], 'roomName': room_name.first().name if room_name.exists() else '', }) device_room = { 'gateways': gateways, 'cameras': cameras, 'sort': list(sort) } return response.json(0, device_room) except Exception as e: return response.json(500, repr(e)) @staticmethod def devices_sort(request_dict, response): """ 家庭设备排序 @param request_dict: 请求参数 @request_dict 设备id: deviceIds @param response: 响应参数 @return: """ device_ids = request_dict.get('deviceIds', None) if not device_ids: return response.json(444) try: device_ids = eval(device_ids) with transaction.atomic(): for i, item in enumerate(device_ids): device_id = item['id'] device_type = item['type'] device_qs = Device_Info.objects.filter(id=device_id, Type=device_type) if device_qs.exists(): # 更新主设备排序 FamilyRoomDevice.objects.filter(device_id=device_id).update(sort=i) else: # 更新子设备排序 FamilyRoomDevice.objects.filter(sub_device=device_id).update(sort=i) return response.json(0) except Exception as e: return response.json(500, repr(e)) @staticmethod def device_category(request_dict, response): """ 获取设备类别排序 @param request_dict: 请求参数 @request_dict 家庭id: family_id @param response: 响应参数 @return: """ family_id = request_dict.get('familyId', None) if not family_id: return response.json(444) try: family_room_device_qs = FamilyRoomDevice.objects.filter(family_id=family_id).values('category').annotate( count=Count('category')).values('category', 'category_sort').order_by('category_sort') for item in family_room_device_qs: if item['category'] == 0: item['category_name'] = '网关子设备类' else: item['category_name'] = '摄像头类' return response.json(0, list(family_room_device_qs)) except Exception as e: return response.json(500, repr(e)) @staticmethod def category_sort(request_dict, response): """ 家庭设备排序 @param request_dict: 请求参数 @request_dict 家庭id: family_id @request_dict 排序列表: category_sort @param response: 响应参数 @return: """ family_id = request_dict.get('familyId', None) category_sort = request_dict.get('categorySort', None) if not all([family_id, category_sort]): return response.json(444) try: category_sort_list = eval(category_sort) family_room_device_qs = FamilyRoomDevice.objects.filter(family_id=family_id) if family_room_device_qs.exists(): with transaction.atomic(): for item in category_sort_list: category = item['category'] category_sort = item['category_sort'] family_room_device_qs.filter(category=category).update(category_sort=category_sort) return response.json(0) except Exception as e: return response.json(500, repr(e))