ASJServer/Controller/SensorGateway/SubDeviceController.py

# -*- 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('serial_number', None)
        device_type = int(request_dict.get('device_type', None))
        nickname = request_dict.get('nickname', None)
        request_dict.get('ieee_addr', None)
        src_addr = request_dict.get('src_addr', None)
        family_id = request_dict.get('family_id', None)
        room_id = request_dict.get('room_id', None)

        if not all([serial_number, device_type, nickname, src_addr, family_id, room_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']
            with transaction.atomic():
                sub_device = GatewaySubDevice.objects.create(device_id=device_id, device_type=device_type,
                                                             nickname=nickname, 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, 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('serial_number', 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', '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: 请求参数
        @param response: 响应结果
        @return: list
        """
        serial_number = request_dict.get('serialNumber', None)
        nickName = request_dict.get('nickName', None)
        gateway_sub_id = request_dict.get('gatewaySubId', None)
        family_id = request_dict.get('familyId', None)
        room_id = request_dict.get('roomId', None)
        if not all([serial_number, gateway_sub_id]):
            return response.json(444)
        try:
            gateway_sub_id = int(gateway_sub_id)
            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)
            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)
                return response.json(0)
            elif all([family_id, room_id]):
                device_id = device_info_qs.first()['id']
                room_id = int(room_id)
                FamilyRoomDevice.objects.filter(device_id=device_id, 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('sub_device_id', 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))