#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ @Copyright (C) ansjer cop Video Technology Co.,Ltd.All rights reserved. @AUTHOR: ASJRD018 @NAME: Ansjer @software: PyCharm @DATE: 2018/5/30 17:53 @Version: python3.6 @MODIFY DECORD:ansjer dev @file: HlsManage.py @Contact: chanjunkai@163.com """ from django.views.generic.base import View from django.utils.decorators import method_decorator from django.views.decorators.csrf import csrf_exempt from Service.ModelService import ModelService from Model.models import Device_Meal from Object.AWS.S3ClassObject import S3ClassObject from Object.AWS.CloudfrontSignUrl import CloudfrontSignUrl from Object.encryHlsVodUrl import getSignUrl from Object.m3u8generate import PlaylistGenerator from Object.TokenObject import TokenObject from Object.ResponseObject import ResponseObject from django.http import HttpResponse ''' 设备端 http://13.56.215.252:82/HlsManager?operation=signPutObj&key=555666.mp4&uid=L59KVYDAEPHR1T6M111A&channel=0 http://13.56.215.252:82/HlsManager?operation=signPostObj&key=ppp.mp4&uid=L59KVYDAEPHR1T6M111A&channel=0 移动端 http://13.56.215.252:82/HlsManager?operation=getHlsVod&uid=L59KVYDAEPHR1T6M111A&channel=0&token=test&key=234234234.ts http://13.56.215.252:82/HlsManager?operation=getAllTs&uid=L59KVYDAEPHR1T6M111A&channel=0&token=test http://13.56.215.252:82/HlsManager?operation=getVodUrl&uid=L59KVYDAEPHR1T6M111A&channel=0&token=test&key=121212.mp4 ''' class HlsManage(View): @method_decorator(csrf_exempt) def dispatch(self, *args, **kwargs): return super(HlsManage, self).dispatch(*args, **kwargs) def get(self, request, *args, **kwargs): request.encoding = 'utf-8' return self.validation(request_dict=request.GET) def post(self, request, *args, **kwargs): request.encoding = 'utf-8' return self.validation(request_dict=request.POST) def validation(self, request_dict, *args, **kwargs): response = ResponseObject() operation = request_dict.get('operation', None) if operation == 'signPutObj': return self.sign_put_obj(request_dict,response) elif operation == 'signPostObj': return self.sign_post_obj(request_dict,response) elif operation == 'getHlsVod': return self.get_hls_vod(request_dict,response) elif operation == 'getAllTs': return self.get_all_ts(request_dict,response) elif operation == 'getVodUrl': return self.get_vod_url(request_dict,response) else: return response.json(444, 'operation') def sign_put_obj(self, request_dict,response): key = request_dict.get('key', None) uid = request_dict.get('uid', None) channel = request_dict.get('channel', None) if key is not None and uid is not None and channel is not None: device_meal_queryset = Device_Meal.objects.filter(uid=uid, channel=channel, status=1) if device_meal_queryset.exists(): path = uid + '_' + channel + '/' + key meal_id = device_meal_queryset[0].rank_id bucket_meal={ 1:'ansjer.meal.1',# 7 2:'ansjer.meal.2',# 30 # 3:'ansjer.meal.3', } s3 = S3ClassObject() sign_url = s3.sign_put_object(key=path, bucket_meal=bucket_meal[meal_id]) if sign_url is not False: return (0, {'url': sign_url}) else: return response.json(48) else: return response.json(444) else: return response.json(444,'key, uid, channel') def sign_post_obj(self, request_dict,response): key = request_dict.get('key', None) uid = request_dict.get('uid', None) channel = request_dict.get('channel', None) if key is not None and uid is not None and channel is not None: device_meal_queryset = Device_Meal.objects.filter(uid=uid, channel=channel, status=1) if device_meal_queryset.exists(): path = uid + '_' + channel + '/' + key meal_id = device_meal_queryset[0].rank_id bucket_meal = { 1: 'ansjer.meal.1', # 7 2: 'ansjer.meal.2', # 30 # 3:'ansjer.meal.3', } s3 = S3ClassObject() data = s3.sign_post_object(key=path, bucket_meal=bucket_meal[meal_id]) if data is not False: return response.json(0, {'url': data}) else: return response.json(48) else: return response.json(444) else: return response.json(444,'key, uid, channel') def get_all_ts(self, request_dict,response): token = request_dict.get('token', None) if token is not None: tko = TokenObject(token) tko.valid() response.lang = tko.lang if tko.code == 0: userID = tko.userID() if userID is not None: uid = request_dict.get('uid', None) channel = request_dict.get('channel', None) if uid is not None and channel is not None: own_permission = ModelService.check_permission(userID=userID, permID=30) own_device = ModelService.check_own_device(userID=userID, UID=uid) if own_permission is True or own_device is True: device_meal_queryset = Device_Meal.objects.filter(uid=uid, channel=channel, status=1) if device_meal_queryset.exists(): meal_id = device_meal_queryset[0].rank_id bucket_meal = { 1: 'ansjer.meal.1', # 7 2: 'ansjer.meal.2', # 30 } s3 = S3ClassObject() data = s3.get_prefix_obj(prefix=uid + '_' + channel + '/', bucket=bucket_meal[meal_id]) # return response.json(0, {'files': data,'bk':bucket_meal[meal_id]}) return response.json(0, {'files': data}) else: return response.json(444) else: return response.json(404) else: return response.json(444,'uid,channel') else: return response.json(310) else: return response.json(tko.code) else: return response.json(311) def get_hls_vod(self, request_dict,response): token = request_dict.get('token', None) if token is not None: tko = TokenObject(token) tko.valid() response.lang = tko.lang if tko.code == 0: userID = tko.userID() if userID is not None: key = request_dict.get('key', None) uid = request_dict.get('uid', None) channel = request_dict.get('channel', None) own_device = ModelService.check_own_device(userID=userID, UID=uid) if own_device is True: device_meal_queryset = Device_Meal.objects.filter(uid=uid, channel=channel, status=1) if device_meal_queryset.exists(): meal_id = device_meal_queryset[0].rank_id cf_domain = { 1: 'http://d3om3d8c0l14oc.cloudfront.net/', 2: 'http://d4neaolnu8h83.cloudfront.net/', } url = cf_domain[meal_id] + uid + '_' + channel + '/' + key sign_url = CloudfrontSignUrl(url=url) playlist_entries = [ { 'name': sign_url, 'duration': '60', } ] playlist = PlaylistGenerator(playlist_entries).generate() response = HttpResponse(content_type='application/force-download', content=playlist) response['Content-Disposition'] = 'attachment; filename=' + key + '.m3u8' # response['X-Sendfile'] = playlist # return response.json(404) return response # return HttpResponse(playlist) else: return response.json(0) else: return response.json(15) else: return response.json(310) else: return response.json(tko.code) else: return response.json(311) # mp4 def get_vod_url(self, request_dict,response): token = request_dict.get('token', None) if token is not None: tko = TokenObject(token) tko.valid() response.lang = tko.lang if tko.code == 0: userID = tko.userID() key = request_dict.get('key', None) uid = request_dict.get('uid', None) channel = request_dict.get('channel', None) if userID is not None: own_device = ModelService.check_own_device(userID=userID, UID=uid) if own_device is True: device_meal_queryset = Device_Meal.objects.filter(uid=uid, channel=channel, status=1) if device_meal_queryset.exists(): meal_id = device_meal_queryset[0].rank_id bucket_meal = { 1: 'ansjer.meal.1', # 7 2: 'ansjer.meal.2', # 30 } bk = bucket_meal[meal_id] path = '/'+bk+'/'+uid+'_'+channel+'/'+key signUrl = getSignUrl(path=path) return response.json(0,{'signUrl':signUrl}) else: return response.json(0) else: return response.json(15) else: return response.json(310) else: return response.json(tko.code) else: return response.json(311)