ASJServer/Controller/CloudVod.py

#!/usr/bin/env python3  
# -*- coding: utf-8 -*-  
"""
@Copyright (C) ansjer cop Video Technology Co.,Ltd.All rights reserved.
@AUTHOR: ASJRD018
@NAME: AnsjerFormal
@software: PyCharm
@DATE: 2018/12/5 9:30
@Version: python3.6
@MODIFY DECORD:ansjer dev
@file: CloudVod.py
@Contact: chanjunkai@163.com
"""
import datetime
import json
import time

import oss2
import paypalrestsdk
from aliyunsdkcore import client
from aliyunsdksts.request.v20150401 import AssumeRoleRequest
from django.http import JsonResponse
from django.utils import timezone
from django.utils.decorators import method_decorator
from django.views.decorators.csrf import csrf_exempt
from django.views.generic.base import View
import urllib

from Ansjer.config import OSS_BUCKET_ENDPOINT, OSS_BUCKET_VOD_LIST, OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET, \
    OSS_REGION_ID, OSS_ROLE_ARN, SERVER_DOMAIN
from Model.models import Device_Meal, Device_Info, Order_Model, Store_Meal
from Object.ResponseObject import ResponseObject
from Object.TokenObject import TokenObject
from Object.UidTokenObject import UidTokenObject
from Service.CommonService import CommonService

'''
# 获取设备推送hls流 证书
http://192.168.136.40:8077/cloudVod/getSts?uidToken=eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ1aWQiOiJGVFNMTDhITTQzN1ozOFdVMTExQSJ9.GIBt8SgY-3yRt9mlihtvRwLM-MT8uVPDKCUQ2yvV3Vo
# 获取存储的播放文件列表
'''


