TestApi.py 43 KB

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