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 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:
        @param response:
        @return: res
        """
        # 该用户已存在该组名的分组
        device_group_qs = DeviceGroup.objects.filter(user_id=user_id).values('id', 'group_name')
        if not device_group_qs.exists():
            return response.json(0)
        res = []
        for device_group in device_group_qs:
            device_quantity = Device_Info.objects.filter(device_group_id=device_group['id']).count()
            device_group['device_quantity'] = device_quantity
            res.append(device_group)
        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')
        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:
            # 查询设备数量不为0的其他分组
            device_group_qs = DeviceGroup.objects.filter(~Q(id=device_group_id), user_id=user_id).\
                values('id', 'group_name')
            if not device_group_qs.exists():
                return response.json(0)
            res = []
            # 查询各组的设备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')
                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)))