# 设备信息添加
class CloudVodView(View):
    @method_decorator(csrf_exempt)
    def dispatch(self, *args, **kwargs):
        return super(CloudVodView, self).dispatch(*args, **kwargs)

    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):
        response = ResponseObject()
        if operation is None:
            return response.json(444, 'error path')
        if operation == 'getSts':
            # 移动侦测获取设备sts证书
            ip = CommonService.get_ip_address(request)
            return self.do_getSts(request_dict, ip, response)
        elif operation == 'payExecute':
            return self.do_paypal_execute(request_dict, response)
        else:
            token = request_dict.get('token', None)
            # 设备主键uid
            tko = TokenObject(token)
            tko.valid()
            response.lang = tko.lang
            if tko.code != 0:
                return response.json(tko.code)
            userID = tko.userID
            if operation == 'getHlsList':
                return self.do_get_hls_list(request_dict, userID, response)
            elif operation == 'createOrder':
                return self.do_create_order(request_dict, userID, response)
            elif operation == 'status':
                return self.do_change_status(request_dict, userID, response)
            else:
                return response.json(414)

    def do_create_order(self, request_dict, userID, response):
        did = request_dict.get('did', None)
        rank = request_dict.get('rank', None)
        channel = request_dict.get('channel', None)
        if not did or not channel or not rank:
            return response.json(444, 'did,channel,rank')
        qs = Device_Info.objects.filter(userID_id=userID, id=did).values("UID")
        if not qs.exists():
            return response.json(13)
        uid = qs[0]['UID']
        nowTime = int(time.time())
        orderID = CommonService.createID()
        if rank == '1':
            price = 30
        elif rank == '2':
            price = 360
        # call_sub_url = "{SERVER_DOMAIN}cloudVod/payExecute?orderID={orderID}".format(SERVER_DOMAIN=SERVER_DOMAIN,orderID=orderID)
        call_sub_url = "http://192.168.136.40:8077/cloudVod/payExecute?orderID={orderID}".format(
            SERVER_DOMAIN=SERVER_DOMAIN, orderID=orderID)
        call_clc_url = "http://192.168.136.40:8077/cloudVod/order/cancle"
        paypalrestsdk.configure({
            "mode": "sandbox",  # sandbox or live
            "client_id": "AfnfDqezODOoWGS-W2Itu-Zl1ay1R95IsGlMqPghPA3KGhkPndNMnQT0bdEewvSv92XAFIfLiinmyhBL",
            "client_secret": "EErLskwYA1xXY3890mHx5OhzgK83B2rNc57zIozGNyKc8i6RJuhPTF9WyhhdZgyDEih0heo1MH9Jk1lj"
        })
        payment = paypalrestsdk.Payment({
            "intent": "sale",
            "payer": {"payment_method": "paypal"},
            "redirect_urls": {"return_url": call_sub_url, "cancel_url": call_clc_url},
            "transactions": [{
                "item_list": {
                    "items": [{"name": "Cloud video", "sku": "1", "price": price, "currency": "USD", "quantity": 1}]},
                "amount": {"total": price, "currency": "USD"},
                "description": "7 days mobile detection storage event 1 month charge"
            }]})

        if payment.create():
            print("Payment created successfully")

        else:
            print(payment.error)
            return response.json(10, payment.error)
        print(payment)
        for link in payment.links:
            if link.rel == "approval_url":
                approval_url = str(link.href)
                print("Redirect for approval: %s" % (approval_url))
                cqs = Order_Model.objects.create(
                    orderID=orderID,
                    UID=uid,
                    channel=channel,
                    userID_id=userID,
                    desc='',
                    price='',
                    addTime=nowTime,
                    updTime=nowTime,
                    rank_id=rank,
                    paypal=approval_url
                )
                return response.json(0, {"redirectUrl": approval_url})
                # return HttpResponseRedirect(approval_url)
        return response.json(10, 'generate_order_false')

    def do_getSts(self, request_dict, ip, response):
        '''
        生成sts上传授权
        '''
        # uidToken = 'eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ1aWQiOiJGVFNMTDhITTQzN1ozOFdVMTExQSJ9.GIBt8SgY-3yRt9mlihtvRwLM-MT8uVPDKCUQ2yvV3Vo'
        uidToken = request_dict.get('uidToken', None)
        utko = UidTokenObject(uidToken)
        if utko.flag is False:
            return response.json(444, 'uidToken')
        UID = utko.UID
        qs = Device_Meal.objects.filter(uid=UID, status=1).values("rank", "channel")
        if not qs.exists():
            return response.json(10, '设备未购买')
        # 套餐id
        rank = qs[0]['rank']
        channel = qs[0]['channel']
        storage = '{uid}/vod{channel}/'.format(uid=UID, channel=channel)

        # bucket_vod_list = {1: 'cloudvod1', 2: 'cloudvod2', 3: 'cloudvod3'}
        # bucket_name = bucket_vod_list[rank]
        # endpoint = 'oss-cn-shenzhen.aliyuncs.com'
        # access_key_id = 'LTAIyMkGfEdogyL9'
        # access_key_secret = '71uIjpsqVOmF7DAITRyRuc259jHOjO'
        # region_id = 'cn-shenzhen'
        # role_arn = 'acs:ram::1901342792446414:role/stsoss'
        bucket_vod_list = OSS_BUCKET_VOD_LIST
        bucket_name = bucket_vod_list[rank]
        endpoint = OSS_BUCKET_ENDPOINT
        access_key_id = OSS_STS_ACCESS_KEY
        access_key_secret = OSS_STS_ACCESS_SECRET
        region_id = OSS_REGION_ID
        role_arn = OSS_ROLE_ARN
        clt = client.AcsClient(access_key_id, access_key_secret, region_id)
        req = AssumeRoleRequest.AssumeRoleRequest()
        # 设置返回值格式为JSON。
        req.set_accept_format('json')
        req.set_RoleArn(role_arn)
        req.set_RoleSessionName(UID)
        req.set_DurationSeconds(3600)
        Resource_access = "acs:oss:*:*:{bucket_name}/{uid_channel}*".format(bucket_name=bucket_name,
                                                                            uid_channel=storage)
        print(Resource_access)
        policys = {
            "Version": "1",
            "Statement": [
                {
                    "Action": [
                        "oss:PutObject",
                        "oss:DeleteObject",
                    ],
                    "Resource": [Resource_access],
                    "Effect": "Allow",
                    "Condition": {
                        # "IpAddress": {"acs:SourceIp": ip}
                        # "IpAddress": {"acs:SourceIp": "120.237.157.184"}
                        "IpAddress": {"acs:SourceIp": "*"}
                    }
                }
            ]
        }
        req.set_Policy(Policy=json.dumps(policys))
        body = clt.do_action(req)
        # 使用RAM账号的AccessKeyId和AccessKeySecret向STS申请临时token。
        token = json.loads(body)
        print(token)
        res = {
            'AccessKeyId': token['Credentials']['AccessKeyId'],
            'AccessKeySecret': token['Credentials']['AccessKeySecret'],
            'SecurityToken': token['Credentials']['SecurityToken'],
            'Expiration': token['Credentials']['Expiration'],
            'expire': '3600',
            'endpoint': endpoint,
            'bucket_name': bucket_name,
            'arn': token['AssumedRoleUser']['Arn'],
            'code': 0,
            'storage': storage,
            'ip': ip
        }
        return JsonResponse(status=200, data=res)

    # 获取设备回放列表
    def do_get_hls_list(self, request_dict, userID, response):
        did = request_dict.get('did', None)
        channel = request_dict.get('channel', None)
        # 目录开始第一个
        # marker = request_dict.get('marker', '')
        daytime = request_dict.get('daytime', None)
        hour = request_dict.get('hour', None)
        if not did or not channel or not daytime:
            return response.json(444, 'did,channel')
        qs = Device_Info.objects.filter(userID_id=userID, id=did).values("UID")
        if not qs.exists():
            return response.json(13)
        uid = qs[0]['UID']
        dm_qs = Device_Meal.objects.filter(uid=uid, channel=channel).values('rank')
        if not dm_qs.exists():
            return response.json(10, '未开通云存储')
        rank = dm_qs[0]["rank"]
        bucket_name = OSS_BUCKET_VOD_LIST[rank]
        auth = oss2.Auth(OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET)
        bucket = oss2.Bucket(auth, OSS_BUCKET_ENDPOINT, bucket_name)
        vod_play_list = []
        prefix = '{uid}/vod{channel}/{daytime}/'.format(uid=uid, channel=channel, daytime=daytime, hour=hour)

        for obj in oss2.ObjectIterator(bucket=bucket, prefix=prefix, delimiter='/', max_keys=2):
            # 通过is_prefix方法判断obj是否为文件夹。
            if obj.is_prefix():  # 文件夹
                print('directory: ' + obj.key)
                ptime = obj.key.split('/')[3]
                url = bucket.sign_url('GET', '{prefix}{name}.m3u8'.format(prefix=obj.key, name=ptime), 60 * 60,params={'x-oss-process': 'hls/sign'})
                # url = bucket.sign_url('GET', obj.key, 60 * 60,params={'x-oss-process': 'hls/sign'})
                urllst = url.split('?')
                url_start = urllib.parse.unquote(urllst[0])
                url_end = urllst[1]
                vod_play_url = '{url_start}?{url_end}'.format(url_start=url_start, url_end=url_end)
                vod_play_list.append({
                    'name': ptime,
                    'sign_url': vod_play_url
                })
            else:  # 文件
                pass
                print('file: ' + obj.key)
        return response.json(0, vod_play_list)

    def do_paypal_execute(self, request_dict, response):
        paymentId = request_dict.get('paymentId', None)
        PayerID = request_dict.get('PayerID', None)
        orderID = request_dict.get('orderID', None)
        if not paymentId or not PayerID or not orderID:
            return response.json(444, 'paymentId,PayerID,orderID')
        paypalrestsdk.configure({
            "mode": "sandbox",  # sandbox or live
            "client_id": "AfnfDqezODOoWGS-W2Itu-Zl1ay1R95IsGlMqPghPA3KGhkPndNMnQT0bdEewvSv92XAFIfLiinmyhBL",
            "client_secret": "EErLskwYA1xXY3890mHx5OhzgK83B2rNc57zIozGNyKc8i6RJuhPTF9WyhhdZgyDEih0heo1MH9Jk1lj"
        })
        # ID of the payment. This ID is provided when creating payment.
        payment = paypalrestsdk.Payment.find(paymentId)
        payres = payment.execute({"payer_id": PayerID})
        print(payres)
        if not payres:
            print(payment.error)  # Error Hash
            return response.json(10, payment.error)
        print("Payment execute successfully")
        order_qs = Order_Model.objects.filter(orderID=orderID)
        order_qs.update(status=1, updTime=int(time.time()))
        order_list = order_qs.values("UID", "rank_id", "channel")
        rank_id = order_list[0]['rank_id']
        UID = order_list[0]['UID']
        channel = order_list[0]['channel']
        smqs = Store_Meal.objects.filter(id=rank_id).values("day")
        if not smqs.exists():
            return response.json(0, '套餐已删除')
        add_days = smqs[0]["day"]
        dmqs = Device_Meal.objects.filter(channel=channel, uid=UID).values("end_time")
        # 续费模式
        if dmqs.exists():
            end_time = dmqs[0]["end_time"]
            now_time = timezone.localtime(timezone.now())
            if now_time > end_time:
                end_time = now_time + datetime.timedelta(days=int(add_days))
            else:
                end_time = end_time + datetime.timedelta(days=int(add_days))
            dmqs.update(end_time=end_time, update_time=now_time)
            return response.json(0)
        # 新增模式
        else:
            end_time = datetime.timedelta(days=add_days) + datetime.datetime.now()
            Device_Meal.objects.create(rank_id=rank_id, uid=UID, channel=channel, status=1, end_time=end_time)
            return response.json(0)

    def do_change_status(self, request_dict, userID, response):
        did = request_dict.get('did', None)
        status = request_dict.get('status', None)
        channel = request_dict.get('channel', None)
        if not did or not status or not channel:
            return response.json(444, 'did,status,channel')
        dvqs = Device_Info.objects.filter(id=did, userID_id=userID).values("UID")
        if not dvqs.exists():
            return response.json(10, '没有设备')
        dmqs = Device_Meal.objects.filter(channel=channel, uid=dvqs[0]["UID"])
        if not dmqs.exists():
            return response.json(10, '未购买')
        now_time = timezone.localtime(timezone.now())
        dmqs.update(update_time=now_time, status=status)
        return response.json(0)