TestApi.py 40 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901
  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 os
  15. import traceback
  16. import botocore
  17. import cv2
  18. from botocore import client
  19. from Ansjer.cn_config.config_formal import CONFIG_INFO
  20. from Controller.DeviceConfirmRegion import Device_Region
  21. from Object.AWS.AmazonS3Util import AmazonS3Util
  22. from Object.RedisObject import RedisObject
  23. from Service.VodHlsService import SplitVodHlsObject
  24. '''
  25. http://192.168.136.40:8077/Test
  26. '''
  27. import json
  28. import time
  29. import urllib
  30. import requests
  31. from Object.AliPayObject import AliPayObject
  32. import boto3
  33. from boto3.session import Session
  34. import oss2
  35. import paypalrestsdk
  36. import logging
  37. from django.http import JsonResponse, HttpResponse
  38. from django.utils.decorators import method_decorator
  39. from django.views.decorators.csrf import csrf_exempt
  40. from django.views.generic.base import View
  41. from django.contrib.auth.hashers import make_password # 对密码加密模块
  42. from Ansjer.config import OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET, OSS_ROLE_ARN, AWS_ACCESS_KEY_ID, \
  43. AWS_SECRET_ACCESS_KEY, SERVER_TYPE, AWS_SES_ACCESS_REGION
  44. from Model.models import Order_Model, Store_Meal, DeviceLogModel, VodBucketModel, \
  45. UIDCompanySerialModel, CompanySerialModel
  46. from Object.ResponseObject import ResponseObject
  47. from Object.TokenObject import TokenObject
  48. from Service.CommonService import CommonService
  49. from Object.m3u8generate import PlaylistGenerator
  50. from Model.models import Device_Info, UID_Bucket, UID_Preview, UidSetModel
  51. from Ansjer.config import SERVER_DOMAIN_SSL
  52. SERVER_DOMAIN = 'http://test.dvema.com/'
  53. ACCESS_KEY = "AKIA2E67UIMD3CYTIWPA"
  54. SECRET_KEY = "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"
  55. LOGGER = logging.getLogger('info')
  56. # 测试接口sdk
  57. class testView(View):
  58. @method_decorator(csrf_exempt)
  59. def dispatch(self, *args, **kwargs):
  60. return super(testView, self).dispatch(*args, **kwargs)
  61. def get(self, request, *args, **kwargs):
  62. request.encoding = 'utf-8'
  63. operation = kwargs.get('operation')
  64. return self.validation(request.GET, request, operation)
  65. def post(self, request, *args, **kwargs):
  66. request.encoding = 'utf-8'
  67. operation = kwargs.get('operation')
  68. return self.validation(request.POST, request, operation)
  69. def put(self, request, *args, **kwargs):
  70. request.encoding = 'utf-8'
  71. operation = kwargs.get('operation')
  72. response = ResponseObject()
  73. return response.json(0, request.body)
  74. def validation(self, request_dict, request, operation):
  75. response = ResponseObject()
  76. # operation => cloudVod/path
  77. if operation is None:
  78. return response.json(444, 'error path')
  79. elif operation == 'tests':
  80. logger = logging.getLogger('info')
  81. logger.info('测试打印')
  82. res = make_password(123456)
  83. return JsonResponse(status=200, data=res, safe=False)
  84. elif operation == 'testMiddleWare':
  85. a = int('a')
  86. return JsonResponse(status=200, safe=False)
  87. elif operation == 'testMiddleWare2':
  88. try:
  89. a = int('a')
  90. return JsonResponse(status=200, safe=False)
  91. except Exception as e:
  92. return response.json(500, repr(e))
  93. elif operation == 'cbu':
  94. return self.createBucket()
  95. elif operation == 'vodList':
  96. ip = CommonService.get_ip_address(request)
  97. userID = '158943594633713800138000'
  98. return self.do_test_query_vod_list(userID, ip, request_dict, response)
  99. elif operation == 'signplaym3u8':
  100. return self.do_sign_play_m3u8(request_dict, response)
  101. elif operation == 'get_sign_sts':
  102. ip = CommonService.get_ip_address(request)
  103. return self.do_test_get_sign_sts(request_dict, ip, response)
  104. elif operation == 'eqi_query':
  105. userID = '158943594633713800138000'
  106. return self.queryInterface(request_dict, userID, response)
  107. elif operation == 'getAlexa':
  108. userID = '158943594633713800138000'
  109. return self.queryInterface(request_dict, userID, response)
  110. elif operation == 'generateToken':
  111. userID = '158943604783713800138000'
  112. return self.generate_token(request_dict, userID)
  113. elif operation == 'test_upload_s3':
  114. userID = '158943604783713800138000'
  115. return self.test_upload_s3(request_dict, response)
  116. elif operation == 'rekognition':
  117. userID = '158943604783713800138000'
  118. return self.testRekognition(request, request_dict)
  119. elif operation == 'ip':
  120. return self.ip(response)
  121. elif operation == 'configType':
  122. return self.configType(response)
  123. elif operation == 'createData':
  124. return self.createData(request_dict, response)
  125. elif operation == 'findPaypalOrder':
  126. return self.findPaypalOrder(request_dict, response)
  127. elif operation == 'comb':
  128. return self.do_comb(request_dict, response)
  129. elif operation == 'count_ts':
  130. return self.count_ts(request_dict, response)
  131. elif operation == 'upload-s3':
  132. return self.file_upload_s3(request, request_dict, response)
  133. elif operation == 'v2/upload-s3':
  134. return self.file_upload_s3_v2(request, request_dict, response)
  135. elif operation == 'download-s3':
  136. return self.file_download_s3(request_dict, response)
  137. elif operation == 'acl-put':
  138. return self.s3_acl_put(request_dict, response)
  139. elif operation == 's3-object-delete':
  140. return self.object_delete(request_dict, response)
  141. elif operation == 'head-bucket':
  142. return self.head_bucket(request_dict, response)
  143. elif operation == 'write_redis_list':
  144. return self.write_redis_list(response)
  145. elif operation == 'read_redis_list':
  146. return self.read_redis_list(response)
  147. elif operation == 'playM3u8':
  148. return self.play_m3u8(request_dict, response)
  149. elif operation == 'generate_video':
  150. return self.generate_video(request_dict, response)
  151. elif operation == 'getSerialNumberInfo': # 序列号信息查询
  152. return self.getSerialNumberInfo(request_dict, response)
  153. else:
  154. return 123
  155. @classmethod
  156. def generate_video(cls, request_dict, response):
  157. # 设计抽取图片规则通过消息随机还是时间随机,调试copy S3对象查询是否携带失效时间
  158. try:
  159. DIR = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
  160. arr_list = ['1666756086.jpeg', '1666831275.jpeg', '1666841492.jpeg']
  161. s3 = AmazonS3Util(AWS_ACCESS_KEY_ID[0], AWS_SECRET_ACCESS_KEY[0], 'cn-northwest-1')
  162. bucket = 'push-cloud-photo'
  163. for item in arr_list:
  164. path = DIR + r'\Ansjer\file\{}'.format(item)
  165. s3_key = 'HA154GVEDH41RY8Y111A/1/{}'.format(item)
  166. s3.download_object(bucket, s3_key, path)
  167. video_dir = DIR + r'\Ansjer\file\result.mp4' # 输出视频的保存路径
  168. fps = 0.5 # 帧率
  169. img_size = (1920, 1080) # 图片尺寸
  170. fourcc = cv2.VideoWriter_fourcc(*"mp4v")
  171. videoWriter = cv2.VideoWriter(video_dir, fourcc, fps, img_size)
  172. for i in arr_list:
  173. img_path = DIR + r'\Ansjer\file\{}'.format(i)
  174. frame = cv2.imread(img_path)
  175. frame = cv2.resize(frame, img_size) # 生成视频 图片尺寸和设定尺寸相同
  176. videoWriter.write(frame) # 将图片写进视频里
  177. os.remove(img_path)
  178. videoWriter.release() # 释放资源
  179. data = open(video_dir, 'rb')
  180. key = 'HA154GVEDH41RY8Y111A/1/20221027.mp4'
  181. s3.upload_file_obj(bucket, key, data)
  182. response_url = s3.generate_file_obj_url(bucket, key)
  183. os.remove(video_dir)
  184. except Exception as e:
  185. print(e)
  186. ex = traceback.format_exc()
  187. LOGGER.info('--->抽取推送图片异常:{}'.format(ex))
  188. return response.json(177, ex)
  189. return response.json(0, response_url)
  190. @classmethod
  191. def head_bucket(cls, request_dict, response):
  192. bucket_name = request_dict.get('bucket', None)
  193. s3 = AmazonS3Util('AKIA2E67UIMD45Y3HL53', 'ckYLg4Lo9ZXJIcJEAKkzf2rWvs8Xth1FCjqiAqUw', 'us-east-1')
  194. s3.bucket_exists(bucket_name)
  195. return response.json(0)
  196. @classmethod
  197. def file_upload_s3(cls, request, request_dict, response):
  198. file = request.FILES.get('file', None)
  199. file_name = file.name
  200. file_name = 'app/images/{}'.format(file_name)
  201. cls.upload_s3(file, file_name)
  202. return response.json(0)
  203. @classmethod
  204. def file_upload_s3_v2(cls, request, request_dict, response):
  205. """
  206. 'private' | 'public-read' | 'public-read-write' | 'authenticated-read'
  207. """
  208. file = request.FILES.get('file', None)
  209. file_name = file.name
  210. # S3下文件夹路径+文件名 组成对象key
  211. file_key = 'app/images/{}'.format(file_name)
  212. s3 = AmazonS3Util(AWS_ACCESS_KEY_ID[1], AWS_SECRET_ACCESS_KEY[1], AWS_SES_ACCESS_REGION)
  213. # 存储桶
  214. bucket = 'ansjerfilemanager'
  215. s3.upload_file_obj(bucket, file_key, file, {'ContentType': file.content_type, 'ACL': 'public-read'})
  216. return response.json(0)
  217. @classmethod
  218. def object_delete(cls, request_dict, response):
  219. file_name = request_dict.get('key', None)
  220. file_key = 'app/images/{}'.format(file_name)
  221. s3 = AmazonS3Util('AKIA2E67UIMD45Y3HL53', 'ckYLg4Lo9ZXJIcJEAKkzf2rWvs8Xth1FCjqiAqUw', 'us-east-1')
  222. bucket = "ansjerfilemanager"
  223. s3.delete_obj(bucket, file_key)
  224. return response.json(0)
  225. @staticmethod
  226. def upload_s3(data, upload_path):
  227. """
  228. 上传对象到对应存储桶
  229. @param data:
  230. @param upload_path:
  231. @return:
  232. """
  233. try:
  234. aws_key = "AKIA2E67UIMD45Y3HL53" # 【你的 aws_access_key】
  235. aws_secret = "ckYLg4Lo9ZXJIcJEAKkzf2rWvs8Xth1FCjqiAqUw" # 【你的 aws_secret_key】
  236. session = Session(aws_access_key_id=aws_key,
  237. aws_secret_access_key=aws_secret,
  238. region_name="us-east-1")
  239. s3 = session.resource("s3")
  240. # client = session.client("s3")
  241. bucket = "ansjerfilemanager" # 【你 bucket 的名字】 # 首先需要保.证 s3 上已经存在该存储桶,否则报错
  242. # upload_key = "test"
  243. s3.Bucket(bucket).put_object(Key=upload_path, Body=data)
  244. return True
  245. except Exception as e:
  246. print(repr(e))
  247. return False
  248. @classmethod
  249. def s3_acl_put(cls, request_dict, response):
  250. """
  251. S3通过存储桶对象设置Acl权限
  252. @param request_dict:
  253. @param response:
  254. @return:
  255. """
  256. try:
  257. key = request_dict.get('key', None)
  258. aws_key = "AKIA2E67UIMD45Y3HL53" # 【你的 aws_access_key】
  259. aws_secret = "ckYLg4Lo9ZXJIcJEAKkzf2rWvs8Xth1FCjqiAqUw" # 【你的 aws_secret_key】
  260. session = Session(aws_access_key_id=aws_key,
  261. aws_secret_access_key=aws_secret,
  262. region_name="us-east-1")
  263. s3 = session.resource("s3")
  264. # client = session.client("s3")
  265. bucket = "ansjerfilemanager" # 【你 bucket 的名字】 # 首先需要保.证 s3 上已经存在该存储桶,否则报错
  266. # upload_key = "test"
  267. obj = s3.Object(bucket, key)
  268. obj.Acl().put(ACL='public-read')
  269. return response.json(0)
  270. except Exception as e:
  271. print(repr(e))
  272. return False
  273. @classmethod
  274. def file_download_s3(cls, request_dict, response):
  275. """
  276. S3通过签名获取对象URL
  277. @param request_dict:
  278. @param response:
  279. @return:
  280. """
  281. key = request_dict.get('key', None)
  282. s3_client = AmazonS3Util('AKIA2E67UIMD45Y3HL53', 'ckYLg4Lo9ZXJIcJEAKkzf2rWvs8Xth1FCjqiAqUw', 'us-east-1')
  283. bucket = "ansjerfilemanager"
  284. response_url = s3_client.generate_file_obj_url(bucket, key)
  285. return response.json(0, response_url)
  286. def do_comb(self, request_dict, response):
  287. import itertools
  288. list1 = [1, 2, 3, 4]
  289. list2 = []
  290. for i in range(1, len(list1) + 1):
  291. iter = itertools.combinations(list1, i)
  292. list2.append(list(iter))
  293. list3 = []
  294. for list_ in list2:
  295. for val in list_:
  296. print('-------------list')
  297. print(type(val))
  298. comb = [str(i) for i in val]
  299. comb_int = int("".join(comb))
  300. list3.append(comb_int)
  301. return HttpResponse(json.dumps(list3))
  302. def findPaypalOrder(self, request_dict, response):
  303. PAYPAL_CRD = {
  304. "mode": "sandbox", # sandbox or live
  305. "client_id": "AVLoQVq3xHZ6FrF4mxHwlCPgVBAw4Fw5RtMkuxmYd23SkUTIY643n2g3KdK-Al8wV05I28lza5uoQbAA",
  306. "client_secret": "EO8kRc8yioDk0i2Qq-QMcVFfwkmyMJorTvBSLDTnxDJJ_wb9VoM_0jkUY9iEng2Flp1ze8wQOGpH5nB2"
  307. }
  308. paypalrestsdk.configure(PAYPAL_CRD)
  309. billing_agreement = paypalrestsdk.BillingAgreement.find('I-7P8BCCKE45HY')
  310. # json_str = json.dumps(billing_agreement, default=lambda o: o.__dict__)
  311. # return HttpResponse(json_str)
  312. print(repr(billing_agreement))
  313. return HttpResponse(repr(billing_agreement))
  314. return response.json(json.dumps(list(payment)))
  315. logger = logging.getLogger('info')
  316. logger.info('------记录aws bucket init时间')
  317. logger.info(int(time.time()))
  318. aws_access_key_id = AWS_ACCESS_KEY_ID[1]
  319. aws_secret_access_key = AWS_SECRET_ACCESS_KEY[1]
  320. session = Session(
  321. aws_access_key_id=aws_access_key_id,
  322. aws_secret_access_key=aws_secret_access_key,
  323. region_name='us-east-1'
  324. )
  325. conn = session.client('s3')
  326. logger.info(int(time.time()))
  327. exit()
  328. # related_resources = payment['transactions'][0]['related_resources']
  329. # if not related_resources:
  330. # return response.json(805)
  331. # sale = related_resources[0]['sale']
  332. # print(related_resources)
  333. # return HttpResponse(sale)
  334. def createBucket(self):
  335. # 查看桶列表
  336. # url = "https://azvod1.s3-ap-northeast-1.amazonaws.com"
  337. # session = Session(ACCESS_KEY, SECRET_KEY)
  338. # s3_client = session.client('s3', endpoint_url=url)
  339. # results = s3_client.list_buckets()
  340. # return JsonResponse(status=200, data={'code': 200, 'msg': results['Buckets']})
  341. # Create bucket
  342. session = Session(ACCESS_KEY, SECRET_KEY)
  343. s3_client = session.client('s3')
  344. # 上传
  345. # 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())
  346. # 下载
  347. resp = s3_client.get_object(Bucket="azvod1", Key="file/rule.txt")
  348. with open('local.txt', 'wb') as f:
  349. f.write(resp['Body'].read())
  350. # 新查询设备字段
  351. def queryInterface(self, request_dict, userID, response):
  352. page = request_dict.get('page', None)
  353. line = request_dict.get('line', None)
  354. NickName = request_dict.get('NickName', None)
  355. uid = request_dict.get('uid', None)
  356. page = int(page)
  357. line = int(line)
  358. response.lang = 'cn'
  359. userID = userID
  360. dvqs = Device_Info.objects.filter(userID_id=userID)
  361. # # 已重置的设备
  362. # dvqs = dvqs.filter(~Q(isExist=2))
  363. if NickName:
  364. dvqs = dvqs.filter(NickName__icontains=NickName)
  365. if uid:
  366. dvqs = dvqs.filter(UID=uid)
  367. # count = dvqs.count()
  368. # 分页
  369. dvql = dvqs[(page - 1) * line:page * line].values('id', 'userID', 'NickName', 'UID', 'View_Account',
  370. 'View_Password', 'ChannelIndex', 'Type', 'isShare',
  371. 'primaryUserID', 'primaryMaster', 'data_joined',
  372. 'version',
  373. 'isVod', 'isExist', 'NotificationMode')
  374. dvls = CommonService.qs_to_list(dvql)
  375. uid_list = []
  376. for dvl in dvls:
  377. uid_list.append(dvl['UID'])
  378. # if dvl['isShare'] is False:
  379. # uid_list.append(dvl['UID'])
  380. ubqs = UID_Bucket.objects.filter(uid__in=uid_list). \
  381. values('bucket__content', 'status', 'channel', 'endTime', 'uid')
  382. upqs = UID_Preview.objects.filter(uid__in=uid_list).order_by('channel').values('id', 'uid', 'channel')
  383. auth = oss2.Auth(OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET)
  384. bucket = oss2.Bucket(auth, 'oss-cn-hongkong.aliyuncs.com', 'statres')
  385. nowTime = int(time.time())
  386. data = []
  387. # 设备拓展信息表
  388. us_qs = UidSetModel.objects.filter(uid__in=uid_list). \
  389. values('uid', 'version', 'nickname', 'detect_interval')
  390. uv_dict = {}
  391. for us in us_qs:
  392. uv_dict[us['uid']] = {'version': us['version'],
  393. 'nickname': us['nickname'],
  394. 'detect_interval': us['detect_interval']}
  395. for p in dvls:
  396. p['vod'] = []
  397. for dm in ubqs:
  398. if p['UID'] == dm['uid']:
  399. if dm['endTime'] > nowTime:
  400. p['vod'].append(dm)
  401. p['preview'] = []
  402. for up in upqs:
  403. if p['UID'] == up['uid']:
  404. obj = 'uid_preview/{uid}/channel_{channel}.png'. \
  405. format(uid=up['uid'], channel=up['channel'])
  406. img_sign = bucket.sign_url('GET', obj, 300)
  407. p['preview'].append(img_sign)
  408. p_uid = p['UID']
  409. if p_uid in uv_dict:
  410. # 设备版本号
  411. p['uid_version'] = uv_dict[p_uid]['version']
  412. p['detect_interval'] = uv_dict[p_uid]['detect_interval']
  413. # 设备昵称 调用影子信息昵称,先阶段不可
  414. if uv_dict[p_uid]['nickname']:
  415. p['NickName'] = uv_dict[p_uid]['nickname']
  416. else:
  417. # 设备版本号
  418. p['uid_version'] = ''
  419. data.append(p)
  420. return response.json(0, data)
  421. # 获取播放列表
  422. def do_test_query_vod_list(self, userID, ip, request_dict, response):
  423. uid = 'GZL2PEFJPLY7W6BG111A'
  424. channel = 2
  425. userID = '158943594633713800138000'
  426. dv_qs = Device_Info.objects.filter(UID=uid, userID_id=userID, isShare=False)
  427. if not dv_qs.exists():
  428. return response.json(12)
  429. vod_play_list = []
  430. bucket_name = 'azvod1'
  431. aws_access_key_id = 'AKIA2E67UIMD45Y3HL53'
  432. aws_secret_access_key = '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'
  433. session = Session(
  434. aws_access_key_id=aws_access_key_id,
  435. aws_secret_access_key=aws_secret_access_key,
  436. region_name='ap-northeast-1'
  437. )
  438. conn = session.client('s3')
  439. thumbspng = '{uid}/vod{channel}/{time}/thumbs.png'. \
  440. format(uid=uid, channel=channel, time=1590485548)
  441. response_url = conn.generate_presigned_url(
  442. 'get_object',
  443. Params={
  444. 'Bucket': bucket_name,
  445. 'Key': thumbspng
  446. },
  447. ExpiresIn=3600
  448. )
  449. # m3u8 = '{uid}/vod{channel}/{time}/{time}.m3u8'. \
  450. # format(uid=uid, channel=channel, time=vod['time'])
  451. thumb_url = response_url
  452. vod_url = 'http://192.168.136.191:8000/testApi/signplaym3u8?' \
  453. 'uid={uid}&channel={channel}&time={time}&sign=tktktktk'. \
  454. format(ip=ip, uid=uid, channel=channel, time=1590485548)
  455. vod_play_list.append({
  456. 'name': 1590485548,
  457. 'sign_url': vod_url,
  458. 'thumb': thumb_url,
  459. 'sec': 12})
  460. return response.json(0, vod_play_list)
  461. # 生成m3u8列表
  462. def do_sign_play_m3u8(self, request_dict, response):
  463. uid = 'GZL2PEFJPLY7W6BG111A'
  464. channel = 2
  465. storeTime = 1591344070
  466. fg = 6
  467. bucket__region = 'ap-northeast-1'
  468. bucket_name = 'azvod1'
  469. aws_access_key_id = 'AKIA2E67UIMD45Y3HL53'
  470. aws_secret_access_key = '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'
  471. session = Session(
  472. aws_access_key_id=aws_access_key_id,
  473. aws_secret_access_key=aws_secret_access_key,
  474. region_name=bucket__region
  475. )
  476. '''
  477. http://test.dvema.com/cloudstorage/signplaym3u8?uid=VVDHCVBYDKFMJRWA111A&channel=1&time=1586940120&sign=tktktktk
  478. '''
  479. conn = session.client('s3')
  480. playlist_entries = []
  481. for i in range(fg):
  482. thumbspng = '{uid}/vod{channel}/{time}/ts{i}.ts'. \
  483. format(uid=uid, channel=channel, time=storeTime, i=i)
  484. response_url = conn.generate_presigned_url(
  485. 'get_object',
  486. Params={
  487. 'Bucket': bucket_name,
  488. 'Key': thumbspng
  489. },
  490. ExpiresIn=86400
  491. )
  492. # m3u8 = '{uid}/vod{channel}/{time}/{time}.m3u8'. \
  493. # format(uid=uid, channel=channel, time=vod['time'])
  494. playlist_entries.append({
  495. 'name': response_url,
  496. 'duration': 10,
  497. })
  498. playlist = PlaylistGenerator(playlist_entries).generate()
  499. response = HttpResponse(playlist)
  500. response['Content-Type'] = 'application/octet-stream'
  501. response['Content-Disposition'] = 'attachment;filename="play.m3u8"'
  502. return response
  503. # return HttpResponse(playlist)
  504. # response = HttpResponse(playlist, content_type="application/vnd.apple.mpegurl")
  505. # # response = HttpResponse(playlist, content_type="application/octet-stream")
  506. # return response
  507. # return HttpResponse(status=200, content=playlist)
  508. def do_pay_by_ali(self, request_dict, userID, response):
  509. uid = request_dict.get('uid', None)
  510. rank = request_dict.get('rank', None)
  511. channel = request_dict.get('channel', None)
  512. qs = Device_Info.objects.filter(userID_id=userID, UID=uid, isShare=False)
  513. if not qs.exists():
  514. return response.json(12)
  515. if not channel or not rank:
  516. return response.json(444, 'channel,rank')
  517. smqs = Store_Meal.objects.filter(id=rank). \
  518. values("currency", "price", "content", "day", "bucket__storeDay", "bucket__region", "type")
  519. if not smqs.exists():
  520. # 套餐不存在
  521. return response.json(173)
  522. if smqs[0]['type'] != 1:
  523. return response.json(10, '不支持支付宝支付')
  524. currency = smqs[0]['currency']
  525. price = smqs[0]['price']
  526. content = smqs[0]['content']
  527. day = smqs[0]['day']
  528. nowTime = int(time.time())
  529. ubqs = UID_Bucket.objects.filter(uid=uid, channel=channel, endTime__gte=nowTime). \
  530. values("bucket__storeDay", "bucket__region")
  531. if ubqs.exists():
  532. if ubqs[0]['bucket__region'] != smqs[0]['bucket__region']:
  533. return response.json(712) # 区域不一致
  534. elif ubqs[0]['bucket__storeDay'] != smqs[0]['bucket__storeDay']:
  535. return response.json(713) # 不可更改套餐
  536. # 续费流程
  537. nowTime = int(time.time())
  538. # 新增流程
  539. orderID = CommonService.createOrderID()
  540. try:
  541. aliPayObj = AliPayObject()
  542. alipay = aliPayObj.conf()
  543. order_string = alipay.api_alipay_trade_wap_pay(
  544. out_trade_no=orderID,
  545. total_amount=price,
  546. subject="测试哟",
  547. return_url="{SERVER_DOMAIN_SSL}cloudVod/payOK".format(SERVER_DOMAIN_SSL=SERVER_DOMAIN_SSL),
  548. notify_url="{SERVER_DOMAIN_SSL}cloudVod/aliPayCallback".format(SERVER_DOMAIN_SSL=SERVER_DOMAIN_SSL)
  549. # return_url="http://192.168.136.40/cloudVod/payOK",
  550. # notify_url="http://192.168.136.40/cloudVod/aliPayCallback"
  551. )
  552. except Exception as e:
  553. print(repr(e))
  554. return response.json(10, repr(e))
  555. if order_string:
  556. redirectUrl = aliPayObj.alipay_prefix + order_string
  557. store_meal_qs = Store_Meal.objects.filter(id=rank, lang__lang='cn', is_show=0).values('lang__title',
  558. 'lang__content')
  559. if store_meal_qs.exists():
  560. store_meal_name = store_meal_qs[0]['lang__title'] + '-' + store_meal_qs[0]['lang__content']
  561. else:
  562. store_meal_name = '未知套餐'
  563. Order_Model.objects.create(orderID=orderID, UID=uid, channel=channel, userID_id=userID, desc=content,
  564. price=price, currency=currency, addTime=nowTime, updTime=nowTime,
  565. endTime=nowTime + int(day) * 3600 * 24, rank_id=rank, payType=1,
  566. store_meal_name=store_meal_name)
  567. return JsonResponse(status=200,
  568. data={'result_code': 0, 'reason': 'success',
  569. 'result': {"redirectUrl": redirectUrl, "orderID": orderID},
  570. 'error_code': 0})
  571. else:
  572. return response.json(10, '生成订单错误')
  573. def do_filter_playlist(self, request_dict, userID, response):
  574. startTime = int(request_dict.get('startTime', None))
  575. endTime = int(request_dict.get('endTime', None))
  576. uid = request_dict.get('uid', None)
  577. channel = request_dict.get('channel', None)
  578. dvqs = Device_Info.objects.filter(UID=uid, userID_id=userID, isShare=False)
  579. if not dvqs.exists():
  580. return response.json(12)
  581. ubqs = UID_Bucket.objects.filter(uid=uid, channel=channel).values('status')
  582. if not ubqs.exists():
  583. return response.json(10, '设备未购买')
  584. split_vod_hls_obj = SplitVodHlsObject()
  585. vodqs = split_vod_hls_obj.get_vod_hls_data(uid=uid, channel=channel).values("start_time", "sec", "bucket_id")
  586. if not vodqs.exists():
  587. return response.json(173)
  588. vod_play_list = []
  589. auth = oss2.Auth(OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET)
  590. bucket_id = vodqs[0]['bucket_id']
  591. vod_bucket_qs = VodBucketModel.objects.filter(id=bucket_id).values('bucket', 'endpoint')
  592. if not vod_bucket_qs.exists():
  593. return response.json(173)
  594. bucket_name = vod_bucket_qs[0]["bucket"]
  595. endpoint = vod_bucket_qs[0]["endpoint"]
  596. for vod in vodqs:
  597. bucket = oss2.Bucket(auth, endpoint, bucket_name)
  598. m3u8 = '{uid}/vod{channel}/{time}/{time}.m3u8'. \
  599. format(uid=uid, channel=channel, time=vod['start_time'])
  600. ts = '{uid}/vod{channel}/{time}/ts0.ts'. \
  601. format(uid=uid, channel=channel, time=vod['start_time'])
  602. url = bucket.sign_url('GET', m3u8, 3600, params={'x-oss-process': 'hls/sign'})
  603. urllst = url.split('?')
  604. url_start = urllib.parse.unquote(urllst[0])
  605. url_end = urllst[1]
  606. vod_play_url = '{url_start}?{url_end}'. \
  607. format(url_start=url_start, url_end=url_end)
  608. thumb = bucket.sign_url('GET', ts, 3600,
  609. params={'x-oss-process': 'video/snapshot,t_10000,m_fast,w_300'})
  610. vod_play_list.append(
  611. {'name': vod['start_time'], 'sign_url': vod_play_url, 'thumb': thumb, 'sec': vod['sec']})
  612. return response.json(0, vod_play_list)
  613. def generate_token(self, request_dict, userID):
  614. # UserIdToken
  615. tko = TokenObject()
  616. userID = request_dict.get('userID', None)
  617. username = request_dict.get('username', None)
  618. res = tko.generate(
  619. data={'userID': userID, 'lang': 'cn', 'user': username, 'm_code': username})
  620. # uidToken
  621. # utko = UidTokenObject()
  622. # res = utko.generate(data={'uid': '4UZSEDP93MJ3X7YB111A','channel': 1})
  623. # from Object.ETkObject import ETkObject
  624. # etkObj = ETkObject(etk='')
  625. # res = etkObj.encrypt("4UZSEDP93MJ3X7YB111A")
  626. return JsonResponse(status=200, data=res, safe=False)
  627. def test_upload_s3(self, request_dict, response):
  628. aws_s3_guonei = boto3.client(
  629. 's3',
  630. aws_access_key_id=AWS_ACCESS_KEY_ID[0],
  631. aws_secret_access_key=AWS_SECRET_ACCESS_KEY[0],
  632. config=botocore.client.Config(signature_version='s3v4'),
  633. region_name='cn-northwest-1'
  634. )
  635. download_link = 'ipctest'
  636. response_url = aws_s3_guonei.generate_presigned_url(
  637. ClientMethod='put_object',
  638. Params={
  639. 'Bucket': 'pc-package',
  640. 'Key': download_link
  641. },
  642. ExpiresIn=3600
  643. )
  644. return response.json(0, {'datas': response_url, 'count': 1})
  645. def testRekognition(self, request, request_dict):
  646. # ip = CommonService.get_ip_address(request)
  647. # ipInfo = CommonService.getIpIpInfo(ip,"CN")
  648. # # print(type(ipInfo))
  649. # # exit(ipInfo)
  650. # addr = CommonService.getAddr(ip)
  651. # dicts = {
  652. # 'ipInfo':ipInfo,
  653. # 'addr':addr,
  654. # }
  655. # return HttpResponse(json.dumps(dicts, ensure_ascii=False),
  656. # content_type="application/json,charset=utf-8")
  657. # client = boto3.client('s3', aws_access_key_id='AKIA2E67UIMD45Y3HL53',aws_secret_access_key='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',region_name='us-east-1')
  658. # exit(request.FILES)
  659. files = request.FILES.get('image')
  660. labels = int(request_dict.get('labels', 5))
  661. minConfidence = int(request_dict.get('minConfidence', 99))
  662. if not files:
  663. return HttpResponse('请上传图片!!!!')
  664. client = boto3.client('rekognition', aws_access_key_id='AKIA2E67UIMD6JD6TN3J',
  665. aws_secret_access_key='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', region_name='us-east-1')
  666. # image = open('E:/photo/a615fa40b8c476bab0f6eeb332e62a5a-1000.jpg', "rb")
  667. response = client.detect_labels(Image={'Bytes': files.read()}, MaxLabels=labels, MinConfidence=minConfidence)
  668. # for obj in response['Labels']:
  669. # exit(obj)
  670. # if obj['Name'] == 'Person':
  671. # jsonstr = json.dumps(obj)
  672. return HttpResponse(json.dumps(response, ensure_ascii=False),
  673. content_type="application/json,charset=utf-8")
  674. def ip(self, response):
  675. ip = '67.220.90.13'
  676. country_id = Device_Region().get_device_region(ip)
  677. print(country_id)
  678. return response.json(0)
  679. def configType(self, response):
  680. print(SERVER_TYPE)
  681. return response.json(0)
  682. def createData(self, request_dict, response):
  683. uid = request_dict.get('uid', None)
  684. # filter_date = datetime.datetime.now()-datetime.timedelta(days=4)
  685. # qs = DeviceLogModel.objects.filter(add_time__lt=filter_date)
  686. DeviceLogModel.objects.create(uid=uid)
  687. return response.json(0)
  688. def count_ts(self, request_dict, response):
  689. year = request_dict.get('year', None)
  690. month = request_dict.get('month', None)
  691. if not year:
  692. year = CommonService.timestamp_to_str(int(time.time()), '%Y')
  693. if not month:
  694. month = CommonService.timestamp_to_str(int(time.time()), '%m')
  695. year = int(year)
  696. month = int(month)
  697. startTime = CommonService.str_to_timestamp('{year}-{month}'.format(year=year, month=month), '%Y-%m')
  698. endTime = CommonService.str_to_timestamp('{year}-{month}'.format(year=year, month=month + 1), '%Y-%m') - 1
  699. split_vod_hls_obj = SplitVodHlsObject()
  700. qsTs = split_vod_hls_obj.get_vod_hls_data(time__gte=startTime, time__lte=endTime).values('fg')
  701. if not qsTs.exists():
  702. return HttpResponse('查无数据')
  703. sumTs = 0 # 总ts个数
  704. sumSec = 0 # 总秒数
  705. for val in qsTs:
  706. fg = int(val['fg'])
  707. sumTs += fg & 0xf
  708. for i in range(15):
  709. shift = (i + 1) * 4
  710. sumSec += (fg >> shift) & 0xf
  711. size = 0
  712. return HttpResponse(
  713. "{year}年{month}月 </br>上传的TS总数:{sumTs} </br> 总秒数:{sumSec} </br> 总大小:{size}GB (1秒约等150KB计算)".format(year=year,
  714. month=month,
  715. sumTs=sumTs,
  716. sumSec=sumSec,
  717. size=size))
  718. @staticmethod
  719. def write_redis_list(response):
  720. redis_obj = RedisObject()
  721. for i in range(10):
  722. redis_obj.rpush('serial_redis_list', i)
  723. return response.json(0)
  724. @staticmethod
  725. def read_redis_list(response):
  726. redis_obj = RedisObject()
  727. serial_redis_list_len = redis_obj.llen('serial_redis_list')
  728. if serial_redis_list_len > 0:
  729. for i in range(serial_redis_list_len):
  730. serial = redis_obj.lpop('serial_redis_list')
  731. print(serial)
  732. return response.json(0)
  733. @staticmethod
  734. def play_m3u8(request_dict, response): # 根据sts播放m3u8 视频流
  735. uid = request_dict.get('uid', None)
  736. channel = request_dict.get('channel', None)
  737. storeTime = request_dict.get('time', None)
  738. now_time = int(time.time())
  739. try:
  740. split_vod_hls_obj = SplitVodHlsObject()
  741. vh_qs = split_vod_hls_obj.get_vod_hls_data(uid=uid, channel=channel, start_time=storeTime,
  742. end_time__gte=now_time).values("sec", "fg", "bucket_id")
  743. if not vh_qs.exists():
  744. return response.json(173)
  745. vod_bucket_qs = VodBucketModel.objects.filter(id=vh_qs[0]['bucket_id']).values('bucket', 'region', 'mold')
  746. if not vod_bucket_qs.exists():
  747. return response.json(173)
  748. fg = int(vh_qs[0]['fg'])
  749. bucket_region = vod_bucket_qs[0]['region']
  750. bucket_name = vod_bucket_qs[0]['bucket']
  751. mold = vod_bucket_qs[0]['mold']
  752. session = Session(
  753. aws_access_key_id=AWS_ACCESS_KEY_ID[mold],
  754. aws_secret_access_key=AWS_SECRET_ACCESS_KEY[mold],
  755. region_name=bucket_region
  756. )
  757. conn = session.client('s3')
  758. playlist_entries = []
  759. # ts_count = fg & 0xf
  760. # fg 64位整型,低四位代表ts文件总数,然后进行位运算,一次移四位,每四位转为十进制即为当前ts文件的秒数
  761. for i in range(15):
  762. shift = (i + 1) * 4
  763. duration = (fg >> shift) & 0xf
  764. if duration > 0:
  765. tsFile = '{uid}/vod{channel}/{time}/ts{i}.ts'. \
  766. format(uid=uid, channel=channel, time=storeTime, i=i)
  767. response_url = conn.generate_presigned_url(
  768. 'get_object',
  769. Params={
  770. 'Bucket': bucket_name,
  771. 'Key': tsFile
  772. },
  773. ExpiresIn=24 * 60 * 60
  774. )
  775. playlist_entries.append({
  776. 'name': response_url,
  777. 'duration': duration,
  778. })
  779. playlist = PlaylistGenerator(playlist_entries).generate()
  780. response = HttpResponse(playlist)
  781. response['Content-Type'] = 'application/octet-stream'
  782. response['Content-Disposition'] = 'attachment;filename="play.m3u8"'
  783. return response
  784. except Exception as e:
  785. print(e)
  786. return response.json(500, repr(e))
  787. @staticmethod
  788. def getSerialNumberInfo(request_dict, response):
  789. logger = logging.getLogger('info')
  790. serial_number = request_dict.get('serialNumber', None)
  791. if not serial_number:
  792. return response.json(444)
  793. serialNumber = serial_number[:9]
  794. serial_number = serial_number[:6]
  795. try:
  796. uid_user_message = {
  797. "uid": "",
  798. "serialNumber": "",
  799. "userID": "",
  800. "username": "",
  801. "primaryUserID": ""
  802. }
  803. data = {
  804. "uid": "",
  805. "serialNumber": serialNumber,
  806. "status": "",
  807. "uid_user_message": uid_user_message
  808. }
  809. company_serial_qs = CompanySerialModel.objects.filter(serial_number=serial_number).values('status')
  810. if not company_serial_qs.exists():
  811. return response.json(173)
  812. if company_serial_qs[0]['status'] == 0:
  813. return response.json(0, {'contents': '序列号未分配'})
  814. uid_company_serial_qs = UIDCompanySerialModel.objects.filter(
  815. company_serial__serial_number=serial_number).values('uid__uid', 'uid__status',
  816. 'company_serial__serial_number')
  817. if not uid_company_serial_qs.exists() and company_serial_qs[0]['status'] != 0:
  818. if company_serial_qs[0]['status'] == 1:
  819. data['status'] = '已分配'
  820. if company_serial_qs[0]['status'] == 2:
  821. data['status'] = '已绑定uid'
  822. if company_serial_qs[0]['status'] == 3:
  823. data['status'] = '已占用'
  824. return response.json(0, data)
  825. for uid_company_serial in uid_company_serial_qs:
  826. data['uid'] = uid_company_serial['uid__uid']
  827. data['serialNumber'] = serialNumber
  828. data['status'] = uid_company_serial['uid__status']
  829. if company_serial_qs[0]['status'] == 1:
  830. data['status'] = '已分配'
  831. if company_serial_qs[0]['status'] == 2:
  832. data['status'] = '绑定uid'
  833. if company_serial_qs[0]['status'] == 3:
  834. data['status'] = '已占用'
  835. uid = uid_company_serial['uid__uid'] if uid_company_serial['uid__uid'] else ''
  836. device_info_qs = Device_Info.objects.filter(UID=uid).values(
  837. 'UID',
  838. 'serial_number',
  839. 'userID_id',
  840. 'primaryUserID',
  841. 'userID__username')
  842. uid_user_message = {
  843. 'uid': device_info_qs[0]['UID'] if device_info_qs.exists() else '',
  844. 'serialNumber': device_info_qs[0]['serial_number'] if device_info_qs.exists() else '',
  845. 'userID': device_info_qs[0]['userID_id'] if device_info_qs.exists() else '',
  846. 'username': device_info_qs[0]['userID__username'] if device_info_qs.exists() else '',
  847. 'primaryUserID': device_info_qs[0]['primaryUserID'] if device_info_qs.exists() else ''
  848. }
  849. data['uid_user_message'] = uid_user_message
  850. return response.json(0, data)
  851. except Exception as e:
  852. logger.info('查询异常:{}'.format(e))
  853. return response.json(500)