TestApi.py 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829
  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. import botocore
  15. from django.db import transaction
  16. from django.views.generic.base import View
  17. import os
  18. from Controller.DeviceConfirmRegion import Device_Region
  19. '''
  20. http://192.168.136.40:8077/Test
  21. '''
  22. import json
  23. import math
  24. import time
  25. import urllib
  26. import datetime
  27. from Object.AliPayObject import AliPayObject
  28. import logging
  29. import boto3
  30. from boto3.session import Session
  31. from botocore.exceptions import ClientError
  32. import oss2
  33. import paypalrestsdk
  34. import logging
  35. import requests
  36. import hashlib
  37. import hmac
  38. from aliyunsdkcore import client
  39. from aliyunsdksts.request.v20150401 import AssumeRoleRequest
  40. from django.http import JsonResponse, HttpResponseRedirect, HttpResponse
  41. from django.utils.decorators import method_decorator
  42. from django.views.decorators.csrf import csrf_exempt
  43. from django.views.generic.base import View
  44. from Object.RedisObject import RedisObject
  45. from django.contrib.auth.hashers import make_password, check_password # 对密码加密模块
  46. from Ansjer.config import OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET, OSS_ROLE_ARN, SERVER_DOMAIN, PAYPAL_CRD, \
  47. SERVER_DOMAIN_SSL, AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, SERVER_TYPE
  48. from Model.models import Device_Info, Order_Model, Store_Meal, VodHlsModel, OssCrdModel, UID_Bucket, CompanySerialModel, \
  49. SerialNumberModel, CompanyModel, VPGModel, Unused_Uid_Meal, StsCrdModel, ExperienceContextModel, DeviceLogModel
  50. from Object.ResponseObject import ResponseObject
  51. from Object.TokenObject import TokenObject
  52. from Object.UidTokenObject import UidTokenObject
  53. from Service.CommonService import CommonService
  54. from Service.ModelService import ModelService
  55. from Object.m3u8generate import PlaylistGenerator
  56. from Model.models import Device_User, Device_Info, UID_Bucket, UID_Preview, UidSetModel, UidPushModel
  57. SERVER_DOMAIN = 'http://test.dvema.com/'
  58. ACCESS_KEY = "AKIA2E67UIMD3CYTIWPA"
  59. SECRET_KEY = "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"
  60. # 测试接口sdk
  61. class testView(View):
  62. @method_decorator(csrf_exempt)
  63. def dispatch(self, *args, **kwargs):
  64. return super(testView, self).dispatch(*args, **kwargs)
  65. def get(self, request, *args, **kwargs):
  66. request.encoding = 'utf-8'
  67. operation = kwargs.get('operation')
  68. return self.validation(request.GET, request, operation)
  69. def post(self, request, *args, **kwargs):
  70. request.encoding = 'utf-8'
  71. operation = kwargs.get('operation')
  72. return self.validation(request.POST, request, operation)
  73. def put(self, request, *args, **kwargs):
  74. request.encoding = 'utf-8'
  75. operation = kwargs.get('operation')
  76. response = ResponseObject()
  77. return response.json(0,request.body)
  78. def validation(self, request_dict, request, operation):
  79. response = ResponseObject()
  80. # operation => cloudVod/path
  81. if operation is None:
  82. return response.json(444, 'error path')
  83. elif operation == 'tests':
  84. logger = logging.getLogger('info')
  85. logger.info('测试打印')
  86. res = make_password(123456)
  87. return JsonResponse(status=200, data=res,safe=False)
  88. elif operation == 'testMiddleWare':
  89. a = int('a')
  90. return JsonResponse(status=200, safe=False)
  91. elif operation == 'testMiddleWare2':
  92. try:
  93. a = int('a')
  94. return JsonResponse(status=200, safe=False)
  95. except Exception as e:
  96. return response.json(500, repr(e))
  97. elif operation == 'cbu':
  98. return self.createBucket()
  99. elif operation == 'vodList':
  100. ip = CommonService.get_ip_address(request)
  101. userID = '158943594633713800138000'
  102. return self.do_test_query_vod_list(userID, ip, request_dict, response)
  103. elif operation == 'signplaym3u8':
  104. return self.do_sign_play_m3u8(request_dict, response)
  105. elif operation == 'get_sign_sts':
  106. ip = CommonService.get_ip_address(request)
  107. return self.do_test_get_sign_sts(request_dict, ip, response)
  108. elif operation == 'eqi_query':
  109. userID = '158943594633713800138000'
  110. return self.queryInterface(request_dict, userID, response)
  111. elif operation == 'getAlexa':
  112. userID = '158943594633713800138000'
  113. return self.queryInterface(request_dict, userID, response)
  114. elif operation == 'generateToken':
  115. userID = '158943604783713800138000'
  116. return self.generate_token(request_dict,userID)
  117. elif operation == 'test_upload_s3':
  118. userID = '158943604783713800138000'
  119. return self.test_upload_s3(request_dict, response)
  120. elif operation == 'rekognition':
  121. userID = '158943604783713800138000'
  122. return self.testRekognition(request,request_dict)
  123. elif operation == 'deleteDevice':
  124. return self.delete_device(response)
  125. elif operation == 'ip':
  126. return self.ip(response)
  127. elif operation == 'configType':
  128. return self.configType(response)
  129. elif operation == 'createData':
  130. return self.createData(request_dict, response)
  131. else:
  132. return 123
  133. def createBucket(self):
  134. #查看桶列表
  135. # url = "https://azvod1.s3-ap-northeast-1.amazonaws.com"
  136. # session = Session(ACCESS_KEY, SECRET_KEY)
  137. # s3_client = session.client('s3', endpoint_url=url)
  138. # results = s3_client.list_buckets()
  139. # return JsonResponse(status=200, data={'code': 200, 'msg': results['Buckets']})
  140. # Create bucket
  141. session = Session(ACCESS_KEY, SECRET_KEY)
  142. s3_client = session.client('s3')
  143. #上传
  144. # 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())
  145. #下载
  146. resp = s3_client.get_object(Bucket="azvod1", Key="file/rule.txt")
  147. with open('local.txt', 'wb') as f:
  148. f.write(resp['Body'].read())
  149. # 新查询设备字段
  150. def queryInterface(self, request_dict, userID, response):
  151. page = request_dict.get('page', None)
  152. line = request_dict.get('line', None)
  153. NickName = request_dict.get('NickName', None)
  154. uid = request_dict.get('uid', None)
  155. page = int(page)
  156. line = int(line)
  157. response.lang = 'cn'
  158. userID = userID
  159. dvqs = Device_Info.objects.filter(userID_id=userID)
  160. # # 已重置的设备
  161. # dvqs = dvqs.filter(~Q(isExist=2))
  162. if NickName:
  163. dvqs = dvqs.filter(NickName__icontains=NickName)
  164. if uid:
  165. dvqs = dvqs.filter(UID=uid)
  166. # count = dvqs.count()
  167. #分页
  168. dvql = dvqs[(page - 1) * line:page * line].values('id', 'userID', 'NickName', 'UID', 'View_Account',
  169. 'View_Password', 'ChannelIndex', 'Type', 'isShare',
  170. 'primaryUserID', 'primaryMaster', 'data_joined',
  171. 'version',
  172. 'isVod', 'isExist', 'NotificationMode')
  173. dvls = CommonService.qs_to_list(dvql)
  174. uid_list = []
  175. for dvl in dvls:
  176. uid_list.append(dvl['UID'])
  177. # if dvl['isShare'] is False:
  178. # uid_list.append(dvl['UID'])
  179. ubqs = UID_Bucket.objects.filter(uid__in=uid_list). \
  180. values('bucket__content', 'status', 'channel', 'endTime', 'uid')
  181. upqs = UID_Preview.objects.filter(uid__in=uid_list).order_by('channel').values('id', 'uid', 'channel')
  182. auth = oss2.Auth(OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET)
  183. bucket = oss2.Bucket(auth, 'oss-cn-hongkong.aliyuncs.com', 'statres')
  184. nowTime = int(time.time())
  185. data = []
  186. # 设备拓展信息表
  187. us_qs = UidSetModel.objects.filter(uid__in=uid_list). \
  188. values('uid', 'version', 'nickname', 'detect_interval')
  189. uv_dict = {}
  190. for us in us_qs:
  191. uv_dict[us['uid']] = {'version': us['version'],
  192. 'nickname': us['nickname'],
  193. 'detect_interval': us['detect_interval']}
  194. for p in dvls:
  195. p['vod'] = []
  196. for dm in ubqs:
  197. if p['UID'] == dm['uid']:
  198. if dm['endTime'] > nowTime:
  199. p['vod'].append(dm)
  200. p['preview'] = []
  201. for up in upqs:
  202. if p['UID'] == up['uid']:
  203. obj = 'uid_preview/{uid}/channel_{channel}.png'. \
  204. format(uid=up['uid'], channel=up['channel'])
  205. img_sign = bucket.sign_url('GET', obj, 300)
  206. p['preview'].append(img_sign)
  207. p_uid = p['UID']
  208. if p_uid in uv_dict:
  209. # 设备版本号
  210. p['uid_version'] = uv_dict[p_uid]['version']
  211. p['detect_interval'] = uv_dict[p_uid]['detect_interval']
  212. # 设备昵称 调用影子信息昵称,先阶段不可
  213. if uv_dict[p_uid]['nickname']:
  214. p['NickName'] = uv_dict[p_uid]['nickname']
  215. else:
  216. # 设备版本号
  217. p['uid_version'] = ''
  218. data.append(p)
  219. return response.json(0, data)
  220. #获取播放列表
  221. def do_test_query_vod_list(self, userID, ip, request_dict, response):
  222. uid = 'GZL2PEFJPLY7W6BG111A'
  223. channel = 2
  224. userID = '158943594633713800138000'
  225. dv_qs = Device_Info.objects.filter(UID=uid, userID_id=userID, isShare=False)
  226. if not dv_qs.exists():
  227. return response.json(12)
  228. vod_play_list = []
  229. bucket_name = 'azvod1'
  230. aws_access_key_id = 'AKIA2E67UIMD45Y3HL53'
  231. aws_secret_access_key = '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'
  232. session = Session(
  233. aws_access_key_id=aws_access_key_id,
  234. aws_secret_access_key=aws_secret_access_key,
  235. region_name='ap-northeast-1'
  236. )
  237. conn = session.client('s3')
  238. thumbspng = '{uid}/vod{channel}/{time}/thumbs.png'. \
  239. format(uid=uid, channel=channel, time=1590485548)
  240. response_url = conn.generate_presigned_url(
  241. 'get_object',
  242. Params={
  243. 'Bucket': bucket_name,
  244. 'Key': thumbspng
  245. },
  246. ExpiresIn=3600
  247. )
  248. # m3u8 = '{uid}/vod{channel}/{time}/{time}.m3u8'. \
  249. # format(uid=uid, channel=channel, time=vod['time'])
  250. thumb_url = response_url
  251. vod_url = 'http://192.168.136.191:8000/testApi/signplaym3u8?' \
  252. 'uid={uid}&channel={channel}&time={time}&sign=tktktktk'. \
  253. format(ip=ip, uid=uid, channel=channel, time=1590485548)
  254. vod_play_list.append({
  255. 'name': 1590485548,
  256. 'sign_url': vod_url,
  257. 'thumb': thumb_url,
  258. 'sec': 12})
  259. return response.json(0, vod_play_list)
  260. #生成m3u8列表
  261. def do_sign_play_m3u8(self, request_dict, response):
  262. uid = 'GZL2PEFJPLY7W6BG111A'
  263. channel = 2
  264. storeTime = 1591344070
  265. fg = 6
  266. bucket__region = 'ap-northeast-1'
  267. bucket_name = 'azvod1'
  268. aws_access_key_id = 'AKIA2E67UIMD45Y3HL53'
  269. aws_secret_access_key = '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'
  270. session = Session(
  271. aws_access_key_id=aws_access_key_id,
  272. aws_secret_access_key=aws_secret_access_key,
  273. region_name=bucket__region
  274. )
  275. '''
  276. http://test.dvema.com/cloudstorage/signplaym3u8?uid=VVDHCVBYDKFMJRWA111A&channel=1&time=1586940120&sign=tktktktk
  277. '''
  278. conn = session.client('s3')
  279. playlist_entries = []
  280. for i in range(fg):
  281. thumbspng = '{uid}/vod{channel}/{time}/ts{i}.ts'. \
  282. format(uid=uid, channel=channel, time=storeTime, i=i)
  283. response_url = conn.generate_presigned_url(
  284. 'get_object',
  285. Params={
  286. 'Bucket': bucket_name,
  287. 'Key': thumbspng
  288. },
  289. ExpiresIn=86400
  290. )
  291. # m3u8 = '{uid}/vod{channel}/{time}/{time}.m3u8'. \
  292. # format(uid=uid, channel=channel, time=vod['time'])
  293. playlist_entries.append({
  294. 'name': response_url,
  295. 'duration': 10,
  296. })
  297. playlist = PlaylistGenerator(playlist_entries).generate()
  298. response = HttpResponse(playlist)
  299. response['Content-Type'] = 'application/octet-stream'
  300. response['Content-Disposition'] = 'attachment;filename="play.m3u8"'
  301. return response
  302. # return HttpResponse(playlist)
  303. # response = HttpResponse(playlist, content_type="application/vnd.apple.mpegurl")
  304. # # response = HttpResponse(playlist, content_type="application/octet-stream")
  305. # return response
  306. # return HttpResponse(status=200, content=playlist)
  307. def do_test_get_sign_sts(self, request_dict, ip, response):
  308. # uid = 'GZL2PEFJPLY7W6BG111A'
  309. # channel = 2
  310. uid = 'VVDHCVBYDKFMJRWA111A'
  311. channel = 1
  312. now_time = int(time.time())
  313. ubqs = UID_Bucket.objects.filter(uid=uid, channel=channel, endTime__gte=now_time). \
  314. values("bucket__mold", "bucket__bucket", "bucket__endpoint", "bucket__region", "endTime")
  315. if ubqs.exists():
  316. if ubqs[0]["bucket__mold"] == 0:
  317. # 阿里云 oss sts
  318. oc_qs = OssCrdModel.objects.filter(uid=uid, channel=channel). \
  319. values("addTime", "data")
  320. if oc_qs.exists():
  321. endTime = int(oc_qs[0]["addTime"]) + 3500
  322. if endTime > now_time:
  323. print(endTime)
  324. print(now_time)
  325. res = json.loads(oc_qs[0]["data"])
  326. return JsonResponse(status=200, data=res)
  327. # 套餐id
  328. storage = '{uid}/vod{channel}/'.format(uid=uid, channel=channel)
  329. bucket_name = ubqs[0]['bucket__bucket']
  330. endpoint = ubqs[0]['bucket__endpoint']
  331. access_key_id = OSS_STS_ACCESS_KEY
  332. access_key_secret = OSS_STS_ACCESS_SECRET
  333. region_id = ubqs[0]['bucket__region']
  334. role_arn = OSS_ROLE_ARN
  335. clt = client.AcsClient(access_key_id, access_key_secret, region_id)
  336. req = AssumeRoleRequest.AssumeRoleRequest()
  337. # 设置返回值格式为JSON。
  338. req.set_accept_format('json')
  339. req.set_RoleArn(role_arn)
  340. req.set_RoleSessionName(uid + '_' + channel)
  341. req.set_DurationSeconds(3600)
  342. Resource_access = "acs:oss:*:*:{bucket_name}/{uid_channel}*". \
  343. format(bucket_name=bucket_name, uid_channel=storage)
  344. print(Resource_access)
  345. policys = {
  346. "Version": "1",
  347. "Statement": [
  348. {
  349. "Action": ["oss:PutObject", "oss:DeleteObject", ],
  350. "Resource": [Resource_access],
  351. "Effect": "Allow",
  352. "Condition": {
  353. "IpAddress": {"acs:SourceIp": ip}
  354. # "IpAddress": {"acs:SourceIp": "120.237.157.184"}
  355. # "IpAddress": {"acs:SourceIp": "*"}
  356. }
  357. }
  358. ]
  359. }
  360. req.set_Policy(Policy=json.dumps(policys))
  361. body = clt.do_action(req)
  362. # 使用RAM账号的AccessKeyId和AccessKeySecret向STS申请临时token。
  363. token = json.loads(body.decode('utf-8'))
  364. print(token)
  365. res = {
  366. 'AccessKeyId': token['Credentials']['AccessKeyId'],
  367. 'AccessKeySecret': token['Credentials']['AccessKeySecret'],
  368. 'SecurityToken': token['Credentials']['SecurityToken'],
  369. 'Expiration': token['Credentials']['Expiration'],
  370. 'expire': '3600',
  371. 'endpoint': endpoint,
  372. 'bucket_name': bucket_name,
  373. 'arn': token['AssumedRoleUser']['Arn'],
  374. 'code': 0,
  375. 'storage': storage,
  376. 'endTime': ubqs[0]['endTime'],
  377. 'ip': ip}
  378. if oc_qs.exists():
  379. oc_qs.update(data=json.dumps(res), addTime=now_time)
  380. else:
  381. OssCrdModel.objects.create \
  382. (uid=uid, channel=channel, data=json.dumps(res), addTime=now_time)
  383. return JsonResponse(status=200, data=res)
  384. elif ubqs[0]["bucket__mold"] == 1:
  385. # 亚马逊 s3 sts
  386. sts_qs = StsCrdModel.objects.filter(uid=uid, channel=channel). \
  387. values("addTime", "data")
  388. if sts_qs.exists():
  389. endTime = int(sts_qs[0]["addTime"]) + 3500
  390. if endTime > now_time:
  391. print(endTime)
  392. print(now_time)
  393. res = json.loads(sts_qs[0]["data"])
  394. return JsonResponse(status=200, data=res)
  395. # 套餐id
  396. storage = '{uid}/vod{channel}/'.format(uid=uid, channel=channel)
  397. bucket_name = ubqs[0]['bucket__bucket']
  398. endpoint = ubqs[0]['bucket__endpoint']
  399. region_id = ubqs[0]['bucket__region']
  400. ###############
  401. REGION_NAME = region_id
  402. boto3_sts = boto3.client(
  403. 'sts',
  404. aws_access_key_id='AKIA2E67UIMD45Y3HL53',
  405. aws_secret_access_key='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',
  406. region_name=REGION_NAME
  407. )
  408. Policy = {
  409. "Version": "2012-10-17",
  410. "Statement": [
  411. {
  412. "Effect": "Allow",
  413. "Action": "s3:*",
  414. "Resource": ["arn:aws:s3:::azvod1/{uid_channel}/*".
  415. format(uid_channel=storage)]
  416. }
  417. ]
  418. }
  419. response = boto3_sts.get_federation_token(
  420. Name='{role_name}'.format(role_name=uid + '_' + channel),
  421. Policy=json.dumps(Policy),
  422. DurationSeconds=7200
  423. )
  424. ###############
  425. res = {
  426. 'AccessKeyId': response['Credentials']['AccessKeyId'],
  427. 'AccessKeySecret': response['Credentials']['SecretAccessKey'],
  428. 'SessionToken': response['Credentials']['SessionToken'],
  429. 'Expiration': response['Credentials']['Expiration'],
  430. 'expire': '3600',
  431. 'endpoint': endpoint,
  432. 'bucket_name': bucket_name,
  433. 'arn': response['FederatedUser']['Arn'],
  434. 'code': 0,
  435. 'storage': storage,
  436. 'endTime': ubqs[0]['endTime'],
  437. 'ip': ip,
  438. }
  439. if sts_qs.exists():
  440. sts_qs.update(data=json.dumps(res, default=str), addTime=now_time)
  441. else:
  442. StsCrdModel.objects.create(uid=uid, channel=channel, data=json.dumps(res, default=str),
  443. addTime=now_time, type=1)
  444. return JsonResponse(status=200, data=res)
  445. else:
  446. res = {'code': 404, 'msg': 'data not exists!'}
  447. return HttpResponse(json.dumps(res, ensure_ascii=False),
  448. content_type="application/json,charset=utf-8")
  449. def do_get_sign_sts(self, request_dict, ip, response):
  450. # uid = 'GZL2PEFJPLY7W6BG111A'
  451. # channel = 2
  452. uid = '86YC8Z192VB1VMKU111A'
  453. channel = 1
  454. now_time = int(time.time())
  455. ubqs = UID_Bucket.objects.filter(uid=uid, channel=channel, endTime__gte=now_time). \
  456. values("bucket__mold", "bucket__bucket", "bucket__endpoint", "bucket__region", "endTime")
  457. if ubqs.exists():
  458. if ubqs[0]["bucket__mold"] == 0:
  459. # 阿里云 oss sts
  460. oc_qs = OssCrdModel.objects.filter(uid=uid, channel=channel). \
  461. values("addTime", "data")
  462. if oc_qs.exists():
  463. endTime = int(oc_qs[0]["addTime"]) + 3500
  464. if endTime > now_time:
  465. print(endTime)
  466. print(now_time)
  467. res = json.loads(oc_qs[0]["data"])
  468. return JsonResponse(status=200, data=res)
  469. # 套餐id
  470. storage = '{uid}/vod{channel}/'.format(uid=uid, channel=channel)
  471. bucket_name = ubqs[0]['bucket__bucket']
  472. endpoint = ubqs[0]['bucket__endpoint']
  473. access_key_id = OSS_STS_ACCESS_KEY
  474. access_key_secret = OSS_STS_ACCESS_SECRET
  475. region_id = ubqs[0]['bucket__region']
  476. role_arn = OSS_ROLE_ARN
  477. clt = client.AcsClient(access_key_id, access_key_secret, region_id)
  478. req = AssumeRoleRequest.AssumeRoleRequest()
  479. # 设置返回值格式为JSON。
  480. req.set_accept_format('json')
  481. req.set_RoleArn(role_arn)
  482. req.set_RoleSessionName(uid + '_' + channel)
  483. req.set_DurationSeconds(3600)
  484. Resource_access = "acs:oss:*:*:{bucket_name}/{uid_channel}*". \
  485. format(bucket_name=bucket_name, uid_channel=storage)
  486. print(Resource_access)
  487. policys = {
  488. "Version": "1",
  489. "Statement": [
  490. {
  491. "Action": ["oss:PutObject", "oss:DeleteObject", ],
  492. "Resource": [Resource_access],
  493. "Effect": "Allow",
  494. "Condition": {
  495. "IpAddress": {"acs:SourceIp": ip}
  496. # "IpAddress": {"acs:SourceIp": "120.237.157.184"}
  497. # "IpAddress": {"acs:SourceIp": "*"}
  498. }
  499. }
  500. ]
  501. }
  502. req.set_Policy(Policy=json.dumps(policys))
  503. body = clt.do_action(req)
  504. # 使用RAM账号的AccessKeyId和AccessKeySecret向STS申请临时token。
  505. token = json.loads(body.decode('utf-8'))
  506. print(token)
  507. res = {
  508. 'AccessKeyId': token['Credentials']['AccessKeyId'],
  509. 'AccessKeySecret': token['Credentials']['AccessKeySecret'],
  510. 'SecurityToken': token['Credentials']['SecurityToken'],
  511. 'Expiration': token['Credentials']['Expiration'],
  512. 'expire': '3600',
  513. 'endpoint': endpoint,
  514. 'bucket_name': bucket_name,
  515. 'arn': token['AssumedRoleUser']['Arn'],
  516. 'code': 0,
  517. 'storage': storage,
  518. 'endTime': ubqs[0]['endTime'],
  519. 'ip': ip}
  520. if oc_qs.exists():
  521. oc_qs.update(data=json.dumps(res), addTime=now_time)
  522. else:
  523. OssCrdModel.objects.create \
  524. (uid=uid, channel=channel, data=json.dumps(res), addTime=now_time)
  525. return JsonResponse(status=200, data=res)
  526. elif ubqs[0]["bucket__mold"] == 1:
  527. # 亚马逊 s3 sts
  528. sts_qs = StsCrdModel.objects.filter(uid=uid, channel=channel). \
  529. values("addTime", "data")
  530. if sts_qs.exists():
  531. endTime = int(sts_qs[0]["addTime"]) + 3500
  532. if endTime > now_time:
  533. print(endTime)
  534. print(now_time)
  535. res = json.loads(sts_qs[0]["data"])
  536. return JsonResponse(status=200, data=res)
  537. # 套餐id
  538. storage = '{uid}/vod{channel}/'.format(uid=uid, channel=channel)
  539. bucket_name = ubqs[0]['bucket__bucket']
  540. endpoint = ubqs[0]['bucket__endpoint']
  541. region_id = ubqs[0]['bucket__region']
  542. ###############
  543. REGION_NAME = region_id
  544. boto3_sts = boto3.client(
  545. 'sts',
  546. aws_access_key_id='AKIA2E67UIMD45Y3HL53',
  547. aws_secret_access_key='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',
  548. region_name=REGION_NAME
  549. )
  550. Policy = {
  551. "Version": "2012-10-17",
  552. "Statement": [
  553. {
  554. "Effect": "Allow",
  555. "Action": "s3:*",
  556. "Resource": ["arn:aws:s3:::azvod1/{uid_channel}/*".
  557. format(uid_channel=storage)]
  558. }
  559. ]
  560. }
  561. response = boto3_sts.get_federation_token(
  562. Name='{role_name}'.format(role_name=uid + '_' + channel),
  563. Policy=json.dumps(Policy),
  564. DurationSeconds=7200
  565. )
  566. ###############
  567. res = {
  568. 'AccessKeyId': response['Credentials']['AccessKeyId'],
  569. 'AccessKeySecret': response['Credentials']['SecretAccessKey'],
  570. 'SessionToken': response['Credentials']['SessionToken'],
  571. 'Expiration': response['Credentials']['Expiration'],
  572. 'expire': '3600',
  573. 'endpoint': endpoint,
  574. 'bucket_name': bucket_name,
  575. 'arn': response['FederatedUser']['Arn'],
  576. 'code': 0,
  577. 'storage': storage,
  578. 'endTime': ubqs[0]['endTime'],
  579. 'ip': ip,
  580. }
  581. if sts_qs.exists():
  582. sts_qs.update(data=json.dumps(res, default=str), addTime=now_time)
  583. else:
  584. StsCrdModel.objects.create(uid=uid, channel=channel, data=json.dumps(res, default=str),
  585. addTime=now_time, type=1)
  586. return JsonResponse(status=200, data=res)
  587. else:
  588. res = {'code': 404, 'msg': 'data not exists!'}
  589. return HttpResponse(json.dumps(res, ensure_ascii=False),
  590. content_type="application/json,charset=utf-8")
  591. def do_pay_by_ali(self, request_dict, userID, response):
  592. uid = request_dict.get('uid', None)
  593. rank = request_dict.get('rank', None)
  594. channel = request_dict.get('channel', None)
  595. qs = Device_Info.objects.filter(userID_id=userID, UID=uid, isShare=False)
  596. if not qs.exists():
  597. return response.json(12)
  598. if not channel or not rank:
  599. return response.json(444, 'channel,rank')
  600. smqs = Store_Meal.objects.filter(id=rank). \
  601. values("currency", "price", "content", "day", "bucket__storeDay", "bucket__region", "type")
  602. if not smqs.exists():
  603. # 套餐不存在
  604. return response.json(173)
  605. if smqs[0]['type'] != 1:
  606. return response.json(10, '不支持支付宝支付')
  607. currency = smqs[0]['currency']
  608. price = smqs[0]['price']
  609. content = smqs[0]['content']
  610. day = smqs[0]['day']
  611. nowTime = int(time.time())
  612. ubqs = UID_Bucket.objects.filter(uid=uid, channel=channel, endTime__gte=nowTime). \
  613. values("bucket__storeDay", "bucket__region")
  614. if ubqs.exists():
  615. if ubqs[0]['bucket__region'] != smqs[0]['bucket__region']:
  616. return response.json(712) # 区域不一致
  617. elif ubqs[0]['bucket__storeDay'] != smqs[0]['bucket__storeDay']:
  618. return response.json(713) # 不可更改套餐
  619. # 续费流程
  620. nowTime = int(time.time())
  621. # 新增流程
  622. orderID = CommonService.createOrderID()
  623. try:
  624. aliPayObj = AliPayObject()
  625. alipay = aliPayObj.conf()
  626. order_string = alipay.api_alipay_trade_wap_pay(
  627. out_trade_no=orderID,
  628. total_amount=price,
  629. subject="测试哟",
  630. return_url="{SERVER_DOMAIN_SSL}cloudVod/payOK".format(SERVER_DOMAIN_SSL=SERVER_DOMAIN_SSL),
  631. notify_url="{SERVER_DOMAIN_SSL}cloudVod/aliPayCallback".format(SERVER_DOMAIN_SSL=SERVER_DOMAIN_SSL)
  632. # return_url="http://192.168.136.40/cloudVod/payOK",
  633. # notify_url="http://192.168.136.40/cloudVod/aliPayCallback"
  634. )
  635. except Exception as e:
  636. print(repr(e))
  637. return response.json(10, repr(e))
  638. if order_string:
  639. redirectUrl = aliPayObj.alipay_prefix + order_string
  640. Order_Model.objects.create(orderID=orderID, UID=uid, channel=channel, userID_id=userID, desc=content,
  641. price=price, currency=currency, addTime=nowTime, updTime=nowTime,
  642. endTime=nowTime + int(day) * 3600 * 24, rank_id=rank, payType=1)
  643. return JsonResponse(status=200,
  644. data={'result_code': 0, 'reason': 'success',
  645. 'result': {"redirectUrl": redirectUrl, "orderID": orderID},
  646. 'error_code': 0})
  647. else:
  648. return response.json(10, '生成订单错误')
  649. def do_filter_playlist(self, request_dict, userID, response):
  650. startTime = int(request_dict.get('startTime', None))
  651. endTime = int(request_dict.get('endTime', None))
  652. uid = request_dict.get('uid', None)
  653. channel = request_dict.get('channel', None)
  654. dvqs = Device_Info.objects.filter(UID=uid, userID_id=userID, isShare=False)
  655. if not dvqs.exists():
  656. return response.json(12)
  657. ubqs = UID_Bucket.objects.filter(uid=uid, channel=channel).values('status')
  658. if not ubqs.exists():
  659. return response.json(10, '设备未购买')
  660. nowTime = int(time.time())
  661. # vodqs = VodHlsModel.objects.filter(uid=uid, channel=channel, time__range=(startTime, endTime),
  662. # endTime__gte=nowTime) \
  663. # .values("time", "sec", "bucket__bucket", "bucket__endpoint", "bucket__region")
  664. vodqs = VodHlsModel.objects.filter(uid=uid, channel=channel) \
  665. .values("time", "sec", "bucket__bucket", "bucket__endpoint", "bucket__region")
  666. vod_play_list = []
  667. print(vodqs)
  668. auth = oss2.Auth(OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET)
  669. for vod in vodqs:
  670. bucket_name = vod["bucket__bucket"]
  671. endpoint = vod["bucket__endpoint"]
  672. bucket = oss2.Bucket(auth, endpoint, bucket_name)
  673. m3u8 = '{uid}/vod{channel}/{time}/{time}.m3u8'. \
  674. format(uid=uid, channel=channel, time=vod['time'])
  675. ts = '{uid}/vod{channel}/{time}/ts0.ts'. \
  676. format(uid=uid, channel=channel, time=vod['time'])
  677. url = bucket.sign_url('GET', m3u8, 3600, params={'x-oss-process': 'hls/sign'})
  678. urllst = url.split('?')
  679. url_start = urllib.parse.unquote(urllst[0])
  680. url_end = urllst[1]
  681. vod_play_url = '{url_start}?{url_end}'. \
  682. format(url_start=url_start, url_end=url_end)
  683. thumb = bucket.sign_url('GET', ts, 3600,
  684. params={'x-oss-process': 'video/snapshot,t_10000,m_fast,w_300'})
  685. vod_play_list.append({'name': vod['time'], 'sign_url': vod_play_url, 'thumb': thumb, 'sec': vod['sec']})
  686. return response.json(0, vod_play_list)
  687. def generate_token(self,request_dict,userID):
  688. # UserIdToken
  689. tko = TokenObject()
  690. res = tko.generate(
  691. data={'userID': 162156677434713800138000, 'lang': 'cn', 'user': '3076898361@qq.com', 'm_code': '12341324321'})
  692. #uidToken
  693. # utko = UidTokenObject()
  694. # res = utko.generate(data={'uid': '4UZSEDP93MJ3X7YB111A','channel': 1})
  695. # from Object.ETkObject import ETkObject
  696. # etkObj = ETkObject(etk='')
  697. # res = etkObj.encrypt("4UZSEDP93MJ3X7YB111A")
  698. return JsonResponse(status=200, data=res,safe=False)
  699. def test_upload_s3(self,request_dict , response):
  700. aws_s3_guonei = boto3.client(
  701. 's3',
  702. aws_access_key_id=AWS_ACCESS_KEY_ID[0],
  703. aws_secret_access_key=AWS_SECRET_ACCESS_KEY[0],
  704. config=botocore.client.Config(signature_version='s3v4'),
  705. region_name='cn-northwest-1'
  706. )
  707. download_link = 'ipctest'
  708. response_url = aws_s3_guonei.generate_presigned_url(
  709. ClientMethod='put_object',
  710. Params={
  711. 'Bucket': 'pc-package',
  712. 'Key': download_link
  713. },
  714. ExpiresIn=3600
  715. )
  716. return response.json(0, {'datas': response_url, 'count': 1})
  717. def testRekognition(self,request,request_dict):
  718. # ip = CommonService.get_ip_address(request)
  719. # ipInfo = CommonService.getIpIpInfo(ip,"CN")
  720. # # print(type(ipInfo))
  721. # # exit(ipInfo)
  722. # addr = CommonService.getAddr(ip)
  723. # dicts = {
  724. # 'ipInfo':ipInfo,
  725. # 'addr':addr,
  726. # }
  727. # return HttpResponse(json.dumps(dicts, ensure_ascii=False),
  728. # content_type="application/json,charset=utf-8")
  729. # client = boto3.client('s3', aws_access_key_id='AKIA2E67UIMD45Y3HL53',aws_secret_access_key='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',region_name='us-east-1')
  730. # exit(request.FILES)
  731. files = request.FILES.get('image')
  732. labels = int(request_dict.get('labels',5))
  733. minConfidence = int(request_dict.get('minConfidence',99))
  734. if not files:
  735. return HttpResponse('请上传图片!!!!')
  736. client = boto3.client('rekognition', aws_access_key_id='AKIA2E67UIMD6JD6TN3J',aws_secret_access_key='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',region_name='us-east-1')
  737. # image = open('E:/photo/a615fa40b8c476bab0f6eeb332e62a5a-1000.jpg', "rb")
  738. response = client.detect_labels(Image={'Bytes':files.read()},MaxLabels=labels,MinConfidence=minConfidence)
  739. # for obj in response['Labels']:
  740. # exit(obj)
  741. # if obj['Name'] == 'Person':
  742. # jsonstr = json.dumps(obj)
  743. return HttpResponse(json.dumps(response, ensure_ascii=False),
  744. content_type="application/json,charset=utf-8")
  745. def delete_device(self, response):
  746. # 定时删除测试账号下的设备数据
  747. try:
  748. userID_list = ['tech01@ansjer.com', 'tech02@ansjer.com', 'tech03@ansjer.com', 'tech04@ansjer.com',
  749. 'tech05@ansjer.com', 'tech06@ansjer.com', 'tech07@ansjer.com', 'tech08@ansjer.com',
  750. 'tech09@ansjer.com', 'tech10@ansjer.com', 'fix01@ansjer.com', 'fix02@ansjer.com',
  751. 'fix03@ansjer.com', 'fix04@ansjer.com', 'fix05@ansjer.com']
  752. device_user = Device_User.objects.filter(username__in=userID_list)
  753. device_info_qs = Device_Info.objects.filter(userID__in=device_user).values('UID')
  754. uid_list = []
  755. for device_info in device_info_qs:
  756. uid_list.append(device_info['UID'])
  757. with transaction.atomic():
  758. # 删除设备云存相关数据
  759. UidSetModel.objects.filter(uid__in=uid_list).delete()
  760. UID_Bucket.objects.filter(uid__in=uid_list).delete()
  761. Unused_Uid_Meal.objects.filter(uid__in=uid_list).delete()
  762. Order_Model.objects.filter(UID__in=uid_list).delete()
  763. StsCrdModel.objects.filter(uid__in=uid_list).delete()
  764. VodHlsModel.objects.filter(uid__in=uid_list).delete()
  765. ExperienceContextModel.objects.filter(uid__in=uid_list).delete()
  766. Device_Info.objects.filter(userID__in=device_user).delete()
  767. return response.json(0)
  768. except Exception as e:
  769. return response.json(500, repr(e))
  770. def ip(self, response):
  771. ip = '67.220.90.13'
  772. country_id = Device_Region().get_device_region(ip)
  773. print(country_id)
  774. return response.json(0)
  775. def configType(self, response):
  776. print(SERVER_TYPE)
  777. return response.json(0)
  778. def createData(self, request_dict, response):
  779. uid = request_dict.get('uid', None)
  780. # filter_date = datetime.datetime.now()-datetime.timedelta(days=4)
  781. # qs = DeviceLogModel.objects.filter(add_time__lt=filter_date)
  782. DeviceLogModel.objects.create(uid=uid)
  783. return response.json(0)