TestApi.py 44 KB

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