ASJServer/Controller/SensorGateway/GatewayFamilyRoomController.py

# -*- encoding: utf-8 -*-
"""
@File    : GatewayFamilyRoomController.py
@Time    : 2022/5/24 19:43
@Author  : stephen
@Email   : zhangdongming@asj6.wecom.work
@Software: PyCharm
"""

from django.db import transaction
from django.db.models import Q, Count, F
from django.views.generic.base import View

from Controller.SensorGateway.EquipmentFamilyController import EquipmentFamilyView
from Model.models import FamilyRoomDevice, FamilyRoom
from Object.ResponseObject import ResponseObject
from Object.TokenObject import TokenObject


# 家庭房间管理
class GatewayFamilyRoomView(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 = request.META.get('HTTP_AUTHORIZATION')
        token = TokenObject(token)
        lang = request_dict.get('lang', token.lang)
        response = ResponseObject(lang)
        if token.code != 0:
            return response.json(token.code)
        app_user_id = token.userID
        # 添加设备关联房间
        if operation == 'device-changes':
            return self.room_device_save(app_user_id, request_dict, response)
        # 房间排序
        elif operation == 'sort':
            return self.room_sort_save(request_dict, response)
        # 房间删除
        elif operation == 'del':
            return self.room_del(app_user_id, request_dict, response)
        # 房间详情
        elif operation == 'details':
            return self.get_room_details(app_user_id, request_dict, response)
        elif operation == 'all-devices':    # 家庭所有设备
            return self.all_devices(request_dict, response)
        elif operation == 'devices-sort':    # 家庭设备排序
            return self.devices_sort(request_dict, response)

    @classmethod
    def room_device_save(cls, app_user_id, request_dict, response):
        """
        房间加入设备or移除设备
        @param app_user_id:
        @param request_dict:
        @param response:
        @return:
        """
        family_id = request_dict.get('familyId', None)
        device_ids = request_dict.get('deviceIds', None)
        room_id = request_dict.get('roomId', None)
        if not all([family_id, room_id]):
            return response.json(444)
        family_id = int(family_id)
        room_id = int(room_id)
        is_owner = EquipmentFamilyView.get_family_owner(app_user_id, family_id)
        if not is_owner:
            return response.json(404)
        try:
            with transaction.atomic():
                room_qs = FamilyRoom.objects.filter(family_id=family_id, id=room_id)
                if not room_qs.exists():
                    return response.json(173)
                qs = FamilyRoomDevice.objects.filter(family_id=family_id, room_id=room_id)
                if qs.exists():
                    qs.update(room_id=0, sort=0)
                if device_ids:
                    device_ids = device_ids.split(',')
                    for i, item in enumerate(device_ids):
                        device_id = int(item)
                        device_qs = FamilyRoomDevice.objects.filter(device_id=device_id)
                        if device_qs.exists():
                            device_qs.update(room_id=room_id, sort=i)
                return response.json(0)
        except Exception as e:
            print(e)
            return response.json(177, repr(e))

    @classmethod
    def room_del(cls, user_id, request_dict, response):
        """
        房间多选删除
        @param user_id: 当前登录用户id
        @param request_dict: 请求参数
        @param response: 响应参数
        @return:
        """
        ids = request_dict.get('roomIds', None)
        if not ids:
            return response.json(444)
        ids = ids.split(',')
        room_id = ids[0]
        room_info = FamilyRoom.objects.filter(id=room_id)
        if not room_info.exists():
            return response.json(173)
        is_owner = EquipmentFamilyView.get_family_owner(user_id, room_info.first().family_id)
        if not is_owner:
            return response.json(404)
        try:
            with transaction.atomic():
                for item in ids:
                    room_id = int(item)
                    room_device = FamilyRoomDevice.objects.filter(room_id=room_id)
                    if room_device.exists():
                        room_device.update(room_id=0)
                    FamilyRoom.objects.filter(id=room_id).delete()
                return response.json(0)
        except Exception as e:
            print(e)
            return response.json(177, repr(e))

    @classmethod
    def room_sort_save(cls, request_dict, response):
        """
        房间排序
        @param request_dict: 请求参数
        @param response: 响应参数
        @return:
        """
        ids = request_dict.get('ids', None)
        if not ids:
            return response.json(444)
        items = ids.split(',')
        for item in items:
            item = item.split('-')
            room_id, sort = item[0], item[1]
            family_room = FamilyRoom.objects.filter(id=int(room_id))
            if family_room.exists():
                family_room.update(sort=int(sort))
        return response.json(0)

    @classmethod
    def get_room_details(cls, app_user_id, request_dict, response):
        """
        房间设备详情(所在当前房间下，和所在家庭不在当前房间下的主设备)
        @param app_user_id:
        @param request_dict:
        @param response:
        @return:
        """
        family_id = request_dict.get('familyId', None)
        room_id = request_dict.get('roomId', None)
        if not all([family_id, room_id]):
            return response.json(444)
        is_owner = EquipmentFamilyView.get_family_owner(app_user_id, family_id)
        if not is_owner:
            return response.json(404)
        family_id = int(family_id)
        room_id = int(room_id)
        room_device_qs = FamilyRoomDevice.objects.filter(family_id=family_id, room_id=room_id).order_by('sort').values(
            'device_id').annotate(count=Count('device_id')).values('device_id', 'device__Type', 'device__NickName')
        device_room = []
        if room_device_qs.exists():
            room_name = FamilyRoom.objects.filter(id=room_id)
            for item in room_device_qs:
                device_room.append({
                    'deviceId': item['device_id'],
                    'deviceType': item['device__Type'],
                    'nickName': item['device__NickName'],
                    'roomName': room_name.first().name if room_name.exists() else '',
                })

        device_not_room = []
        device_not_room_qs = FamilyRoomDevice.objects.filter(family_id=family_id)
        device_not_room_qs = device_not_room_qs.filter(~Q(room_id=room_id)).values('device_id').annotate(
            count=Count('device_id')).values('room_id', 'device_id', 'device__Type', 'device__NickName')
        if device_not_room_qs.exists():
            for item in device_not_room_qs:
                name = ''
                if room_device_qs.exists():
                    family_room_qs = FamilyRoom.objects.filter(id=item['room_id'])
                    if family_room_qs.exists():
                        name = family_room_qs.first().name
                device_not_room.append({
                    'deviceId': item['device_id'],
                    'deviceType': item['device__Type'],
                    'nickName': item['device__NickName'],
                    'roomName': name
                })
        return response.json(0, {'deviceRooms': device_room, 'deviceNotRooms': device_not_room})

    @staticmethod
    def all_devices(request_dict, response):
        """
        家庭所有设备(网关和摄像头设备)
        @param request_dict: 请求参数
        @request_dict familyId: 家庭id
        @param response: 响应参数
        @return:
        """
        family_id = request_dict.get('familyId', None)
        if not family_id:
            return response.json(444)
        try:
            family_room_device_qs = FamilyRoomDevice.objects.filter(family_id=family_id, sub_device=0).\
                annotate(device_id=F('device__id'), type=F('device__Type'), nickname=F('device__NickName')).\
                values('device_id', 'type', 'nickname', 'room_id').order_by('sort')
            if not family_room_device_qs.exists():
                return response.json(0, [])
            # 查询房间名称
            for device in family_room_device_qs:
                room_id = device['room_id']
                if room_id == 0:
                    device['room'] = ''
                else:
                    family_room_qs = FamilyRoom.objects.filter(id=room_id).values('name')
                    device['room'] = family_room_qs[0]['name'] if family_room_qs.exists() else ''
                device.pop('room_id')
            return response.json(0, list(family_room_device_qs))
        except Exception as e:
            return response.json(500, repr(e))

    @staticmethod
    def devices_sort(request_dict, response):
        """
        家庭设备排序
        @param request_dict: 请求参数
        @request_dict 设备id: deviceIds
        @param response: 响应参数
        @return:
        """
        device_ids = request_dict.get('deviceIds', None)
        if not device_ids:
            return response.json(444)
        try:
            device_ids = device_ids.split(',')
            with transaction.atomic():
                for i, item in enumerate(device_ids):
                    device_id = int(item)
                    FamilyRoomDevice.objects.filter(device_id=device_id).update(sort=i)
            return response.json(0)
        except Exception as e:
            return response.json(500, repr(e))