# -*- coding: utf-8 -*- """ @Time ： 2022/5/25 15:17 @Auth ： Locky @File ：SubDeviceController.py @IDE ：PyCharm """ import time from django.db import transaction from django.views import View from Model.models import Device_Info, GatewaySubDevice, FamilyRoomDevice from Object.ResponseObject import ResponseObject from Object.TokenObject import TokenObject class GatewaySubDeviceView(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 == 'add': # 添加子设备 return self.add(request_dict, user_id, response) elif operation == 'query': # 查询子设备 return self.query(request_dict, user_id, response) elif operation == 'update': # 更新子设备信息 return self.sensor_update(user_id, request_dict, response) elif operation == 'delete': # 删除子设备 return self.delete(request_dict, user_id, response) else: return response.json(414) @staticmethod def add(request_dict, user_id, response): """ 添加子设备 @param request_dict: 请求参数 @request_dict serial_number: 序列号 @request_dict device_type: 设备类型 @request_dict nickname: 设备名 @request_dict src_addr: 短地址 @request_dict family_id: 家庭id @request_dict room_id: 房间id @param user_id: 用户id @param response: 响应对象 @return: response 响应对象 """ serial_number = request_dict.get('serialNumber', None) device_type = int(request_dict.get('deviceType', None)) nickname = request_dict.get('nickName', None) ieee_addr = request_dict.get('ieeeAddr', None) src_addr = request_dict.get('srcAddr', None) family_id = request_dict.get('familyId', None) room_id = request_dict.get('roomId', 0) if not all([serial_number, device_type, nickname, src_addr, family_id]): return response.json(444) now_time = int(time.time()) try: # 查询网关设备数据 device_info_qs = Device_Info.objects.filter(userID_id=user_id, serial_number=serial_number).values('id') if not device_info_qs.exists(): return response.json(14) device_id = device_info_qs[0]['id'] # 查询是否已添加过该子设备 sub_device_qs = GatewaySubDevice.objects.filter(device_id=device_id, ieee_addr=ieee_addr) if sub_device_qs.exists(): return response.json(174) with transaction.atomic(): sub_device = GatewaySubDevice.objects.create(device_id=device_id, device_type=device_type, nickname=nickname, ieee_addr=ieee_addr, src_addr=src_addr, status=1, created_time=now_time, updated_time=now_time) FamilyRoomDevice.objects.create(family_id=family_id, room_id=room_id, device_id=device_id, sub_device=sub_device.id, created_time=now_time, updated_time=now_time) return response.json(0) except Exception as e: return response.json(500, repr(e)) @staticmethod def query(request_dict, user_id, response): """ 查询子设备 @param request_dict: 请求参数 @request_dict serial_number: 序列号 @param user_id: 用户id @param response: 响应对象 @return: response """ serial_number = request_dict.get('serialNumber', None) if not all([serial_number]): return response.json(444) try: device_info_qs = Device_Info.objects.filter(userID_id=user_id, serial_number=serial_number).values('id') if not device_info_qs.exists(): return response.json(14) device_id = device_info_qs[0]['id'] count = GatewaySubDevice.objects.filter(device_id=device_id).count() gateway_sub_device_qs = GatewaySubDevice.objects.filter(device_id=device_id).values('device_type', 'nickname', 'ieee_addr', 'src_addr', 'status') gateway_sub_device_list = [gateway_sub_device for gateway_sub_device in gateway_sub_device_qs] res = { 'count': count, 'gateway_sub_device_list': gateway_sub_device_list } return response.json(0, res) except Exception as e: return response.json(500, repr(e)) @classmethod def sensor_update(cls, user_id, request_dict, response): """ 传感器修改 @param user_id: 登录用户id @param request_dict: 请求参数 @request_dict gateway_sub_id: 子设备id @request_dict nickName: 设备名 @request_dict room_id: 房间id @param response: 响应结果 @return: list """ gateway_sub_id = int(request_dict.get('gatewaySubId', None)) nickName = request_dict.get('nickName', None) room_id = request_dict.get('roomId', None) if not all([gateway_sub_id]): return response.json(444) try: gateway_sub_qs = GatewaySubDevice.objects.filter(id=gateway_sub_id) if not gateway_sub_qs.exists(): return response.json(173) if nickName: gateway_sub_qs.update(nickname=nickName) elif room_id: FamilyRoomDevice.objects.filter(sub_device=gateway_sub_id).update(room_id=room_id) return response.json(0) except Exception as e: print(e.args) return response.json(500, repr(e)) @staticmethod def delete(request_dict, user_id, response): """ 更新子设备信息 @param request_dict: 请求参数 @request_dict sub_device_id: 子设备id @param user_id: 用户id @param response: 响应对象 @return: response """ sub_device_id = request_dict.get('subDeviceId', None) if not all([sub_device_id]): return response.json(444) try: GatewaySubDevice.objects.filter(id=sub_device_id).delete() return response.json(0) except Exception as e: return response.json(500, repr(e))