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 Q
from django.views import View
from django.db.models import Count, Sum
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: res 包含分组信息和设备总数的响应
        """

        # 一次查询获取所有分组及对应的设备数量
        device_group_qs = DeviceGroup.objects.filter(user_id=user_id) \
            .annotate(device_quantity=Count('device_info', distinct=True)) \
            .values('id', 'group_name', 'device_quantity')

        # 计算设备总数
        total_result = device_group_qs.aggregate(total=Sum('device_quantity'))
        device_total = total_result['total'] or 0  # 处理没有分组的情况

        res = {
            'total': device_total,
            'device_groups': list(device_group_qs)  # 直接转换为列表
        }

        return response.json(0, res)

    @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)))