ASJServer/Controller/IcloudService/IcloudService.py

# -*- coding: utf-8 -*-
"""
@Author ： peng
@Time ： 2023-6-7 18:26:35
@File ：IcloudMeal.py
"""
import time

from django.db.models import Sum, Q
from django.http import HttpResponse
from django.views import View

from Model.models import IcloudUseDetails, IcloudService, VodBucketModel, IcloudStorageRecord, Device_Info
from Object.AWS.AmazonS3Util import AmazonS3Util
from Object.RedisObject import RedisObject
from Object.ResponseObject import ResponseObject
from Object.TokenObject import TokenObject
from Ansjer.config import ACCESS_KEY_ID, SECRET_ACCESS_KEY, REGION_NAME, SERVER_DOMAIN, AWS_ACCESS_KEY_ID, \
    AWS_SECRET_ACCESS_KEY
from Service.VodHlsService import SplitVodHlsObject


class IcloudServiceView(View):
    def get(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        operation = kwargs.get('operation')
        return self.validation(request.GET, operation, request)

    def post(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        operation = kwargs.get('operation')
        return self.validation(request.POST, operation, request)

    def validation(self, request_dict, operation, request):
        response = ResponseObject()
        tko = TokenObject(request.META.get('HTTP_AUTHORIZATION'))
        response.lang = tko.lang
        user_id = tko.userID
        if tko.code != 0:
            return response.json(tko.code)
        if operation == 'vodUpload':  # 云存上传云盘
            return self.vod_upload(request_dict, response, user_id)
        elif operation == 'get-scanning-status':  # 确认app是否扫码
            return self.get_scanning_status(request_dict, response)
        else:
            return response.json(404)

    @staticmethod
    def vod_upload(request_dict, response, user_id):
        """
        云存上传云盘
        @param request_dict: 请求参数
        @param user_id: 用户id
        @request_dict serial_number: 序列号
        @param response: 响应对象
        @return: response
        """
        uid = request_dict.get('uid', None)
        channel = request_dict.get('channel', None)
        start_time = request_dict.get('start_time', None)
        if not all([uid, channel, start_time]):
            return response.json(444, {'error param': 'uid,channel,start_time'})
        device_qs = Device_Info.objects.filter(UID=uid, userID=user_id).values('NickName')
        if not device_qs.exists():
            return response.json(173)
        # 查询是否开通云盘
        use_details_qs = IcloudUseDetails.objects.filter(user_id=user_id).values('id', 'bucket_id', 'use_size')
        if not use_details_qs.exists():
            return response.json(173)
        use_details_id = use_details_qs[0]['id']
        bucket_id = use_details_qs[0]['bucket_id']
        use_size = use_details_qs[0]['use_size']
        now_time = int(time.time())
        nickname = device_qs[0]['NickName']
        # 判断云盘是否还有容量
        all_size = IcloudService.objects.filter(Q(use_details_id=use_details_id), Q(use_status=0),
                                                Q(end_time__gt=now_time) | Q(end_time=0)).values(
            'size').aggregate(total_size=Sum('count'))['total_size']
        all_size = all_size if all_size else 0
        if use_size > all_size:
            return response.json(910)
        split_vod_hls_obj = SplitVodHlsObject()
        vod_hls = split_vod_hls_obj.get_vod_hls_data(uid=uid, channel=channel, start_time=start_time).values(
            'bucket_id', 'fg', 'sec')
        if not vod_hls.exists():
            return response.json(173)
        source_bucket_id = vod_hls[0]['bucket_id']
        fg = vod_hls[0]['fg']
        sec = vod_hls[0]['sec']
        bucket_qs = VodBucketModel.objects.filter(id=bucket_id).values('bucket')
        if not bucket_qs.exists():
            return response.json(173)
        to_bucket_name = bucket_qs[0]['bucket']

        source_bucket_qs = VodBucketModel.objects.filter(id=source_bucket_id).values('bucket', 'region', 'mold')
        if not source_bucket_qs.exists():
            return response.json(173)
        bucket_region = source_bucket_qs[0]['region']
        source_bucket_name = source_bucket_qs[0]['bucket']
        mold = source_bucket_qs[0]["mold"]
        ts_list = []
        ts_size = 0
        try:
            s3_obj = AmazonS3Util(
                AWS_ACCESS_KEY_ID[mold],
                AWS_SECRET_ACCESS_KEY[mold],
                bucket_region
            )
            for i in range(15):
                shift = (i + 1) * 4
                duration = (fg >> shift) & 0xf
                if duration > 0:
                    ts_file = '{uid}/vod{channel}/{time}/ts{i}.ts'.format(uid=uid, channel=channel, time=start_time,
                                                                          i=i)
                    ts_list.append(ts_file)
                    ts_size += s3_obj.get_object_size(source_bucket_name, ts_file)
            temp_size = ts_size + use_size
            if temp_size > all_size:
                return response.json(910)
            for item in ts_list:
                key_name = '{user_id}/ts_file/'.format(user_id=user_id) + item
                s3_obj.copy_single_obj(source_bucket_name, item, to_bucket_name, key_name)
            IcloudStorageRecord.objects.create(user_id=user_id, uid=uid, channel=channel, time_stamp=start_time,
                                               nickname=nickname,
                                               sec=sec, bucket_id=bucket_id, fg=fg, size=ts_size, store_type=1)
            use_details_qs.update(size=temp_size)
            return response.json(0)
        except Exception as e:
            print(e)
            return response.json(500)