CloudStorage.py 39 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877
  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: AnsjerFormal
  7. @software: PyCharm
  8. @DATE: 2018/12/5 9:30
  9. @Version: python3.6
  10. @MODIFY DECORD:ansjer dev
  11. @file: cloudstorage.py
  12. @Contact: chanjunkai@163.com
  13. """
  14. import json
  15. import time
  16. import urllib
  17. import boto3
  18. import oss2
  19. import paypalrestsdk
  20. from aliyunsdkcore import client
  21. from aliyunsdksts.request.v20150401 import AssumeRoleRequest
  22. from boto3.session import Session
  23. from django.http import JsonResponse, HttpResponseRedirect, HttpResponse
  24. from django.views.generic.base import View
  25. from Ansjer.config import OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET, OSS_ROLE_ARN, SERVER_DOMAIN, PAYPAL_CRD, \
  26. SERVER_DOMAIN_SSL
  27. from Model.models import Device_Info, Order_Model, Store_Meal, VodHlsModel, OssCrdModel, UID_Bucket, StsCrdModel
  28. from Object.AliPayObject import AliPayObject
  29. from Object.ResponseObject import ResponseObject
  30. from Object.TokenObject import TokenObject
  31. from Object.UidTokenObject import UidTokenObject
  32. from Service.CommonService import CommonService
  33. from Object.m3u8generate import PlaylistGenerator
  34. SERVER_DOMAIN = 'http://test.dvema.com/'
  35. '''
  36. eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VySUQiOiIxNTY3NzU4MjE4NjE5MTM4MDAxMzgwMDAiLCJleHAiOjE1ODk1MjM0OTIsIm1fY29kZSI6IjEyMzQxMzI0MzIxNCIsImxhbmciOiJlbiIsInVzZXIiOiIxMzExOTY1NzcxMyJ9.e2NdhJtbXrDngZTSmOX_52Y-oxyfUEXjZD_qNxg6VrU
  37. 生成订单
  38. http://test.dvema.com/cloudstorage/createpayorder?token=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VySUQiOiIxNTY3NzU4MjE4NjE5MTM4MDAxMzgwMDAiLCJleHAiOjE1ODk1MjM0OTIsIm1fY29kZSI6IjEyMzQxMzI0MzIxNCIsImxhbmciOiJlbiIsInVzZXIiOiIxMzExOTY1NzcxMyJ9.e2NdhJtbXrDngZTSmOX_52Y-oxyfUEXjZD_qNxg6VrU&uid=VVDHCVBYDKFMJRWA111A&channel=1&pay_type=0&commodity_code=aws_us_vod_7_val_30
  39. 开启开关
  40. http://test.dvema.com/cloudstorage/changevodstatus?token=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VySUQiOiIxNTY3NzU4MjE4NjE5MTM4MDAxMzgwMDAiLCJleHAiOjE1ODk1MjM0OTIsIm1fY29kZSI6IjEyMzQxMzI0MzIxNCIsImxhbmciOiJlbiIsInVzZXIiOiIxMzExOTY1NzcxMyJ9.e2NdhJtbXrDngZTSmOX_52Y-oxyfUEXjZD_qNxg6VrU&uid=VVDHCVBYDKFMJRWA111A&channel=1&status=1
  41. # 获取播放列表
  42. http://test.dvema.com/cloudstorage/queryvodlist?token=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VySUQiOiIxNTY3NzU4MjE4NjE5MTM4MDAxMzgwMDAiLCJleHAiOjE1ODk1MjM0OTIsIm1fY29kZSI6IjEyMzQxMzI0MzIxNCIsImxhbmciOiJlbiIsInVzZXIiOiIxMzExOTY1NzcxMyJ9.e2NdhJtbXrDngZTSmOX_52Y-oxyfUEXjZD_qNxg6VrU&uid=VVDHCVBYDKFMJRWA111A&startTime=1&endTime=1954687458
  43. http://test.dvema.com/cloudstorage/storeplaylist?uidToken=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1aWQiOiJWVkRIQ1ZCWURLRk1KUldBMTExQSIsImNoYW5uZWwiOiIxIn0.eGwi5QKyrXi4WSKRbrUG7iFTChv_Utec2hSnqZkDKt8
  44. http://test.dvema.com/cloudstorage/storeplaylist?uidToken=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1aWQiOiJWVkRIQ1ZCWURLRk1KUldBMTExQSIsImNoYW5uZWwiOiIxIn0.eGwi5QKyrXi4WSKRbrUG7iFTChv_Utec2hSnqZkDKt8&time=1586940120&sec=20&fg=10
  45. '''
  46. # 设备信息添加
  47. class CloudStorageView(View):
  48. def get(self, request, *args, **kwargs):
  49. request.encoding = 'utf-8'
  50. operation = kwargs.get('operation')
  51. return self.validation(request.GET, request, operation)
  52. def post(self, request, *args, **kwargs):
  53. request.encoding = 'utf-8'
  54. operation = kwargs.get('operation')
  55. return self.validation(request.POST, request, operation)
  56. def validation(self, request_dict, request, operation):
  57. response = ResponseObject()
  58. if operation is None:
  59. return response.json(444, 'error path')
  60. elif operation == 'testpaly.m3u8':
  61. return self.do_test_play_m3u8(request_dict, response)
  62. elif operation == 'dopaypalcallback':
  63. return self.do_pay_by_paypal_callback(request_dict, response)
  64. elif operation == 'doalicallback':
  65. return self.do_pay_by_ali_callback(request)
  66. elif operation == 'getsignsts':
  67. ip = CommonService.get_ip_address(request)
  68. return self.do_get_sign_sts(request_dict, ip, response)
  69. elif operation == 'storeplaylist':
  70. return self.do_store_playlist(request_dict, response)
  71. elif operation == 'signplaym3u8':
  72. return self.do_sign_play_m3u8(request_dict, response)
  73. elif operation == 'payOK':
  74. return self.do_pay_ok()
  75. else:
  76. token = request_dict.get('token', None)
  77. # 设备主键uid
  78. tko = TokenObject(token)
  79. response.lang = tko.lang
  80. if tko.code != 0:
  81. return response.json(tko.code)
  82. userID = tko.userID
  83. if operation == 'createpayorder':
  84. return self.do_create_pay_order(request_dict, userID, response)
  85. elif operation == 'changevodstatus':
  86. return self.do_change_vod_status(request_dict, userID, response)
  87. elif operation == 'queryvodlist':
  88. return self.do_query_vod_list(request_dict, userID, response)
  89. elif operation == 'commoditylist':
  90. return self.do_commodity_list(request_dict, userID, response)
  91. elif operation == 'queryorder':
  92. return self.do_query_order(request_dict, userID, response)
  93. else:
  94. return response.json(414)
  95. def do_test_play_m3u8(self, request_dict, response):
  96. playlist_entries = [
  97. {
  98. 'name': 'https://azvod1.s3-ap-northeast-1.amazonaws.com/test/output000.ts',
  99. 'duration': 10,
  100. },
  101. {
  102. 'name': 'https://azvod1.s3-ap-northeast-1.amazonaws.com/test/output000.ts',
  103. 'duration': 10,
  104. },
  105. {
  106. 'name': 'https://azvod1.s3-ap-northeast-1.amazonaws.com/test/output000.ts',
  107. 'duration': 10,
  108. },
  109. ]
  110. playlist = PlaylistGenerator(playlist_entries).generate()
  111. # return HttpResponse(playlist)
  112. # response = HttpResponse(playlist, content_type="application/octet-stream")
  113. response = HttpResponse(playlist, content_type="application/vnd.apple.mpegurl")
  114. return response
  115. def do_query_order(self, request_dict, userID, response):
  116. return
  117. def do_commodity_list(self, request_dict, userID, response):
  118. mold = request_dict.get('mold', None)
  119. qs = Store_Meal.objects
  120. if mold:
  121. qs = qs.filter(bucket__mold=1)
  122. else:
  123. qs = qs.filter(bucket__mold=0)
  124. qs = qs.values("id", "title", "content", "price",
  125. "day", "currency", "bucket__storeDay",
  126. "bucket__bucket", "bucket__area", "type",
  127. "commodity_code", "commodity_type")
  128. if qs.exists():
  129. ql = list(qs)
  130. from operator import itemgetter
  131. from itertools import groupby
  132. ql.sort(key=itemgetter('bucket__area'))
  133. res = []
  134. for area, items in groupby(ql, key=itemgetter('bucket__area')):
  135. res_c = {'area': area, 'items': list(items)}
  136. res.append(res_c)
  137. result = {
  138. 'meals': res,
  139. 'extra':
  140. {
  141. 'cloud_banner': 'https://www.dvema.com/web/images/cloud_cn_banner.png',
  142. 'cloud_en_baner': 'https://www.dvema.com/web/images/cloud_en_banner.png'
  143. }
  144. }
  145. return response.json(0, result)
  146. else:
  147. return response.json(0)
  148. def do_sign_play_m3u8(self, request_dict, response):
  149. uid = request_dict.get('uid', None)
  150. channel = request_dict.get('channel', None)
  151. storeTime = request_dict.get('time', None)
  152. now_time = int(time.time())
  153. vh_qs = VodHlsModel.objects.filter(uid=uid, channel=channel, time=int(storeTime), endTime__gte=now_time). \
  154. values("sec", "fg", "bucket__bucket", "bucket__endpoint", "bucket__region", "bucket__mold")
  155. if not vh_qs.exists():
  156. return response.json()
  157. sec = vh_qs[0]['sec']
  158. fg = vh_qs[0]['fg']
  159. bucket__region = vh_qs[0]['bucket__region']
  160. bucket_name = vh_qs[0]['bucket__bucket']
  161. aws_access_key_id = 'AKIA2E67UIMD45Y3HL53'
  162. aws_secret_access_key = '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'
  163. session = Session(
  164. aws_access_key_id=aws_access_key_id,
  165. aws_secret_access_key=aws_secret_access_key,
  166. region_name=bucket__region
  167. )
  168. conn = session.client('s3')
  169. playlist_entries = []
  170. for i in range(fg):
  171. thumbspng = '{uid}/vod{channel}/{time}/thumbs.png'. \
  172. format(uid=uid, channel=channel, time=storeTime)
  173. response_url = conn.generate_presigned_url(
  174. 'get_object',
  175. Params={
  176. 'Bucket': bucket_name,
  177. 'Key': thumbspng
  178. },
  179. ExpiresIn=3600
  180. )
  181. # m3u8 = '{uid}/vod{channel}/{time}/{time}.m3u8'. \
  182. # format(uid=uid, channel=channel, time=vod['time'])
  183. playlist_entries.append({
  184. 'name': response_url,
  185. 'duration': 10,
  186. })
  187. playlist = PlaylistGenerator(playlist_entries).generate()
  188. response = HttpResponse(playlist, content_type="application/octet-stream")
  189. return response
  190. return HttpResponse(status=200, content=playlist)
  191. def do_get_sign_sts(self, request_dict, ip, response):
  192. uidToken = request_dict.get('uidToken', None)
  193. utko = UidTokenObject(uidToken)
  194. if utko.flag is False:
  195. return response.json(444, 'uidToken')
  196. uid = utko.UID
  197. channel = utko.channel
  198. print(channel)
  199. print(uid)
  200. now_time = int(time.time())
  201. ubqs = UID_Bucket.objects.filter(uid=uid, channel=channel, endTime__gte=now_time). \
  202. values("bucket__mold", "bucket__bucket", "bucket__endpoint", "bucket__region", "endTime")
  203. if ubqs.exists():
  204. if ubqs[0]["bucket__mold"] == 0:
  205. # 阿里云 oss sts
  206. oc_qs = OssCrdModel.objects.filter(uid=uid, channel=channel). \
  207. values("addTime", "data")
  208. if oc_qs.exists():
  209. endTime = int(oc_qs[0]["addTime"]) + 3500
  210. if endTime > now_time:
  211. print(endTime)
  212. print(now_time)
  213. res = json.loads(oc_qs[0]["data"])
  214. return JsonResponse(status=200, data=res)
  215. # 套餐id
  216. storage = '{uid}/vod{channel}/'.format(uid=uid, channel=channel)
  217. bucket_name = ubqs[0]['bucket__bucket']
  218. endpoint = ubqs[0]['bucket__endpoint']
  219. access_key_id = OSS_STS_ACCESS_KEY
  220. access_key_secret = OSS_STS_ACCESS_SECRET
  221. region_id = ubqs[0]['bucket__region']
  222. role_arn = OSS_ROLE_ARN
  223. clt = client.AcsClient(access_key_id, access_key_secret, region_id)
  224. req = AssumeRoleRequest.AssumeRoleRequest()
  225. # 设置返回值格式为JSON。
  226. req.set_accept_format('json')
  227. req.set_RoleArn(role_arn)
  228. req.set_RoleSessionName(uid + '_' + channel)
  229. req.set_DurationSeconds(3600)
  230. Resource_access = "acs:oss:*:*:{bucket_name}/{uid_channel}*". \
  231. format(bucket_name=bucket_name, uid_channel=storage)
  232. print(Resource_access)
  233. policys = {
  234. "Version": "1",
  235. "Statement": [
  236. {
  237. "Action": ["oss:PutObject", "oss:DeleteObject", ],
  238. "Resource": [Resource_access],
  239. "Effect": "Allow",
  240. "Condition": {
  241. "IpAddress": {"acs:SourceIp": ip}
  242. # "IpAddress": {"acs:SourceIp": "120.237.157.184"}
  243. # "IpAddress": {"acs:SourceIp": "*"}
  244. }
  245. }
  246. ]
  247. }
  248. req.set_Policy(Policy=json.dumps(policys))
  249. body = clt.do_action(req)
  250. # 使用RAM账号的AccessKeyId和AccessKeySecret向STS申请临时token。
  251. token = json.loads(body.decode('utf-8'))
  252. print(token)
  253. res = {
  254. 'AccessKeyId': token['Credentials']['AccessKeyId'],
  255. 'AccessKeySecret': token['Credentials']['AccessKeySecret'],
  256. 'SecurityToken': token['Credentials']['SecurityToken'],
  257. 'Expiration': token['Credentials']['Expiration'],
  258. 'expire': '3600',
  259. 'endpoint': endpoint,
  260. 'bucket_name': bucket_name,
  261. 'arn': token['AssumedRoleUser']['Arn'],
  262. 'code': 0,
  263. 'storage': storage,
  264. 'endTime': ubqs[0]['endTime'],
  265. 'ip': ip}
  266. if oc_qs.exists():
  267. oc_qs.update(data=json.dumps(res), addTime=now_time)
  268. else:
  269. OssCrdModel.objects.create \
  270. (uid=uid, channel=channel, data=json.dumps(res), addTime=now_time)
  271. return JsonResponse(status=200, data=res)
  272. elif ubqs[0]["bucket__mold"] == 1:
  273. # 亚马逊 s3 sts
  274. sts_qs = StsCrdModel.objects.filter(uid=uid, channel=channel). \
  275. values("addTime", "data")
  276. if sts_qs.exists():
  277. endTime = int(sts_qs[0]["addTime"]) + 3500
  278. if endTime > now_time:
  279. print(endTime)
  280. print(now_time)
  281. res = json.loads(sts_qs[0]["data"])
  282. return JsonResponse(status=200, data=res)
  283. # 套餐id
  284. storage = '{uid}/vod{channel}/'.format(uid=uid, channel=channel)
  285. bucket_name = ubqs[0]['bucket__bucket']
  286. endpoint = ubqs[0]['bucket__endpoint']
  287. region_id = ubqs[0]['bucket__region']
  288. ###############
  289. REGION_NAME = region_id
  290. boto3_sts = boto3.client(
  291. 'sts',
  292. aws_access_key_id='AKIA2E67UIMD45Y3HL53',
  293. aws_secret_access_key='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',
  294. region_name=REGION_NAME
  295. )
  296. Policy = {
  297. "Version": "2012-10-17",
  298. "Statement": [
  299. {
  300. "Effect": "Allow",
  301. "Action": "s3:*",
  302. "Resource": ["arn:aws:s3:::azvod1/{uid_channel}/*".
  303. format(uid_channel=storage)]
  304. }
  305. ]
  306. }
  307. response = boto3_sts.get_federation_token(
  308. Name='{role_name}'.format(role_name=uid + '_' + channel),
  309. Policy=json.dumps(Policy),
  310. DurationSeconds=7200
  311. )
  312. ###############
  313. res = {
  314. 'AccessKeyId': response['Credentials']['AccessKeyId'],
  315. 'AccessKeySecret': response['Credentials']['SecretAccessKey'],
  316. 'SessionToken': response['Credentials']['SessionToken'],
  317. 'Expiration': response['Credentials']['Expiration'],
  318. 'expire': '3600',
  319. 'endpoint': endpoint,
  320. 'bucket_name': bucket_name,
  321. 'arn': response['FederatedUser']['Arn'],
  322. 'code': 0,
  323. 'storage': storage,
  324. 'endTime': ubqs[0]['endTime'],
  325. 'ip': ip,
  326. }
  327. if sts_qs.exists():
  328. sts_qs.update(data=json.dumps(res), addTime=now_time)
  329. else:
  330. StsCrdModel.objects.create(uid=uid, channel=channel, data=json.dumps(res, default=str),
  331. addTime=now_time, type=1)
  332. return JsonResponse(status=200, data=res)
  333. else:
  334. res = {'code': 404, 'msg': 'data not exists!'}
  335. return HttpResponse(json.dumps(res, ensure_ascii=False),
  336. content_type="application/json,charset=utf-8")
  337. def do_query_vod_list(self, request_dict, userID, response):
  338. startTime = int(request_dict.get('startTime', None))
  339. endTime = int(request_dict.get('endTime', None))
  340. uid = request_dict.get('uid', None)
  341. channel = request_dict.get('channel', None)
  342. dv_qs = Device_Info.objects.filter(UID=uid, userID_id=userID, isShare=False)
  343. if not dv_qs.exists():
  344. return response.json(12)
  345. now_time = int(time.time())
  346. vh_qs = VodHlsModel.objects.filter(uid=uid, channel=channel,
  347. time__range=(startTime, endTime),
  348. endTime__gte=now_time). \
  349. values("time", "sec", "bucket__bucket", "bucket__endpoint", "bucket__region", "bucket__mold")
  350. vod_play_list = []
  351. print("---")
  352. print(vh_qs)
  353. auth = oss2.Auth(OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET)
  354. for vod in vh_qs:
  355. bucket__mold = vod["bucket__mold"]
  356. bucket_name = vod["bucket__bucket"]
  357. endpoint = vod["bucket__endpoint"]
  358. bucket__region = vod["bucket__region"]
  359. if bucket__mold == 0:
  360. bucket = oss2.Bucket(auth, endpoint, bucket_name)
  361. m3u8 = '{uid}/vod{channel}/{time}/{time}.m3u8'. \
  362. format(uid=uid, channel=channel, time=vod['time'])
  363. ts = '{uid}/vod{channel}/{time}/ts0.ts'. \
  364. format(uid=uid, channel=channel, time=vod['time'])
  365. url = bucket.sign_url('GET', m3u8, 3600, params={'x-oss-process': 'hls/sign'})
  366. urllst = url.split('?')
  367. url_start = urllib.parse.unquote(urllst[0])
  368. url_end = urllst[1]
  369. vod_play_url = '{url_start}?{url_end}'. \
  370. format(url_start=url_start, url_end=url_end)
  371. thumb = bucket.sign_url('GET', ts, 3600,
  372. params={'x-oss-process': 'video/snapshot,t_10000,m_fast,w_300'})
  373. vod_play_list.append({'name': vod['time'], 'sign_url': vod_play_url, 'thumb': thumb, 'sec': vod['sec']})
  374. elif bucket__mold == 1:
  375. aws_access_key_id = 'AKIA2E67UIMD45Y3HL53'
  376. aws_secret_access_key = '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'
  377. session = Session(
  378. aws_access_key_id=aws_access_key_id,
  379. aws_secret_access_key=aws_secret_access_key,
  380. region_name=bucket__region
  381. )
  382. conn = session.client('s3')
  383. thumbspng = '{uid}/vod{channel}/{time}/thumbs.png'. \
  384. format(uid=uid, channel=channel, time=vod['time'])
  385. response_url = conn.generate_presigned_url(
  386. 'get_object',
  387. Params={
  388. 'Bucket': bucket_name,
  389. 'Key': thumbspng
  390. },
  391. ExpiresIn=3600
  392. )
  393. # m3u8 = '{uid}/vod{channel}/{time}/{time}.m3u8'. \
  394. # format(uid=uid, channel=channel, time=vod['time'])
  395. thumb_url = response_url
  396. vod_url = 'http://cloudstroage.dvema.com/cloudstorage/signplaym3u8?uid={uid}&channel={channel}&time={time}&sign=tktktktk'. \
  397. format(uid=uid, channel=channel, time=vod['time'])
  398. vod_play_list.append({
  399. 'name': vod['time'],
  400. 'sign_url': vod_url,
  401. 'thumb': thumb_url,
  402. 'sec': vod['sec']})
  403. return response.json(0, vod_play_list)
  404. def do_store_playlist(self, request_dict, response):
  405. uidToken = request_dict.get('uidToken', None)
  406. storeTime = request_dict.get('time', None)
  407. sec = request_dict.get('sec', None)
  408. fg = request_dict.get('fg', None)
  409. utko = UidTokenObject(uidToken)
  410. if utko.flag is False:
  411. return response.json(444, 'uidToken')
  412. if not uidToken or not storeTime or not sec:
  413. return response.json(0)
  414. pass
  415. UID = utko.UID
  416. channel = utko.channel
  417. print('UID:')
  418. print(UID)
  419. print('channel:')
  420. print(channel)
  421. now_time = int(time.time())
  422. ubqs = UID_Bucket.objects.filter(uid=UID, channel=channel, endTime__gte=now_time). \
  423. values("bucket__mold", "bucket__bucket", "bucket__endpoint",
  424. "bucket__region", "status", "bucket__storeDay", "bucket__id")
  425. if ubqs.exists():
  426. ub_qs_data = ubqs[0]
  427. store_day = ub_qs_data['bucket__storeDay']
  428. bucket_id = ub_qs_data['bucket__id']
  429. end_time = int(storeTime) + store_day * 86400
  430. if ub_qs_data['status'] == 1:
  431. if ub_qs_data["bucket__mold"] == 0:
  432. oss_crd_qs = OssCrdModel.objects.filter(uid=UID, channel=channel)
  433. if oss_crd_qs.exists():
  434. VodHlsModel.objects.create(
  435. uid=UID,
  436. channel=channel,
  437. time=storeTime,
  438. endTime=end_time,
  439. bucket_id=bucket_id
  440. )
  441. res = {'code': 0, 'msg': '存储成功'}
  442. return HttpResponse(json.dumps(res, ensure_ascii=False),
  443. content_type="application/json,charset=utf-8")
  444. else:
  445. res = {'code': 404, 'msg': '设备未购买'}
  446. return HttpResponse(json.dumps(res, ensure_ascii=False),
  447. content_type="application/json,charset=utf-8")
  448. elif ub_qs_data['bucket__mold'] == 1:
  449. sts_qs = StsCrdModel.objects.filter(uid=UID, channel=channel, type=1). \
  450. values("addTime", "data")
  451. if sts_qs.exists():
  452. VodHlsModel.objects.create(
  453. uid=UID,
  454. channel=channel,
  455. time=storeTime,
  456. endTime=end_time,
  457. bucket_id=bucket_id,
  458. fg=fg,
  459. )
  460. res = {'code': 0, 'msg': '存储成功'}
  461. return HttpResponse(json.dumps(res, ensure_ascii=False),
  462. content_type="application/json,charset=utf-8")
  463. else:
  464. res = {'code': 404, 'msg': '设备未购买'}
  465. return HttpResponse(json.dumps(res, ensure_ascii=False),
  466. content_type="application/json,charset=utf-8")
  467. else:
  468. res = {'code': 404, 'msg': '设备未开启'}
  469. return HttpResponse(json.dumps(res, ensure_ascii=False), content_type="application/json,charset=utf-8")
  470. else:
  471. res = {'code': 404, 'msg': '设备未购买'}
  472. return HttpResponse(json.dumps(res, ensure_ascii=False), content_type="application/json,charset=utf-8")
  473. def do_change_vod_status(self, request_dict, userID, response):
  474. uid = request_dict.get('uid', None)
  475. status = request_dict.get('status', None)
  476. channel = request_dict.get('channel', None)
  477. if all([uid, status, channel]):
  478. dv_qs = Device_Info.objects.filter(UID=uid, userID_id=userID, isShare=False, isExist=1)
  479. if dv_qs.exists():
  480. now_time = int(time.time())
  481. try:
  482. ubqs = UID_Bucket.objects.filter(uid=uid, channel=channel, endTime__gte=now_time).first()
  483. except Exception as e:
  484. print(repr(e))
  485. return response.json(10, '未购买')
  486. else:
  487. ubqs.status = status
  488. ubqs.save()
  489. if status == 0:
  490. return response.json(0)
  491. else:
  492. utko = UidTokenObject()
  493. utko.generate(data={'uid': uid, 'channel': channel})
  494. uidTK = utko.token
  495. uidTkUrl = "{SERVER_DOMAIN}cloudstorage/getsignsts?uidToken={uidToken}". \
  496. format(uidToken=uidTK, SERVER_DOMAIN=SERVER_DOMAIN)
  497. storeHlsUrl = "{SERVER_DOMAIN}cloudstorage/storeplaylist?uidToken={uidToken}". \
  498. format(uidToken=uidTK, SERVER_DOMAIN=SERVER_DOMAIN)
  499. return response.json(0, {'uidTkUrl': uidTkUrl, 'storeHlsUrl': storeHlsUrl})
  500. else:
  501. return response.json(12)
  502. else:
  503. return response.json(444, 'uid,status,channel')
  504. def do_pay_error(self):
  505. response = HttpResponse()
  506. response.content = '''
  507. <!DOCTYPE html>
  508. <html>
  509. <head>
  510. <!--浏览器不缓存-->
  511. <meta http-equiv="Pragma" content="no-cache">
  512. <meta http-equiv="Cache-Control" content="no-cache">
  513. <meta http-equiv="Expires" content="0">
  514. <!--utf-8-->
  515. <meta http-equiv="content-type" content="text/html;charset=utf-8">
  516. <!-- viewport的<meta>标签,这个标签可以修改在大部分的移动设备上面的显示,为了确保适当的绘制和触屏缩放。-->
  517. <meta name="viewport" content="width=device-width, initial-scale=1.0">
  518. <link rel="shortcut icon" href="https://test.dvema.com/web/images/favicon.ico" type="image/x-icon" charset="utf-8"/>
  519. <title>Trading particulars</title>
  520. <style>
  521. .title_head{
  522. height: 50px;
  523. border-radius: 5px;
  524. background-color: #c3c6c7;
  525. text-align: center;
  526. line-height: 50px;
  527. }
  528. .content{
  529. text-align: center;
  530. margin-top: 50px;
  531. font-size: 20px;
  532. color : #ec7648
  533. }
  534. .content_img{
  535. width: 60px;
  536. height: 60px;
  537. }
  538. .bottom{
  539. margin-bottom: 10px;
  540. margin-top: 250px;
  541. color : #ec7648
  542. }
  543. .bottom_div{
  544. border: 1px solid #ec7648;
  545. line-height: 38px;
  546. text-align: center;
  547. width: 100px;
  548. height: 38px;
  549. border-radius: 5px;
  550. }
  551. .bottom_div:hover{
  552. background-color: #dde4e2;
  553. }
  554. </style>
  555. </head>
  556. <body>
  557. <div class="title_head">Trading particulars</div>
  558. <div class="content">
  559. <p >
  560. <img src="https://test.dvema.com/web/images/failed.jpg" class="content_img">
  561. <br />
  562. Payment failure
  563. </p>
  564. </div>
  565. <center class="bottom">
  566. <div class="bottom_div" onclick="payOKButton()">
  567. Finish
  568. </div>
  569. </center>
  570. <script> // 点击付款成功按钮
  571. function payOKButton() {
  572. // 复杂数据
  573. console.log('success')
  574. window.webkit.messageHandlers.jsCallOC.postMessage({"status": 0});
  575. }
  576. </script>
  577. </body>
  578. </html>
  579. '''
  580. return response
  581. def do_pay_ok(self):
  582. response = HttpResponse()
  583. response.content = '''
  584. <!DOCTYPE html>
  585. <html>
  586. <head>
  587. <!--浏览器不缓存-->
  588. <meta http-equiv="Pragma" content="no-cache">
  589. <meta http-equiv="Cache-Control" content="no-cache">
  590. <meta http-equiv="Expires" content="0">
  591. <!--utf-8-->
  592. <meta http-equiv="content-type" content="text/html;charset=utf-8">
  593. <!-- viewport的<meta>标签,这个标签可以修改在大部分的移动设备上面的显示,为了确保适当的绘制和触屏缩放。-->
  594. <meta name="viewport" content="width=device-width, initial-scale=1.0">
  595. <link rel="shortcut icon" href="https://test.dvema.com/web/images/favicon.ico" type="image/x-icon" charset="utf-8"/>
  596. <title>Trading particulars</title>
  597. <style>
  598. .title_head{
  599. height: 50px;
  600. border-radius: 5px;
  601. background-color: #c3c6c7;
  602. text-align: center;
  603. line-height: 50px;
  604. }
  605. .content{
  606. text-align: center;
  607. margin-top: 50px;
  608. font-size: 20px;
  609. color : green
  610. }
  611. .content_img{
  612. width: 60px;
  613. height: 60px;
  614. }
  615. .bottom{
  616. margin-bottom: 10px;
  617. margin-top: 250px;
  618. color : green
  619. }
  620. .bottom_div{
  621. border: 1px solid green;
  622. line-height: 38px;
  623. text-align: center;
  624. width: 100px;
  625. height: 38px;
  626. border-radius: 5px;
  627. }
  628. .bottom_div:hover{
  629. background-color: #dde4e2;
  630. }
  631. </style>
  632. </head>
  633. <body>
  634. <div class="title_head">Trading particulars</div>
  635. <div class="content">
  636. <p >
  637. <img src="https://test.dvema.com/web/images/timg.jpg" class="content_img">
  638. <br />
  639. Successful payment
  640. </p>
  641. </div>
  642. <center class="bottom">
  643. <div class="bottom_div" onclick="payOKButton()">
  644. Finish
  645. </div>
  646. </center>
  647. <script> // 点击付款成功按钮
  648. function payOKButton() {
  649. // 复杂数据
  650. console.log('success')
  651. window.webkit.messageHandlers.jsCallOC.postMessage({"status": 1});
  652. }
  653. </script>
  654. </body>
  655. </html>
  656. '''
  657. return response
  658. def do_pay_by_ali_callback(self, request):
  659. response = ResponseObject()
  660. data = request.POST.dict()
  661. signature = data["sign"]
  662. data.pop('sign')
  663. orderID = data['out_trade_no']
  664. order_qs = Order_Model.objects.filter(orderID=orderID, status=0)
  665. print(order_qs)
  666. if not order_qs.exists():
  667. return response.json(404)
  668. print(json.dumps(data))
  669. print(signature)
  670. # verify
  671. aliPayObj = AliPayObject()
  672. alipay = aliPayObj.conf()
  673. success = alipay.verify(data, signature)
  674. if success and data["trade_status"] in ("TRADE_SUCCESS", "TRADE_FINISHED"):
  675. print("trade succeed")
  676. nowTime = int(time.time())
  677. order_list = order_qs.values("UID", "commodity_code", "channel")
  678. UID = order_list[0]['UID']
  679. channel = order_list[0]['channel']
  680. commodity_code = order_list[0]['commodity_code']
  681. order_qs.update(status=1, updTime=nowTime)
  682. smqs = Store_Meal.objects.filter(commodity_code=commodity_code). \
  683. values("day", "bucket_id", "bucket__storeDay")
  684. bucketId = smqs[0]['bucket_id']
  685. if not smqs.exists():
  686. return response.json(0, '套餐已删除')
  687. addTime = int(smqs[0]["day"]) * 24 * 3600
  688. ubqs = UID_Bucket.objects.filter(uid=UID, channel=channel). \
  689. values("bucket_id", "endTime", "bucket__storeDay")
  690. nowTime = int(time.time())
  691. if ubqs.exists():
  692. ubqs_count = ubqs.count()
  693. ubq = ubqs[ubqs_count - 1, ubqs_count]
  694. new_starTime = ubq['endTime'] + 1
  695. ub_cqs = UID_Bucket.objects.create(uid=UID, channel=channel, bucket_id=bucketId,
  696. endTime=new_starTime + addTime)
  697. uid_bucket_id = ub_cqs.id
  698. else:
  699. ub_cqs = UID_Bucket.objects.create(uid=UID, channel=channel, bucket_id=bucketId,
  700. endTime=nowTime + addTime)
  701. uid_bucket_id = ub_cqs.id
  702. order_qs.update \
  703. (payTime=nowTime, status=1, updTime=nowTime, uid_bucket_id=uid_bucket_id)
  704. red_url = "{SERVER_DOMAIN}cloudstorage/payOK".format(SERVER_DOMAIN=SERVER_DOMAIN)
  705. return HttpResponseRedirect(red_url)
  706. return response.json(0, signature)
  707. def do_pay_by_paypal_callback(self, request_dict, response):
  708. paymentId = request_dict.get('paymentId', None)
  709. PayerID = request_dict.get('PayerID', None)
  710. orderID = request_dict.get('orderID', None)
  711. if not paymentId or not PayerID or not orderID:
  712. red_url = "{SERVER_DOMAIN}cloudstorage/payError".format(SERVER_DOMAIN=SERVER_DOMAIN)
  713. return HttpResponseRedirect(red_url)
  714. paypalrestsdk.configure(PAYPAL_CRD)
  715. # ID of the payment. This ID is provided when creating payment.
  716. payment = paypalrestsdk.Payment.find(paymentId)
  717. payres = payment.execute({"payer_id": PayerID})
  718. print(payres)
  719. if not payres:
  720. red_url = "{SERVER_DOMAIN}cloudstorage/payError".format(SERVER_DOMAIN=SERVER_DOMAIN)
  721. return HttpResponseRedirect(red_url)
  722. print("Payment execute successfully")
  723. order_qs = Order_Model.objects.filter(orderID=orderID)
  724. nowTime = int(time.time())
  725. order_list = order_qs.values("UID", "channel", "commodity_code")
  726. UID = order_list[0]['UID']
  727. channel = order_list[0]['channel']
  728. commodity_code = order_list[0]['commodity_code']
  729. smqs = Store_Meal.objects.filter(commodity_code=commodity_code). \
  730. values("day", "bucket_id", "bucket__storeDay")
  731. bucketId = smqs[0]['bucket_id']
  732. if not smqs.exists():
  733. return response.json(0, '套餐已删除')
  734. addTime = int(smqs[0]["day"]) * 24 * 3600
  735. # ##
  736. ubqs = UID_Bucket.objects.filter(uid=UID, channel=channel, endTime__gte=nowTime). \
  737. values("bucket__storeDay", "bucket__region", "endTime")
  738. if ubqs.exists():
  739. ubqs_count = ubqs.count()
  740. print(ubqs_count)
  741. ubq = ubqs[ubqs_count - 1]
  742. print(ubq)
  743. new_starTime = ubq['endTime'] + 1
  744. ub_cqs = UID_Bucket.objects.create \
  745. (uid=UID, channel=channel, bucket_id=bucketId,
  746. endTime=new_starTime + addTime)
  747. uid_bucket_id = ub_cqs.id
  748. else:
  749. ub_cqs = UID_Bucket.objects.create \
  750. (uid=UID, channel=channel, bucket_id=bucketId, endTime=nowTime + addTime)
  751. uid_bucket_id = ub_cqs.id
  752. order_qs.update(status=1, updTime=nowTime, uid_bucket_id=uid_bucket_id)
  753. # return response.json(0)
  754. red_url = "{SERVER_DOMAIN}cloudstorage/payOK".format(SERVER_DOMAIN=SERVER_DOMAIN)
  755. return HttpResponseRedirect(red_url)
  756. def do_create_pay_order(self, request_dict, userID, response):
  757. uid = request_dict.get('uid', None)
  758. channel = request_dict.get('channel', None)
  759. commodity_code = request_dict.get('commodity_code', None)
  760. pay_type = int(request_dict.get('pay_type', None))
  761. dv_qs = Device_Info.objects.filter(userID_id=userID, UID=uid, isShare=False, isExist=1)
  762. if not dv_qs.exists():
  763. return response.json(12)
  764. smqs = Store_Meal.objects.filter(commodity_code=commodity_code, type=pay_type). \
  765. values('currency', 'price', 'content', 'day', 'commodity_type', 'title', 'content')
  766. if not smqs.exists():
  767. return response.json(10, '套餐不存在')
  768. currency = smqs[0]['currency']
  769. price = smqs[0]['price']
  770. content = smqs[0]['content']
  771. day = smqs[0]['day']
  772. commodity_type = smqs[0]['commodity_type']
  773. # ubqs = UID_Bucket.objects.filter(uid=uid, channel=channel, endTime__gte=nowTime). \
  774. # values("bucket__storeDay", "bucket__region", "endTime")
  775. # if ubqs.exists():
  776. # ubqs_count = ubqs.count()
  777. # ubq = ubqs[ubqs_count - 1, ubqs_count]
  778. # new_starTime = ubq['endTime'] + 1
  779. nowTime = int(time.time())
  780. orderID = CommonService.createOrderID()
  781. if pay_type == 0:
  782. call_sub_url = "{SERVER_DOMAIN}cloudstorage/dopaypalcallback?orderID={orderID}". \
  783. format(SERVER_DOMAIN=SERVER_DOMAIN, orderID=orderID)
  784. # call_sub_url = "http://192.168.136.40:8077/cloudstorage/payExecute?orderID={orderID}".format(
  785. # SERVER_DOMAIN=SERVER_DOMAIN, orderID=orderID)
  786. call_clc_url = "{SERVER_DOMAIN}cloudstorage/payError".format(SERVER_DOMAIN=SERVER_DOMAIN)
  787. paypalrestsdk.configure(PAYPAL_CRD)
  788. payment = paypalrestsdk.Payment({
  789. "intent": "sale",
  790. "payer": {"payment_method": "paypal"},
  791. "redirect_urls": {"return_url": call_sub_url, "cancel_url": call_clc_url},
  792. "transactions": [{
  793. "item_list": {"items": [
  794. {"name": "Cloud video", "sku": "1", "price": price, "currency": "USD", "quantity": 1}]},
  795. "amount": {"total": price, "currency": currency},
  796. "description": content}]})
  797. if payment.create():
  798. print("Payment created successfully")
  799. else:
  800. print(payment.error)
  801. return response.json(10, payment.error)
  802. print(payment)
  803. for link in payment.links:
  804. if link.rel == "approval_url":
  805. approval_url = str(link.href)
  806. print("Redirect for approval: %s" % (approval_url))
  807. Order_Model.objects.create(orderID=orderID, UID=uid, channel=channel, userID_id=userID,
  808. desc=content, payType=pay_type, payTime=nowTime,
  809. price=price, currency=currency, addTime=nowTime, updTime=nowTime,
  810. endTime=(nowTime + int(day) * 3600 * 24), pay_url=approval_url,
  811. commodity_code=commodity_code, commodity_type=commodity_type)
  812. return response.json(0, {"redirectUrl": approval_url, "orderID": orderID})
  813. return response.json(10, 'generate_order_false')
  814. else:
  815. try:
  816. aliPayObj = AliPayObject()
  817. alipay = aliPayObj.conf()
  818. subject = smqs[0]['title'] + smqs[0]['content']
  819. order_string = alipay.api_alipay_trade_wap_pay(
  820. out_trade_no=orderID,
  821. total_amount=price,
  822. subject=subject,
  823. return_url="{SERVER_DOMAIN_SSL}cloudstorage/payOK".format(SERVER_DOMAIN_SSL=SERVER_DOMAIN_SSL),
  824. notify_url="{SERVER_DOMAIN_SSL}cloudstorage/doalicallback".format(
  825. SERVER_DOMAIN_SSL=SERVER_DOMAIN_SSL)
  826. # return_url="http://192.168.136.40/cloudstorage/payOK",
  827. # notify_url="http://192.168.136.40/cloudstorage/aliPayCallback"
  828. )
  829. except Exception as e:
  830. print(repr(e))
  831. return response.json(10, repr(e))
  832. else:
  833. if order_string:
  834. redirectUrl = aliPayObj.alipay_prefix + order_string
  835. Order_Model.objects.create(orderID=orderID, UID=uid, channel=channel, userID_id=userID,
  836. desc=content, payType=pay_type, payTime=nowTime,
  837. price=price, currency=currency, addTime=nowTime, updTime=nowTime,
  838. endTime=(nowTime + int(day) * 3600 * 24), pay_url=redirectUrl,
  839. commodity_code=commodity_code, commodity_type=commodity_type)
  840. return JsonResponse(status=200, data={'result_code': 0, 'reason': 'success',
  841. 'result': {"redirectUrl": redirectUrl, "orderID": orderID},
  842. 'error_code': 0})
  843. else:
  844. return response.json(10, '生成订单错误')