TestApi.py 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674
  1. #!/usr/bin/env python3
  2. # -*- coding: utf-8 -*-
  3. """
  4. @Copyright (C) ansjer cop Video Technology Co.,Ltd.All rights reserved.
  5. @AUTHOR: ASJRD018
  6. @NAME: Ansjer
  7. @software: PyCharm
  8. @DATE: 2018/5/22 13:58
  9. @Version: python3.6
  10. @MODIFY DECORD:ansjer dev
  11. @file: Test.py
  12. @Contact: chanjunkai@163.com
  13. """
  14. from django.views.generic.base import View
  15. import os
  16. '''
  17. http://192.168.136.40:8077/Test
  18. '''
  19. import json
  20. import math
  21. import time
  22. import urllib
  23. from Object.AliPayObject import AliPayObject
  24. import logging
  25. import boto3
  26. from boto3.session import Session
  27. from botocore.exceptions import ClientError
  28. import oss2
  29. import paypalrestsdk
  30. import logging
  31. from aliyunsdkcore import client
  32. from aliyunsdksts.request.v20150401 import AssumeRoleRequest
  33. from django.http import JsonResponse, HttpResponseRedirect, HttpResponse
  34. from django.utils.decorators import method_decorator
  35. from django.views.decorators.csrf import csrf_exempt
  36. from django.views.generic.base import View
  37. from Ansjer.config import OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET, OSS_ROLE_ARN, SERVER_DOMAIN, PAYPAL_CRD, \
  38. SERVER_DOMAIN_SSL
  39. from Model.models import Device_Info, Order_Model, Store_Meal, VodHlsModel, OssCrdModel, UID_Bucket
  40. from Object.ResponseObject import ResponseObject
  41. from Object.TokenObject import TokenObject
  42. from Object.UidTokenObject import UidTokenObject
  43. from Service.CommonService import CommonService
  44. from Service.ModelService import ModelService
  45. from Object.m3u8generate import PlaylistGenerator
  46. from Model.models import Device_User, Device_Info, UID_Bucket, UID_Preview, UidSetModel, UidPushModel
  47. SERVER_DOMAIN = 'http://test.dvema.com/'
  48. ACCESS_KEY = "AKIA2E67UIMD3CYTIWPA"
  49. SECRET_KEY = "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"
  50. # 测试接口sdk
  51. class testView(View):
  52. @method_decorator(csrf_exempt)
  53. def dispatch(self, *args, **kwargs):
  54. return super(testView, self).dispatch(*args, **kwargs)
  55. def get(self, request, *args, **kwargs):
  56. request.encoding = 'utf-8'
  57. operation = kwargs.get('operation')
  58. return self.validation(request.GET, request, operation)
  59. def post(self, request, *args, **kwargs):
  60. request.encoding = 'utf-8'
  61. operation = kwargs.get('operation')
  62. return self.validation(request.POST, request, operation)
  63. def validation(self, request_dict, request, operation):
  64. response = ResponseObject()
  65. # operation => cloudVod/path
  66. if operation is None:
  67. return response.json(444, 'error path')
  68. elif operation == 'tests':
  69. userID = '158943594633713800138000'
  70. return self.do_filter_playlist(request_dict, userID, response)
  71. elif operation == 'cbu':
  72. return self.createBucket()
  73. elif operation == 'vodList':
  74. ip = CommonService.get_ip_address(request)
  75. userID = '158943594633713800138000'
  76. return self.do_test_query_vod_list(userID, ip, request_dict, response)
  77. elif operation == 'signplaym3u8':
  78. return self.do_sign_play_m3u8(request_dict, response)
  79. elif operation == 'get_sign_sts':
  80. ip = CommonService.get_ip_address(request)
  81. return self.do_test_get_sign_sts(request_dict, ip, response)
  82. elif operation == 'eqi_query':
  83. userID = '158943594633713800138000'
  84. return self.queryInterface(request_dict, userID, response)
  85. else:
  86. return 12344444
  87. def createBucket(self):
  88. #查看桶列表
  89. # url = "https://azvod1.s3-ap-northeast-1.amazonaws.com"
  90. # session = Session(ACCESS_KEY, SECRET_KEY)
  91. # s3_client = session.client('s3', endpoint_url=url)
  92. # results = s3_client.list_buckets()
  93. # return JsonResponse(status=200, data={'code': 200, 'msg': results['Buckets']})
  94. # Create bucket
  95. session = Session(ACCESS_KEY, SECRET_KEY)
  96. s3_client = session.client('s3')
  97. #上传
  98. # s3_client.put_object(Bucket="azvod1", Key="file/rule.txt", Body=open(r"E:\download\Shadowsocks-4.1.10.0\user-rule.txt", 'rb').read())
  99. #下载
  100. resp = s3_client.get_object(Bucket="azvod1", Key="file/rule.txt")
  101. with open('local.txt', 'wb') as f:
  102. f.write(resp['Body'].read())
  103. # 新查询设备字段
  104. def queryInterface(self, request_dict, userID, response):
  105. page = request_dict.get('page', None)
  106. line = request_dict.get('line', None)
  107. NickName = request_dict.get('NickName', None)
  108. uid = request_dict.get('uid', None)
  109. page = int(page)
  110. line = int(line)
  111. response.lang = 'cn'
  112. userID = userID
  113. dvqs = Device_Info.objects.filter(userID_id=userID)
  114. # # 已重置的设备
  115. # dvqs = dvqs.filter(~Q(isExist=2))
  116. if NickName:
  117. dvqs = dvqs.filter(NickName__icontains=NickName)
  118. if uid:
  119. dvqs = dvqs.filter(UID=uid)
  120. # count = dvqs.count()
  121. #分页
  122. dvql = dvqs[(page - 1) * line:page * line].values('id', 'userID', 'NickName', 'UID', 'View_Account',
  123. 'View_Password', 'ChannelIndex', 'Type', 'isShare',
  124. 'primaryUserID', 'primaryMaster', 'data_joined',
  125. 'version',
  126. 'isVod', 'isExist', 'NotificationMode')
  127. dvls = CommonService.qs_to_list(dvql)
  128. uid_list = []
  129. for dvl in dvls:
  130. uid_list.append(dvl['UID'])
  131. # if dvl['isShare'] is False:
  132. # uid_list.append(dvl['UID'])
  133. ubqs = UID_Bucket.objects.filter(uid__in=uid_list). \
  134. values('bucket__content', 'status', 'channel', 'endTime', 'uid')
  135. upqs = UID_Preview.objects.filter(uid__in=uid_list).order_by('channel').values('id', 'uid', 'channel')
  136. auth = oss2.Auth(OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET)
  137. bucket = oss2.Bucket(auth, 'oss-cn-hongkong.aliyuncs.com', 'statres')
  138. nowTime = int(time.time())
  139. data = []
  140. # 设备拓展信息表
  141. us_qs = UidSetModel.objects.filter(uid__in=uid_list). \
  142. values('uid', 'version', 'nickname', 'detect_interval')
  143. uv_dict = {}
  144. for us in us_qs:
  145. uv_dict[us['uid']] = {'version': us['version'],
  146. 'nickname': us['nickname'],
  147. 'detect_interval': us['detect_interval']}
  148. for p in dvls:
  149. p['vod'] = []
  150. for dm in ubqs:
  151. if p['UID'] == dm['uid']:
  152. if dm['endTime'] > nowTime:
  153. p['vod'].append(dm)
  154. p['preview'] = []
  155. for up in upqs:
  156. if p['UID'] == up['uid']:
  157. obj = 'uid_preview/{uid}/channel_{channel}.png'. \
  158. format(uid=up['uid'], channel=up['channel'])
  159. img_sign = bucket.sign_url('GET', obj, 300)
  160. p['preview'].append(img_sign)
  161. p_uid = p['UID']
  162. if p_uid in uv_dict:
  163. # 设备版本号
  164. p['uid_version'] = uv_dict[p_uid]['version']
  165. p['detect_interval'] = uv_dict[p_uid]['detect_interval']
  166. # 设备昵称 调用影子信息昵称,先阶段不可
  167. if uv_dict[p_uid]['nickname']:
  168. p['NickName'] = uv_dict[p_uid]['nickname']
  169. else:
  170. # 设备版本号
  171. p['uid_version'] = ''
  172. data.append(p)
  173. return response.json(0, data)
  174. #获取播放列表
  175. def do_test_query_vod_list(self, userID, ip, request_dict, response):
  176. uid = 'GZL2PEFJPLY7W6BG111A'
  177. channel = 2
  178. userID = '158943594633713800138000'
  179. dv_qs = Device_Info.objects.filter(UID=uid, userID_id=userID, isShare=False)
  180. if not dv_qs.exists():
  181. return response.json(12)
  182. vod_play_list = []
  183. bucket_name = 'azvod1'
  184. aws_access_key_id = 'AKIA2E67UIMD45Y3HL53'
  185. aws_secret_access_key = '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'
  186. session = Session(
  187. aws_access_key_id=aws_access_key_id,
  188. aws_secret_access_key=aws_secret_access_key,
  189. region_name='ap-northeast-1'
  190. )
  191. conn = session.client('s3')
  192. thumbspng = '{uid}/vod{channel}/{time}/thumbs.png'. \
  193. format(uid=uid, channel=channel, time=1590485548)
  194. response_url = conn.generate_presigned_url(
  195. 'get_object',
  196. Params={
  197. 'Bucket': bucket_name,
  198. 'Key': thumbspng
  199. },
  200. ExpiresIn=3600
  201. )
  202. # m3u8 = '{uid}/vod{channel}/{time}/{time}.m3u8'. \
  203. # format(uid=uid, channel=channel, time=vod['time'])
  204. thumb_url = response_url
  205. vod_url = 'http://192.168.136.191:8000/testApi/signplaym3u8?' \
  206. 'uid={uid}&channel={channel}&time={time}&sign=tktktktk'. \
  207. format(ip=ip, uid=uid, channel=channel, time=1590485548)
  208. vod_play_list.append({
  209. 'name': 1590485548,
  210. 'sign_url': vod_url,
  211. 'thumb': thumb_url,
  212. 'sec': 12})
  213. return response.json(0, vod_play_list)
  214. #生成m3u8列表
  215. def do_sign_play_m3u8(self, request_dict, response):
  216. uid = 'GZL2PEFJPLY7W6BG111A'
  217. channel = 2
  218. storeTime = 1591344070
  219. fg = 6
  220. bucket__region = 'ap-northeast-1'
  221. bucket_name = 'azvod1'
  222. aws_access_key_id = 'AKIA2E67UIMD45Y3HL53'
  223. aws_secret_access_key = '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'
  224. session = Session(
  225. aws_access_key_id=aws_access_key_id,
  226. aws_secret_access_key=aws_secret_access_key,
  227. region_name=bucket__region
  228. )
  229. '''
  230. http://test.dvema.com/cloudstorage/signplaym3u8?uid=VVDHCVBYDKFMJRWA111A&channel=1&time=1586940120&sign=tktktktk
  231. '''
  232. conn = session.client('s3')
  233. playlist_entries = []
  234. for i in range(fg):
  235. thumbspng = '{uid}/vod{channel}/{time}/ts{i}.ts'. \
  236. format(uid=uid, channel=channel, time=storeTime, i=i)
  237. response_url = conn.generate_presigned_url(
  238. 'get_object',
  239. Params={
  240. 'Bucket': bucket_name,
  241. 'Key': thumbspng
  242. },
  243. ExpiresIn=86400
  244. )
  245. # m3u8 = '{uid}/vod{channel}/{time}/{time}.m3u8'. \
  246. # format(uid=uid, channel=channel, time=vod['time'])
  247. playlist_entries.append({
  248. 'name': response_url,
  249. 'duration': 10,
  250. })
  251. playlist = PlaylistGenerator(playlist_entries).generate()
  252. response = HttpResponse(playlist)
  253. response['Content-Type'] = 'application/octet-stream'
  254. response['Content-Disposition'] = 'attachment;filename="play.m3u8"'
  255. return response
  256. # return HttpResponse(playlist)
  257. # response = HttpResponse(playlist, content_type="application/vnd.apple.mpegurl")
  258. # # response = HttpResponse(playlist, content_type="application/octet-stream")
  259. # return response
  260. # return HttpResponse(status=200, content=playlist)
  261. def do_test_get_sign_sts(self, request_dict, ip, response):
  262. # uid = 'GZL2PEFJPLY7W6BG111A'
  263. # channel = 2
  264. uid = 'VVDHCVBYDKFMJRWA111A'
  265. channel = 1
  266. now_time = int(time.time())
  267. ubqs = UID_Bucket.objects.filter(uid=uid, channel=channel, endTime__gte=now_time). \
  268. values("bucket__mold", "bucket__bucket", "bucket__endpoint", "bucket__region", "endTime")
  269. if ubqs.exists():
  270. if ubqs[0]["bucket__mold"] == 0:
  271. # 阿里云 oss sts
  272. oc_qs = OssCrdModel.objects.filter(uid=uid, channel=channel). \
  273. values("addTime", "data")
  274. if oc_qs.exists():
  275. endTime = int(oc_qs[0]["addTime"]) + 3500
  276. if endTime > now_time:
  277. print(endTime)
  278. print(now_time)
  279. res = json.loads(oc_qs[0]["data"])
  280. return JsonResponse(status=200, data=res)
  281. # 套餐id
  282. storage = '{uid}/vod{channel}/'.format(uid=uid, channel=channel)
  283. bucket_name = ubqs[0]['bucket__bucket']
  284. endpoint = ubqs[0]['bucket__endpoint']
  285. access_key_id = OSS_STS_ACCESS_KEY
  286. access_key_secret = OSS_STS_ACCESS_SECRET
  287. region_id = ubqs[0]['bucket__region']
  288. role_arn = OSS_ROLE_ARN
  289. clt = client.AcsClient(access_key_id, access_key_secret, region_id)
  290. req = AssumeRoleRequest.AssumeRoleRequest()
  291. # 设置返回值格式为JSON。
  292. req.set_accept_format('json')
  293. req.set_RoleArn(role_arn)
  294. req.set_RoleSessionName(uid + '_' + channel)
  295. req.set_DurationSeconds(3600)
  296. Resource_access = "acs:oss:*:*:{bucket_name}/{uid_channel}*". \
  297. format(bucket_name=bucket_name, uid_channel=storage)
  298. print(Resource_access)
  299. policys = {
  300. "Version": "1",
  301. "Statement": [
  302. {
  303. "Action": ["oss:PutObject", "oss:DeleteObject", ],
  304. "Resource": [Resource_access],
  305. "Effect": "Allow",
  306. "Condition": {
  307. "IpAddress": {"acs:SourceIp": ip}
  308. # "IpAddress": {"acs:SourceIp": "120.237.157.184"}
  309. # "IpAddress": {"acs:SourceIp": "*"}
  310. }
  311. }
  312. ]
  313. }
  314. req.set_Policy(Policy=json.dumps(policys))
  315. body = clt.do_action(req)
  316. # 使用RAM账号的AccessKeyId和AccessKeySecret向STS申请临时token。
  317. token = json.loads(body.decode('utf-8'))
  318. print(token)
  319. res = {
  320. 'AccessKeyId': token['Credentials']['AccessKeyId'],
  321. 'AccessKeySecret': token['Credentials']['AccessKeySecret'],
  322. 'SecurityToken': token['Credentials']['SecurityToken'],
  323. 'Expiration': token['Credentials']['Expiration'],
  324. 'expire': '3600',
  325. 'endpoint': endpoint,
  326. 'bucket_name': bucket_name,
  327. 'arn': token['AssumedRoleUser']['Arn'],
  328. 'code': 0,
  329. 'storage': storage,
  330. 'endTime': ubqs[0]['endTime'],
  331. 'ip': ip}
  332. if oc_qs.exists():
  333. oc_qs.update(data=json.dumps(res), addTime=now_time)
  334. else:
  335. OssCrdModel.objects.create \
  336. (uid=uid, channel=channel, data=json.dumps(res), addTime=now_time)
  337. return JsonResponse(status=200, data=res)
  338. elif ubqs[0]["bucket__mold"] == 1:
  339. # 亚马逊 s3 sts
  340. sts_qs = StsCrdModel.objects.filter(uid=uid, channel=channel). \
  341. values("addTime", "data")
  342. if sts_qs.exists():
  343. endTime = int(sts_qs[0]["addTime"]) + 3500
  344. if endTime > now_time:
  345. print(endTime)
  346. print(now_time)
  347. res = json.loads(sts_qs[0]["data"])
  348. return JsonResponse(status=200, data=res)
  349. # 套餐id
  350. storage = '{uid}/vod{channel}/'.format(uid=uid, channel=channel)
  351. bucket_name = ubqs[0]['bucket__bucket']
  352. endpoint = ubqs[0]['bucket__endpoint']
  353. region_id = ubqs[0]['bucket__region']
  354. ###############
  355. REGION_NAME = region_id
  356. boto3_sts = boto3.client(
  357. 'sts',
  358. aws_access_key_id='AKIA2E67UIMD45Y3HL53',
  359. aws_secret_access_key='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',
  360. region_name=REGION_NAME
  361. )
  362. Policy = {
  363. "Version": "2012-10-17",
  364. "Statement": [
  365. {
  366. "Effect": "Allow",
  367. "Action": "s3:*",
  368. "Resource": ["arn:aws:s3:::azvod1/{uid_channel}/*".
  369. format(uid_channel=storage)]
  370. }
  371. ]
  372. }
  373. response = boto3_sts.get_federation_token(
  374. Name='{role_name}'.format(role_name=uid + '_' + channel),
  375. Policy=json.dumps(Policy),
  376. DurationSeconds=7200
  377. )
  378. ###############
  379. res = {
  380. 'AccessKeyId': response['Credentials']['AccessKeyId'],
  381. 'AccessKeySecret': response['Credentials']['SecretAccessKey'],
  382. 'SessionToken': response['Credentials']['SessionToken'],
  383. 'Expiration': response['Credentials']['Expiration'],
  384. 'expire': '3600',
  385. 'endpoint': endpoint,
  386. 'bucket_name': bucket_name,
  387. 'arn': response['FederatedUser']['Arn'],
  388. 'code': 0,
  389. 'storage': storage,
  390. 'endTime': ubqs[0]['endTime'],
  391. 'ip': ip,
  392. }
  393. if sts_qs.exists():
  394. sts_qs.update(data=json.dumps(res, default=str), addTime=now_time)
  395. else:
  396. StsCrdModel.objects.create(uid=uid, channel=channel, data=json.dumps(res, default=str),
  397. addTime=now_time, type=1)
  398. return JsonResponse(status=200, data=res)
  399. else:
  400. res = {'code': 404, 'msg': 'data not exists!'}
  401. return HttpResponse(json.dumps(res, ensure_ascii=False),
  402. content_type="application/json,charset=utf-8")
  403. def do_get_sign_sts(self, request_dict, ip, response):
  404. # uid = 'GZL2PEFJPLY7W6BG111A'
  405. # channel = 2
  406. uid = 'VVDHCVBYDKFMJRWA111A'
  407. channel = 1
  408. now_time = int(time.time())
  409. ubqs = UID_Bucket.objects.filter(uid=uid, channel=channel, endTime__gte=now_time). \
  410. values("bucket__mold", "bucket__bucket", "bucket__endpoint", "bucket__region", "endTime")
  411. if ubqs.exists():
  412. if ubqs[0]["bucket__mold"] == 0:
  413. # 阿里云 oss sts
  414. oc_qs = OssCrdModel.objects.filter(uid=uid, channel=channel). \
  415. values("addTime", "data")
  416. if oc_qs.exists():
  417. endTime = int(oc_qs[0]["addTime"]) + 3500
  418. if endTime > now_time:
  419. print(endTime)
  420. print(now_time)
  421. res = json.loads(oc_qs[0]["data"])
  422. return JsonResponse(status=200, data=res)
  423. # 套餐id
  424. storage = '{uid}/vod{channel}/'.format(uid=uid, channel=channel)
  425. bucket_name = ubqs[0]['bucket__bucket']
  426. endpoint = ubqs[0]['bucket__endpoint']
  427. access_key_id = OSS_STS_ACCESS_KEY
  428. access_key_secret = OSS_STS_ACCESS_SECRET
  429. region_id = ubqs[0]['bucket__region']
  430. role_arn = OSS_ROLE_ARN
  431. clt = client.AcsClient(access_key_id, access_key_secret, region_id)
  432. req = AssumeRoleRequest.AssumeRoleRequest()
  433. # 设置返回值格式为JSON。
  434. req.set_accept_format('json')
  435. req.set_RoleArn(role_arn)
  436. req.set_RoleSessionName(uid + '_' + channel)
  437. req.set_DurationSeconds(3600)
  438. Resource_access = "acs:oss:*:*:{bucket_name}/{uid_channel}*". \
  439. format(bucket_name=bucket_name, uid_channel=storage)
  440. print(Resource_access)
  441. policys = {
  442. "Version": "1",
  443. "Statement": [
  444. {
  445. "Action": ["oss:PutObject", "oss:DeleteObject", ],
  446. "Resource": [Resource_access],
  447. "Effect": "Allow",
  448. "Condition": {
  449. "IpAddress": {"acs:SourceIp": ip}
  450. # "IpAddress": {"acs:SourceIp": "120.237.157.184"}
  451. # "IpAddress": {"acs:SourceIp": "*"}
  452. }
  453. }
  454. ]
  455. }
  456. req.set_Policy(Policy=json.dumps(policys))
  457. body = clt.do_action(req)
  458. # 使用RAM账号的AccessKeyId和AccessKeySecret向STS申请临时token。
  459. token = json.loads(body.decode('utf-8'))
  460. print(token)
  461. res = {
  462. 'AccessKeyId': token['Credentials']['AccessKeyId'],
  463. 'AccessKeySecret': token['Credentials']['AccessKeySecret'],
  464. 'SecurityToken': token['Credentials']['SecurityToken'],
  465. 'Expiration': token['Credentials']['Expiration'],
  466. 'expire': '3600',
  467. 'endpoint': endpoint,
  468. 'bucket_name': bucket_name,
  469. 'arn': token['AssumedRoleUser']['Arn'],
  470. 'code': 0,
  471. 'storage': storage,
  472. 'endTime': ubqs[0]['endTime'],
  473. 'ip': ip}
  474. if oc_qs.exists():
  475. oc_qs.update(data=json.dumps(res), addTime=now_time)
  476. else:
  477. OssCrdModel.objects.create \
  478. (uid=uid, channel=channel, data=json.dumps(res), addTime=now_time)
  479. return JsonResponse(status=200, data=res)
  480. elif ubqs[0]["bucket__mold"] == 1:
  481. # 亚马逊 s3 sts
  482. sts_qs = StsCrdModel.objects.filter(uid=uid, channel=channel). \
  483. values("addTime", "data")
  484. if sts_qs.exists():
  485. endTime = int(sts_qs[0]["addTime"]) + 3500
  486. if endTime > now_time:
  487. print(endTime)
  488. print(now_time)
  489. res = json.loads(sts_qs[0]["data"])
  490. return JsonResponse(status=200, data=res)
  491. # 套餐id
  492. storage = '{uid}/vod{channel}/'.format(uid=uid, channel=channel)
  493. bucket_name = ubqs[0]['bucket__bucket']
  494. endpoint = ubqs[0]['bucket__endpoint']
  495. region_id = ubqs[0]['bucket__region']
  496. ###############
  497. REGION_NAME = region_id
  498. boto3_sts = boto3.client(
  499. 'sts',
  500. aws_access_key_id='AKIA2E67UIMD45Y3HL53',
  501. aws_secret_access_key='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',
  502. region_name=REGION_NAME
  503. )
  504. Policy = {
  505. "Version": "2012-10-17",
  506. "Statement": [
  507. {
  508. "Effect": "Allow",
  509. "Action": "s3:*",
  510. "Resource": ["arn:aws:s3:::azvod1/{uid_channel}/*".
  511. format(uid_channel=storage)]
  512. }
  513. ]
  514. }
  515. response = boto3_sts.get_federation_token(
  516. Name='{role_name}'.format(role_name=uid + '_' + channel),
  517. Policy=json.dumps(Policy),
  518. DurationSeconds=7200
  519. )
  520. ###############
  521. res = {
  522. 'AccessKeyId': response['Credentials']['AccessKeyId'],
  523. 'AccessKeySecret': response['Credentials']['SecretAccessKey'],
  524. 'SessionToken': response['Credentials']['SessionToken'],
  525. 'Expiration': response['Credentials']['Expiration'],
  526. 'expire': '3600',
  527. 'endpoint': endpoint,
  528. 'bucket_name': bucket_name,
  529. 'arn': response['FederatedUser']['Arn'],
  530. 'code': 0,
  531. 'storage': storage,
  532. 'endTime': ubqs[0]['endTime'],
  533. 'ip': ip,
  534. }
  535. if sts_qs.exists():
  536. sts_qs.update(data=json.dumps(res, default=str), addTime=now_time)
  537. else:
  538. StsCrdModel.objects.create(uid=uid, channel=channel, data=json.dumps(res, default=str),
  539. addTime=now_time, type=1)
  540. return JsonResponse(status=200, data=res)
  541. else:
  542. res = {'code': 404, 'msg': 'data not exists!'}
  543. return HttpResponse(json.dumps(res, ensure_ascii=False),
  544. content_type="application/json,charset=utf-8")
  545. def do_pay_by_ali(self, request_dict, userID, response):
  546. uid = request_dict.get('uid', None)
  547. rank = request_dict.get('rank', None)
  548. channel = request_dict.get('channel', None)
  549. qs = Device_Info.objects.filter(userID_id=userID, UID=uid, isShare=False)
  550. if not qs.exists():
  551. return response.json(12)
  552. if not channel or not rank:
  553. return response.json(444, 'channel,rank')
  554. smqs = Store_Meal.objects.filter(id=rank). \
  555. values("currency", "price", "content", "day", "bucket__storeDay", "bucket__region", "type")
  556. if not smqs.exists():
  557. # 套餐不存在
  558. return response.json(173)
  559. if smqs[0]['type'] != 1:
  560. return response.json(10, '不支持支付宝支付')
  561. currency = smqs[0]['currency']
  562. price = smqs[0]['price']
  563. content = smqs[0]['content']
  564. day = smqs[0]['day']
  565. nowTime = int(time.time())
  566. ubqs = UID_Bucket.objects.filter(uid=uid, channel=channel, endTime__gte=nowTime). \
  567. values("bucket__storeDay", "bucket__region")
  568. if ubqs.exists():
  569. if ubqs[0]['bucket__region'] != smqs[0]['bucket__region']:
  570. return response.json(712) # 区域不一致
  571. elif ubqs[0]['bucket__storeDay'] != smqs[0]['bucket__storeDay']:
  572. return response.json(713) # 不可更改套餐
  573. # 续费流程
  574. nowTime = int(time.time())
  575. # 新增流程
  576. orderID = CommonService.createOrderID()
  577. try:
  578. aliPayObj = AliPayObject()
  579. alipay = aliPayObj.conf()
  580. order_string = alipay.api_alipay_trade_wap_pay(
  581. out_trade_no=orderID,
  582. total_amount=price,
  583. subject="测试哟",
  584. return_url="{SERVER_DOMAIN_SSL}cloudVod/payOK".format(SERVER_DOMAIN_SSL=SERVER_DOMAIN_SSL),
  585. notify_url="{SERVER_DOMAIN_SSL}cloudVod/aliPayCallback".format(SERVER_DOMAIN_SSL=SERVER_DOMAIN_SSL)
  586. # return_url="http://192.168.136.40/cloudVod/payOK",
  587. # notify_url="http://192.168.136.40/cloudVod/aliPayCallback"
  588. )
  589. except Exception as e:
  590. print(repr(e))
  591. return response.json(10, repr(e))
  592. if order_string:
  593. redirectUrl = aliPayObj.alipay_prefix + order_string
  594. Order_Model.objects.create(orderID=orderID, UID=uid, channel=channel, userID_id=userID, desc=content,
  595. price=price, currency=currency, addTime=nowTime, updTime=nowTime,
  596. endTime=nowTime + int(day) * 3600 * 24, rank_id=rank, payType=1)
  597. return JsonResponse(status=200,
  598. data={'result_code': 0, 'reason': 'success',
  599. 'result': {"redirectUrl": redirectUrl, "orderID": orderID},
  600. 'error_code': 0})
  601. else:
  602. return response.json(10, '生成订单错误')
  603. def do_filter_playlist(self, request_dict, userID, response):
  604. startTime = int(request_dict.get('startTime', None))
  605. endTime = int(request_dict.get('endTime', None))
  606. uid = request_dict.get('uid', None)
  607. channel = request_dict.get('channel', None)
  608. dvqs = Device_Info.objects.filter(UID=uid, userID_id=userID, isShare=False)
  609. if not dvqs.exists():
  610. return response.json(12)
  611. ubqs = UID_Bucket.objects.filter(uid=uid, channel=channel).values('status')
  612. if not ubqs.exists():
  613. return response.json(10, '设备未购买')
  614. nowTime = int(time.time())
  615. # vodqs = VodHlsModel.objects.filter(uid=uid, channel=channel, time__range=(startTime, endTime),
  616. # endTime__gte=nowTime) \
  617. # .values("time", "sec", "bucket__bucket", "bucket__endpoint", "bucket__region")
  618. vodqs = VodHlsModel.objects.filter(uid=uid, channel=channel) \
  619. .values("time", "sec", "bucket__bucket", "bucket__endpoint", "bucket__region")
  620. vod_play_list = []
  621. print(vodqs)
  622. auth = oss2.Auth(OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET)
  623. for vod in vodqs:
  624. bucket_name = vod["bucket__bucket"]
  625. endpoint = vod["bucket__endpoint"]
  626. bucket = oss2.Bucket(auth, endpoint, bucket_name)
  627. m3u8 = '{uid}/vod{channel}/{time}/{time}.m3u8'. \
  628. format(uid=uid, channel=channel, time=vod['time'])
  629. ts = '{uid}/vod{channel}/{time}/ts0.ts'. \
  630. format(uid=uid, channel=channel, time=vod['time'])
  631. url = bucket.sign_url('GET', m3u8, 3600, params={'x-oss-process': 'hls/sign'})
  632. urllst = url.split('?')
  633. url_start = urllib.parse.unquote(urllst[0])
  634. url_end = urllst[1]
  635. vod_play_url = '{url_start}?{url_end}'. \
  636. format(url_start=url_start, url_end=url_end)
  637. thumb = bucket.sign_url('GET', ts, 3600,
  638. params={'x-oss-process': 'video/snapshot,t_10000,m_fast,w_300'})
  639. vod_play_list.append({'name': vod['time'], 'sign_url': vod_play_url, 'thumb': thumb, 'sec': vod['sec']})
  640. return response.json(0, vod_play_list)