ASJServer/Controller/UserDevice/UserDeviceShareController.py

# -*- encoding: utf-8 -*-
"""
@File    : UserDeviceShareController.py
@Time    : 2023/1/7 15:05
@Author  : stephen
@Email   : zhangdongming@asj6.wecom.work
@Software: PyCharm
"""
import json
import logging
import threading
import time

import boto3
import botocore
from botocore import client
from django.db import transaction
from django.db.models import Q
from django.views import View

from Model.models import DeviceSharePermission, DeviceChannelUserSet, DeviceChannelUserPermission, UidChannelSetModel, \
    Device_Info
from Object.ContentSecurityObject import ContentSecurity
from Object.ResponseObject import ResponseObject
from Object.TokenObject import TokenObject
from Service.UserDeviceService import UserDeviceService
from Ansjer.config import CONFIG_CN, CONFIG_INFO, CONFIG_TEST
from Model.models import DeviceWallpaper
from Object.AWS.AmazonS3Util import AmazonS3Util
from django.conf import settings
AWS_ACCESS_KEY_ID = settings.AWS_ACCESS_KEY_ID
AWS_SECRET_ACCESS_KEY = settings.AWS_SECRET_ACCESS_KEY
LOGGER = logging.getLogger('info')


class UserDeviceShareView(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 = TokenObject(request.META.get('HTTP_AUTHORIZATION'))
        lang = request_dict.get('lang', token.lang)
        response = ResponseObject(lang)
        if token.code != 0:
            return response.json(token.code)
        if operation == 'user-permissions':
            return self.get_user_share_permission(request_dict, response)
        elif operation == 'permissions-save':
            return self.user_channel_permission_save(request_dict, response)
        elif operation == 'permissions-test':
            return self.synch_share_device_permission(response)
        elif operation == 'getWallpaperList':
            return self.get_wallpaper_list(request_dict, response)
        elif operation == 'getUploadWallpaper':
            return self.get_upload_wallpaper(request_dict, response)
        elif operation == 'notifyUploadWallpaper':
            return self.notify_upload_wallpaper(request_dict, response)
        elif operation == 'delWallpaper':
            return self.del_wallpaper(request_dict, response)
        elif operation == 'selectWallpaper':
            return self.select_wallpaper(request_dict, response)
        else:
            return response.json(404)

    @classmethod
    def get_user_share_permission(cls, request_dict, response):
        """
        获取用户分享权限
        @param request_dict: 设备uid
        @param response: 响应对象
        @return: permission List
        """
        try:
            uid = request_dict.get('uid', None)
            user_id = request_dict.get('userId', None)
            channel_count = request_dict.get('channelCount', None)
            if not all([user_id, uid, channel_count]):
                return response(444, 'uid, userId and channelCount is required')
            user_permission_qs = DeviceChannelUserSet.objects.filter(user_id=user_id, uid=uid) \
                .values('id', 'channels')
            if not user_permission_qs.exists():
                return response.json(0, {})
            up_id = user_permission_qs[0]['id']
            channel_permission_qs = DeviceChannelUserPermission.objects.filter(channel_user_id=up_id) \
                .values('permission_id', 'channel_user_id')
            if not channel_permission_qs.exists():
                return response.json(0, {})
            channel_list = list(range(1, int(channel_count) + 1))
            share_channel_list = user_permission_qs[0]['channels'].split(',')
            c_list = []
            for channel in channel_list:
                is_select = 1 if str(channel) in share_channel_list else 0
                c_list.append({'channelIndex': channel, 'isSelect': is_select})
            p_list = []
            permission_qs = DeviceSharePermission.objects.all()
            share_permission_list = [item['permission_id'] for item in channel_permission_qs]
            for item in permission_qs:
                is_select = 1 if item.id in share_permission_list else 0
                p_list.append({'permissionId': item.id, 'code': item.code, 'isSelect': is_select})
            data = {'channels': c_list, 'permissions': p_list}
            return response.json(0, data)
        except Exception as e:
            LOGGER.info('异常详情,errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    @classmethod
    def user_channel_permission_save(cls, request_dict, response):
        """
        主用户分享设备时设置通道权限保存
        """
        try:
            uid = request_dict.get('uid', None)
            channels = request_dict.get('channels', None)  # 通道集合，多个','隔开
            user_id = request_dict.get('userId', None)
            permission_ids = request_dict.get('permissionIds', None)  # 权限集合，多个','隔开
            if not all([user_id, uid, channels, permission_ids]):
                return response.json(444)
            p_ids = []
            device_user_set = DeviceChannelUserSet.objects.filter(user_id=user_id, uid=uid)
            now_time = int(time.time())
            with transaction.atomic():
                is_delete = False
                if not device_user_set.exists():
                    device_set = {'uid': uid, 'user_id': user_id, 'channels': channels,
                                  'created_time': now_time, 'updated_time': now_time}
                    device_user_set = DeviceChannelUserSet.objects.create(**device_set)
                    channel_user_id = device_user_set.id
                else:
                    DeviceChannelUserSet.objects.update(channels=channels, updated_time=now_time)
                    channel_user_id = device_user_set.first().id
                    is_delete = True
                if ',' in permission_ids:
                    p_ids = [int(val) for val in permission_ids.split(',')]
                if is_delete:
                    DeviceChannelUserPermission.objects.filter(
                        channel_user_id=channel_user_id).delete()
                if not p_ids:
                    channel_permission = {'permission_id': int(permission_ids),
                                          'channel_user_id': channel_user_id,
                                          'created_time': now_time}
                    DeviceChannelUserPermission.objects.create(**channel_permission)
                else:
                    channel_permission_list = []
                    for item in p_ids:
                        channel_permission_list.append(DeviceChannelUserPermission(
                            permission_id=int(item),
                            channel_user_id=channel_user_id,
                            created_time=now_time))
                    DeviceChannelUserPermission.objects.bulk_create(channel_permission_list)
                return response.json(0)
        except Exception as e:
            LOGGER.info('异常详情,errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    @staticmethod
    def qrcode_share_channel_permission_save(user_id, uid):
        """
        二维码分享保存通道权限
        @param user_id: 用户id
        @param uid: 用户设备ID
        @return: True | False
        """
        try:
            if not all([user_id, uid]):
                return False
            with transaction.atomic():
                ds_qs = DeviceChannelUserSet.objects.filter(user_id=user_id, uid=uid)
                if ds_qs.exists():
                    return True
                UserDeviceService.update_device_channel(uid)
                channel_qs = UidChannelSetModel.objects.filter(uid__uid=uid).values('channel')
                if not channel_qs.exists():
                    return False
                channel_list = [str(val['channel']) for val in channel_qs]
                channel_str = ','.join(channel_list)
                now_time = int(time.time())
                device_set = {'uid': uid, 'user_id': user_id, 'channels': channel_str,
                              'created_time': now_time, 'updated_time': now_time}
                device_user_set = DeviceChannelUserSet.objects.create(**device_set)
                channel_permission_qs = DeviceSharePermission.objects \
                    .all().values('id', 'code').order_by('sort')
                user_set_id = device_user_set.id
                channel_permission_list = []
                for item in channel_permission_qs:
                    channel_permission_list.append(DeviceChannelUserPermission(
                        permission_id=item['id'],
                        channel_user_id=user_set_id,
                        created_time=now_time))
                DeviceChannelUserPermission.objects.bulk_create(channel_permission_list)
            return True
        except Exception as e:
            LOGGER.info('异常详情,errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
            return False

    @staticmethod
    def synch_share_device_permission(response):
        """
        同步分析设备权限
        @param response: 响应结果
        """
        device_info_qs = Device_Info.objects \
            .filter(~Q(Type__in=[1, 2, 3, 4, 10001]), ~Q(primaryUserID=''), isShare=1) \
            .values('userID_id', 'UID').order_by('-data_joined')
        if not device_info_qs.exists():
            return response.json(0)
        for item in device_info_qs:
            UserDeviceShareView.qrcode_share_channel_permission_save(item['userID_id'], item['UID'])
        return response.json(0)

    @classmethod
    def get_wallpaper_list(cls, request_dict, response):
        """
        获取设备壁纸列表
        @param request_dict:
        @param response:
        @return:
        """
        try:
            device_type = int(request_dict.get('deviceType', None))
            uid = request_dict.get('uid', None)
            channel = int(request_dict.get('channel', 1))
            LOGGER.info('获取设备壁纸列表参数:{}'.format(request_dict))
            if not all([device_type, uid]):
                return response.json(444)

            # 查询用户自定义壁纸
            user_wallpaper_qs = DeviceWallpaper.objects.filter(channel=channel, uid=uid,
                                                               device_type=device_type, parent_id=0, status=1)

            # 查询系统默认壁纸
            def_wallpaper_qs = DeviceWallpaper.objects.filter(classification=1, device_type=device_type, uid='',
                                                              status=1)

            # 查询用户选中的壁纸
            user_checked_qs = DeviceWallpaper.objects.filter(channel=channel, uid=uid,
                                                             device_type=device_type, parent_id__gt=0, status=1)
            checked_id = user_checked_qs[0].parent_id if user_checked_qs.exists() else 0

            wallpaper_list = []
            if def_wallpaper_qs.exists() or user_wallpaper_qs.exists():
                # 初始化存储桶客户端
                if CONFIG_CN == CONFIG_INFO or CONFIG_TEST == CONFIG_INFO:
                    s3 = boto3.client(
                        's3',
                        aws_access_key_id=AWS_ACCESS_KEY_ID[0],
                        aws_secret_access_key=AWS_SECRET_ACCESS_KEY[0],
                        config=botocore.client.Config(signature_version='s3v4'),
                        region_name='cn-northwest-1'
                    )
                else:
                    s3 = boto3.client(
                        's3',
                        aws_access_key_id=AWS_ACCESS_KEY_ID[1],
                        aws_secret_access_key=AWS_SECRET_ACCESS_KEY[1],
                        config=botocore.client.Config(signature_version='s3v4'),
                        region_name='us-east-1'
                    )

                # 处理系统默认壁纸和用户自定义壁纸
                all_wallpapers_qs = def_wallpaper_qs.union(user_wallpaper_qs)
                for item in all_wallpapers_qs:
                    obj_key = item.obj_prefix + item.obj_name
                    params = {
                        'Key': obj_key,
                        'Bucket': 'ansjerfilemanager',
                    }
                    response_url = s3.generate_presigned_url(
                        'get_object', Params=params, ExpiresIn=3600)
                    wallpaper = {
                        'id': item.id,
                        'url': response_url,
                        'state': 1 if checked_id == item.id else 0,
                        'classification': item.classification
                    }
                    wallpaper_list.append(wallpaper)

            return response.json(0, {'wallpapers': wallpaper_list})

        except Exception as e:
            LOGGER.error('查询设备壁纸异常:errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
            return response.json(5)

    @classmethod
    def get_upload_wallpaper(cls, request_dict, response):
        """
        获取设备壁纸上传链接
        @param request_dict:
        @param response:
        @return:
        """
        try:
            uid = request_dict.get('uid', None)
            channel = int(request_dict.get('channel', 1))

            if not uid:
                return response.json(444)

            # 初始化存储桶客户端
            if CONFIG_CN == CONFIG_INFO or CONFIG_TEST == CONFIG_INFO:
                s3 = boto3.client(
                    's3',
                    aws_access_key_id=AWS_ACCESS_KEY_ID[0],
                    aws_secret_access_key=AWS_SECRET_ACCESS_KEY[0],
                    config=botocore.client.Config(signature_version='s3v4'),
                    region_name='cn-northwest-1'
                )
            else:
                s3 = boto3.client(
                    's3',
                    aws_access_key_id=AWS_ACCESS_KEY_ID[1],
                    aws_secret_access_key=AWS_SECRET_ACCESS_KEY[1],
                    config=botocore.client.Config(signature_version='s3v4'),
                    region_name='us-east-1'
                )

            # 生成唯一的文件名
            file_name = f"{int(time.time())}.png"
            obj_key = f"app/static/device-wallpaper/{uid}/{file_name}"

            # 生成预签名的 URL
            presigned_url = s3.generate_presigned_url(
                'put_object',
                Params={
                    'Bucket': 'ansjerfilemanager',
                    'Key': obj_key,
                },
                ExpiresIn=600
            )

            # 返回预签名的URL和字段
            return response.json(0, {
                'uploadUrl': presigned_url,
                'fileName': file_name,
            })

        except Exception as e:
            LOGGER.error('获取上传链接异常:errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
            return response.json(5)

    @classmethod
    def notify_upload_wallpaper(cls, request_dict, response):
        """
        确认壁纸上传成功，创建DeviceWallpaper
        @param request_dict:
        @param response:
        @return:
        """
        try:
            uid = request_dict.get('uid', None)
            file_name = request_dict.get('fileName', None)
            channel = int(request_dict.get('channel', 1))

            if not all([uid, file_name]):
                return response.json(444)

            device_info_qs = Device_Info.objects.filter(UID=uid).values('id', 'Type')
            device_type = device_info_qs[0]['Type'] if device_info_qs.exists() else ''

            # 初始化存储桶客户端
            if CONFIG_CN == CONFIG_INFO or CONFIG_TEST == CONFIG_INFO:
                s3 = boto3.client(
                    's3',
                    aws_access_key_id=AWS_ACCESS_KEY_ID[0],
                    aws_secret_access_key=AWS_SECRET_ACCESS_KEY[0],
                    config=botocore.client.Config(signature_version='s3v4'),
                    region_name='cn-northwest-1'
                )
            else:
                s3 = boto3.client(
                    's3',
                    aws_access_key_id=AWS_ACCESS_KEY_ID[1],
                    aws_secret_access_key=AWS_SECRET_ACCESS_KEY[1],
                    config=botocore.client.Config(signature_version='s3v4'),
                    region_name='us-east-1'
                )

            obj_key = f"app/static/device-wallpaper/{uid}/{file_name}"
            params = {'Bucket': "ansjerfilemanager", 'Key': obj_key}
            image_url = s3.generate_presigned_url('get_object', Params=params)
            service = 'profilePhotoCheck'
            LOGGER.info('壁纸链接:{}'.format(image_url))
            service_dict = {'imageUrl': image_url}
            service_parameters = json.dumps(service_dict)
            legal = ContentSecurity().image_review(service, service_parameters)
            if not legal:
                # 异步删除
                thread = threading.Thread(target=cls.delete_wallpaper, args=(obj_key,))
                thread.start()
                return response.json(184)

            DeviceWallpaper.objects.create(
                device_type=device_type,
                storage_type=1,
                channel=channel,
                uid=uid,
                obj_prefix=f'app/static/device-wallpaper/{uid}/',
                obj_name=file_name,
                classification=2,  # 自定义壁纸分类
                status=1,
                parent_id=0,
                created_time=int(time.time())
            )
            return response.json(0)

        except Exception as e:
            LOGGER.error('壁纸创建失败:errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
            return response.json(5)

    @classmethod
    def del_wallpaper(cls, request_dict, response):
        """
        删除设备壁纸
        """
        wallpaper_id = request_dict.get('wallpaperId', None)
        if not all([wallpaper_id]):
            return response.json(444)
        try:
            wallpaper_id = int(wallpaper_id)
            if DeviceWallpaper.objects.filter(id=wallpaper_id, classification=1).exists():
                return response.json(176, "系统图片不能删除")
            device_wallpaper_qs = DeviceWallpaper.objects.filter(id=wallpaper_id)
            device_wallpaper = device_wallpaper_qs.values("obj_prefix", "obj_name", "uid", "parent_id")
            uid = device_wallpaper[0]['uid']
            obj_name = device_wallpaper[0]['obj_name']

            # 使用中壁纸被删除
            use_device_wallpaper = DeviceWallpaper.objects.filter(uid=uid, parent_id__gt=0).first()
            if use_device_wallpaper and use_device_wallpaper.parent_id == wallpaper_id:
                system_device_wallpaper = DeviceWallpaper.objects.filter(id=1).first()
                use_device_wallpaper.parent_id = 1
                use_device_wallpaper.storage_type = system_device_wallpaper.storage_type
                use_device_wallpaper.obj_name = system_device_wallpaper.obj_name
                use_device_wallpaper.obj_prefix = system_device_wallpaper.obj_prefix
                use_device_wallpaper.classification = system_device_wallpaper.classification
                use_device_wallpaper.channel = system_device_wallpaper.channel
                use_device_wallpaper.save()

            if CONFIG_CN == CONFIG_INFO or CONFIG_TEST == CONFIG_INFO:
                s3 = AmazonS3Util(
                    AWS_ACCESS_KEY_ID[0], AWS_SECRET_ACCESS_KEY[0], 'cn-northwest-1'
                )
            else:
                s3 = AmazonS3Util(
                    AWS_ACCESS_KEY_ID[1], AWS_SECRET_ACCESS_KEY[1], 'us-east-1'
                )

            s3.delete_obj("ansjerfilemanager", f"app/static/device-wallpaper/{uid}/{obj_name}")
            device_wallpaper_qs.delete()

            return response.json(0)

        except Exception as e:
            LOGGER.info('删除壁纸异常:{}'.format(e))
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    @classmethod
    def select_wallpaper(cls, request_dict, response):
        """
        用户选取壁纸
        @param request_dict:
        @param response:
        @return:
        """
        try:
            wallpaper_id = request_dict.get('wallpaperId', None)
            uid = request_dict.get('uid', None)
            if not all([wallpaper_id, uid]):
                return response.json(444)
            device_wallpaper = DeviceWallpaper.objects.filter(id=wallpaper_id).values("device_type", "storage_type",
                                                                                      "obj_prefix",
                                                                                      "obj_name",
                                                                                      "classification",
                                                                                      "channel").first()
            device_wallpaper_qs = DeviceWallpaper.objects.filter(uid=uid, parent_id__gt=0)
            if device_wallpaper_qs.exists():
                device_wallpaper_qs.update(
                    storage_type=device_wallpaper["storage_type"],
                    classification=device_wallpaper["classification"],
                    obj_prefix=device_wallpaper["obj_prefix"],
                    obj_name=device_wallpaper["obj_name"],
                    parent_id=wallpaper_id
                )
            else:
                DeviceWallpaper.objects.create(
                    device_type=device_wallpaper["device_type"],
                    storage_type=device_wallpaper["storage_type"],
                    obj_name=device_wallpaper["obj_name"],
                    obj_prefix=device_wallpaper["obj_prefix"],
                    classification=device_wallpaper["classification"],
                    uid=uid,
                    channel=device_wallpaper["channel"],
                    status=1,
                    parent_id=wallpaper_id,
                    created_time=int(time.time())
                )
            return response.json(0)
        except Exception as e:
            LOGGER.error('用户选取壁纸异常:errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
            return response.json(5)

    @staticmethod
    def delete_wallpaper(file_key):
        try:
            if CONFIG_CN == CONFIG_INFO or CONFIG_TEST == CONFIG_INFO:
                s3 = AmazonS3Util(
                    AWS_ACCESS_KEY_ID[0], AWS_SECRET_ACCESS_KEY[0], 'cn-northwest-1'
                )
            else:
                s3 = AmazonS3Util(
                    AWS_ACCESS_KEY_ID[1], AWS_SECRET_ACCESS_KEY[1], 'us-east-1'
                )

            s3.delete_obj("ansjerfilemanager", file_key)
            return True
        except Exception as e:
            LOGGER.info('删除违规壁纸异常:errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
            return False