TestApi.py 43 KB

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