ASJServer/Controller/SensorGateway/SubDeviceController.py

# -*- coding: utf-8 -*-
"""
@Author ： Rocky
@Time ： 2022/5/25 15:17
@File ：SubDeviceController.py
"""
import time
from collections import OrderedDict

from django.db import transaction
from django.db.models import Count
from django.views import View

from Model.models import Device_Info, GatewaySubDevice, FamilyRoomDevice, SensorRecord
from Object.ResponseObject import ResponseObject
from Object.TokenObject import TokenObject
from Service.CommonService import CommonService


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_sub_device(request_dict, user_id, response)
        elif operation == 'query':  # 查询子设备
            return self.query(request_dict, user_id, response)
        elif operation == 'update':  # 更新子设备信息
            return self.sensor_update(request_dict, response)
        elif operation == 'delete':  # 删除子设备
            return self.delete(request_dict, user_id, response)
        elif operation == 'records/tem-hum':  # 查询温湿度传感器记录
            return self.records_tem_hum(request_dict, response)
        elif operation == 'records':  # 查询其他传感器记录
            return self.records(request_dict, response)
        elif operation == 'records-date':  # 查询传感器记录日期
            return self.records_date(request_dict, response)
        else:
            return response.json(414)

    @staticmethod
    def add_sub_device(request_dict, user_id, response):
        """
        添加子设备
        @param request_dict: 请求参数
        @request_dict serialNumber: 序列号
        @request_dict deviceType: 设备类型
        @request_dict nickName: 设备名
        @request_dict ieeeAddr: 长地址
        @request_dict srcAddr: 短地址
        @request_dict mac: mac地址
        @request_dict deviceModel: 设备型号
        @request_dict manufacturer: 厂家名称
        @request_dict firmwareVersion: 固件版本
        @request_dict hardwareVersion: 硬件版本
        @request_dict familyId: 家庭id
        @request_dict roomId: 房间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)
        mac = request_dict.get('mac', '')
        device_model = request_dict.get('deviceModel', '')
        manufacturer = request_dict.get('manufacturer', '')
        sensor_serial = request_dict.get('sensorSerial', '')
        firmware_version = request_dict.get('firmwareVersion', '')
        hardware_version = request_dict.get('hardwareVersion', '')

        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, mac=mac, device_model=device_model,
                                                             manufacturer=manufacturer, sensor_serial=sensor_serial,
                                                             firmware_version=firmware_version,
                                                             hardware_version=hardware_version,
                                                             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, request_dict, response):
        """
        传感器修改
        @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)
            if 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))

    @staticmethod
    def records_tem_hum(request_dict, response):
        """
        查询温湿度传感器记录
        @param request_dict: 请求参数
        @request_dict gatewaySubId: 子设备id
        @request_dict cycle: 时间周期
        @request_dict eventType: 事件类型, 18:温度,19:湿度
        @param response: 响应对象
        @return: response
        """
        sub_device_id = request_dict.get('gatewaySubId', None)
        cycle = request_dict.get('cycle', None)
        event_type = request_dict.get('eventType', None)
        if not all([sub_device_id, cycle, event_type]):
            return response.json(444, {'error param': 'gatewaySubId or cycle or eventType'})
        now_time = int(time.time())
        try:
            record_dict = OrderedDict()
            record_list = []
            if cycle == 'Hours':
                start_time = now_time - 24 * 60 * 60
                sensor_record_qs = SensorRecord.objects.filter(gateway_sub_device_id=sub_device_id,
                                                               event_type=event_type,
                                                               created_time__range=(start_time, now_time)). \
                    values('alarm', 'created_time').order_by('created_time')
                if not sensor_record_qs.exists():
                    return response.json(0, {'records': [], 'time': now_time})

                for sensor_record in sensor_record_qs:
                    created_time = time.strftime('%m/%d %H:%M %w', time.localtime(sensor_record['created_time']))
                    hour = int(created_time[-7:-5])
                    minute = int(created_time[-4:-2])
                    if hour != 23 and minute > 30:  # 不为23时且分钟大于30,hour+1
                        hour += 1
                    alarm = float(sensor_record['alarm'])
                    # 组织数据,record_dict:{"0": [1.0, 2.0, 3.0], "1": [1.0, 2.0, 3.0]...}
                    if str(hour) in record_dict:
                        record_dict[str(hour)].append(alarm)
                    else:
                        record_dict[str(hour)] = [alarm]

            elif cycle == 'Week':
                start_time = now_time - 24 * 60 * 60 * 7
                sensor_record_qs = SensorRecord.objects.filter(gateway_sub_device_id=sub_device_id,
                                                               event_type=event_type,
                                                               created_time__range=(start_time, now_time)). \
                    values('alarm', 'created_time').order_by('created_time')
                if not sensor_record_qs.exists():
                    return response.json(0, {'records': [], 'time': now_time})

                for sensor_record in sensor_record_qs:
                    created_time = time.strftime('%m/%d %H:%M %w', time.localtime(sensor_record['created_time']))
                    week = int(created_time[-1:])
                    alarm = float(sensor_record['alarm'])
                    # 组织数据,record_dict:{"0": [1.0, 2.0, 3.0], "1": [1.0, 2.0, 3.0]...}
                    if str(week) in record_dict:
                        record_dict[str(week)].append(alarm)
                    else:
                        record_dict[str(week)] = [alarm]

            elif cycle == 'Month':
                start_time = now_time - 24 * 60 * 60 * 30
                sensor_record_qs = SensorRecord.objects.filter(gateway_sub_device_id=sub_device_id,
                                                               event_type=event_type,
                                                               created_time__range=(start_time, now_time)). \
                    values('alarm', 'created_time').order_by('created_time')
                if not sensor_record_qs.exists():
                    return response.json(0, {'records': [], 'time': now_time})

                for sensor_record in sensor_record_qs:
                    created_time = time.strftime('%m/%d %H:%M %w', time.localtime(sensor_record['created_time']))
                    month = int(created_time[:2])
                    day = int(created_time[3:5])
                    date = str(month) + '/' + str(day)
                    alarm = float(sensor_record['alarm'])
                    # 组织数据,record_dict:{"0": [1.0, 2.0, 3.0], "1": [1.0, 2.0, 3.0]...}
                    if date in record_dict:
                        record_dict[date].append(alarm)
                    else:
                        record_dict[date] = [alarm]

            # 组织响应数据列表,value为每 小时/天 的平均值
            for k, v in record_dict.items():
                record_list.append({'key': k, 'value': round(sum(v) / len(v), 1)})

            res = {
                'records': record_list,
                'time': now_time
            }
            return response.json(0, res)
        except Exception as e:
            return response.json(500, repr(e))

    @staticmethod
    def records(request_dict, response):
        """
        查询其他传感器记录
        @param request_dict: 请求参数
        @request_dict gatewaySubId: 子设备id
        @request_dict page: 页数
        @request_dict size: 条数
        @request_dict startTime: 开始时间
        @request_dict endTime: 结束时间
        @param response: 响应对象
        @return: response
        """
        sub_device_id = request_dict.get('gatewaySubId', None)
        page = request_dict.get('page', None)
        size = request_dict.get('size', None)
        if not all([sub_device_id, page, size]):
            return response.json(444, {'error param': 'gatewaySubId or page or size'})

        start_time = request_dict.get('startTime', None)
        end_time = request_dict.get('endTime', None)

        try:
            page, size = int(page), int(size)
            if start_time and end_time:
                sensor_record_qs = SensorRecord.objects.filter(gateway_sub_device_id=sub_device_id,
                                                               created_time__range=(start_time, end_time)). \
                                       values('alarm', 'created_time').order_by('-created_time')[
                                   (page - 1) * size:page * size]
            else:
                sensor_record_qs = SensorRecord.objects.filter(gateway_sub_device_id=sub_device_id). \
                                       values('alarm', 'created_time').order_by('-created_time')[
                                   (page - 1) * size:page * size]
            if not sensor_record_qs.exists():
                return response.json(0, [])
            return response.json(0, list(sensor_record_qs))
        except Exception as e:
            return response.json(500, repr(e))

    @staticmethod
    def records_date(request_dict, response):
        """
        查询传感器记录日期
        @param request_dict: 请求参数
        @request_dict gatewaySubId: 子设备id
        @request_dict startTime: 开始时间
        @request_dict endTime: 结束时间
        @param response: 响应对象
        @return: response
        """
        sub_device_id = request_dict.get('gatewaySubId', None)
        if not sub_device_id:
            return response.json(444, {'error param': 'gatewaySubId'})

        try:
            sensor_record_qs = SensorRecord.objects.extra(
                select={'date': "FROM_UNIXTIME(created_time,'%%Y-%%m-%%d')"}).values('date'). \
                                   filter(gateway_sub_device_id=sub_device_id). \
                                   annotate(count=Count('created_time')). \
                                   order_by('-date')[:31]
            record_date_list = []
            for sensor_record in sensor_record_qs:
                record_date_list.append({
                    'timestamp': CommonService.str_to_timestamp(sensor_record['date'], '%Y-%m-%d'),
                    'count': sensor_record['count'],
                    'format': sensor_record['date'],
                })
            return response.json(0, record_date_list)
        except Exception as e:
            return response.json(500, repr(e))