ASJServer/Controller/DeviceManagement/DeviceGroupController.py

# @Author    : Rocky
# @File      : DeviceGroupController.py
# @Time      : 2025/1/22 9:51
from django.db import transaction
from django.db.models import Count
from django.db.models import Q
from django.views import View

from Model.models import DeviceGroup, Device_Info
from Service.CommonService import CommonService


class DeviceGroupView(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 == 'queryGroupQuantity':  # 查询分组数量
            return self.query_group_quantity(request_dict, user_id, response)
        elif operation == 'queryGroupDevice':  # 查询分组设备
            return self.query_group_device(request_dict, user_id, response)
        elif operation == 'queryOtherGroupDevice':  # 查询其他分组设备
            return self.query_other_group_device(request_dict, user_id, response)
        elif operation == 'creat':  # 创建分组
            return self.creat(request_dict, user_id, response)
        elif operation == 'update':  # 重命名分组
            return self.update(request_dict, user_id, response)
        elif operation == 'delete':  # 删除分组
            return self.delete(request_dict, user_id, response)
        elif operation == 'transferGroup':  # 转移分组
            return self.transfer_group(request_dict, user_id, response)
        else:
            return response.json(414)

    @staticmethod
    def query_group_quantity(request_dict, user_id, response):
        """
        查询分组数量（优化版：减少数据库查询次数，提升性能）
        @param request_dict: 请求参数
        @param user_id: 用户ID
        @param response: 响应对象
        @return: 包含分组信息和设备总数的响应
        """

        # 1. 一次查询获取所有分组ID和名称（仅1次数据库查询）
        device_groups = list(DeviceGroup.objects.filter(user_id=user_id).values('id', 'group_name'))
        if not device_groups:
            return response.json(0, {'total': 0, 'device_groups': []})

        # 2. 提取所有分组ID，批量查询设备数量（仅1次数据库查询）
        group_ids = [group['id'] for group in device_groups]
        device_counts = dict(
            Device_Info.objects.filter(device_group_id__in=group_ids)
            .values('device_group_id')
            .annotate(count=Count('id'))  # 聚合计算每个分组的设备数
            .values_list('device_group_id', 'count')  # 转换为(id, count)格式
        )

        # 3. 内存中组装数据（无额外数据库查询）
        device_total = 0
        for group in device_groups:
            # 从字典中获取数量，默认0（避免不存在的分组ID报错）
            count = device_counts.get(group['id'], 0)
            group['device_quantity'] = count
            device_total += count

        return response.json(0, {
            'total': device_total,
            'device_groups': device_groups
        })

    @staticmethod
    def query_group_device(request_dict, user_id, response):
        """
        查询分组设备
        @param request_dict:
        @param user_id:
        @param response:
        @return: res
        """
        device_group_id = request_dict.get('device_group_id', None)
        if not device_group_id:
            return response.json(444)
        device_info_qs = Device_Info.objects.filter(device_group_id=device_group_id).values('NickName', 'Type')
        if not device_info_qs.exists():
            return response.json(0, [])
        res = list(device_info_qs)
        return response.json(0, res)

    @staticmethod
    def query_other_group_device(request_dict, user_id, response):
        """
        查询其他分组设备
        @param request_dict:
        @param user_id:
        @param response:
        @return: res
        """
        device_group_id = request_dict.get('device_group_id', None)
        if not device_group_id:
            return response.json(444)
        try:
            res = []
            # 查询全部设备
            if device_group_id != '0':
                device_info_qs = Device_Info.objects.filter(~Q(device_group_id=device_group_id), userID=user_id). \
                    values('id', 'NickName', 'Type')
                if device_info_qs.exists():
                    data = {
                        'id': 0,
                        'group_name': 'all'
                    }
                    device_info_list = []
                    for device_info in device_info_qs:
                        device_info_list.append(device_info)
                    data['device_info_list'] = device_info_list
                    res.append(data)

            # 查询设备数量不为0的其他分组
            device_group_qs = DeviceGroup.objects.filter(~Q(id=device_group_id), user_id=user_id). \
                values('id', 'group_name')
            if device_group_qs.exists():
                # 查询各组的设备id和昵称
                for device_group in device_group_qs:
                    device_info_list = []
                    device_info_qs = Device_Info.objects.filter(device_group_id=device_group['id']). \
                        values('id', 'NickName', 'Type')
                    for device_info in device_info_qs:
                        device_info_list.append(device_info)
                    device_group['device_info_list'] = device_info_list
                    res.append(device_group)
            return response.json(0, res)
        except Exception as e:
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    @staticmethod
    def creat(request_dict, user_id, response):
        """
        创建分组
        @param request_dict:
        @param user_id:
        @param response:
        @return: res
        """
        group_name = request_dict.get('group_name', None)
        if not group_name:
            return response.json(444)
        # 该用户已存在该组名的分组
        device_group_qs = DeviceGroup.objects.filter(user_id=user_id, group_name=group_name)
        if device_group_qs.exists():
            return response.json(174)
        DeviceGroup.objects.create(user_id=user_id, group_name=group_name)
        return response.json(0)

    @staticmethod
    def update(request_dict, user_id, response):
        """
        重命名分组
        @param request_dict:
        @param user_id:
        @param response:
        @return: res
        """
        device_group_id = request_dict.get('device_group_id', None)
        group_name = request_dict.get('group_name', None)
        if not all([device_group_id, group_name]):
            return response.json(444)
        # 该用户已存在该组名的分组
        device_group_qs = DeviceGroup.objects.filter(user_id=user_id, group_name=group_name)
        if device_group_qs.exists():
            return response.json(174)
        DeviceGroup.objects.filter(id=device_group_id).update(group_name=group_name)
        return response.json(0)

    @staticmethod
    def delete(request_dict, user_id, response):
        """
        删除分组
        @param request_dict:
        @param user_id:
        @param response:
        @return: res
        """
        device_group_id = request_dict.get('device_group_id', None)
        if not device_group_id:
            return response.json(444)
        try:
            device_group_id = int(device_group_id)
            with transaction.atomic():
                # 删除设备分组数据,重置设备信息表的分组id
                DeviceGroup.objects.filter(id=device_group_id).delete()
                Device_Info.objects.filter(device_group_id=device_group_id).update(device_group_id=0)
            return response.json(0)
        except Exception as e:
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    @staticmethod
    def transfer_group(request_dict, user_id, response):
        """
        转移分组
        @param request_dict:
        @param user_id:
        @param response:
        @return: res
        """
        device_group_id = request_dict.get('device_group_id', None)
        device_id_list = request_dict.get('device_id_list', None)
        if not all([device_group_id, device_id_list]):
            return response.json(444)
        try:
            device_group_id = int(device_group_id)
            device_id_list = device_id_list.split(',')
            # 更新设备信息表的分组id
            Device_Info.objects.filter(id__in=device_id_list).update(device_group_id=device_group_id)
            return response.json(0)
        except Exception as e:
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))