IcloudService.py 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558
  1. # -*- coding: utf-8 -*-
  2. """
  3. @Author : peng
  4. @Time : 2023-6-7 18:26:35
  5. @File :IcloudMeal.py
  6. """
  7. from Ansjer.config import LOGGER
  8. import time
  9. import jwt
  10. from django.db.models import Sum, Q
  11. from django.http import HttpResponse
  12. from django.views import View
  13. from Model.models import IcloudUseDetails, IcloudService, VodBucketModel, IcloudStorageRecord, Device_Info
  14. from Object.AWS.AmazonS3Util import AmazonS3Util
  15. from Object.RedisObject import RedisObject
  16. from Object.ResponseObject import ResponseObject
  17. from Object.TokenObject import TokenObject
  18. from Ansjer.config import ACCESS_KEY_ID, SECRET_ACCESS_KEY, REGION_NAME, SERVER_DOMAIN, AWS_ACCESS_KEY_ID, \
  19. AWS_SECRET_ACCESS_KEY, SERVER_DOMAIN_SSL, OAUTH_ACCESS_TOKEN_SECRET
  20. from Object.m3u8generate import PlaylistGenerator
  21. from Service.VodHlsService import SplitVodHlsObject
  22. class IcloudServiceView(View):
  23. def get(self, request, *args, **kwargs):
  24. request.encoding = 'utf-8'
  25. operation = kwargs.get('operation')
  26. return self.validation(request.GET, operation, request)
  27. def post(self, request, *args, **kwargs):
  28. request.encoding = 'utf-8'
  29. operation = kwargs.get('operation')
  30. return self.validation(request.POST, operation, request)
  31. def validation(self, request_dict, operation, request):
  32. response = ResponseObject()
  33. if operation == 'icloudPlay': # 云盘播放
  34. return self.icloud_play(request_dict, response)
  35. else:
  36. tko = TokenObject(request.META.get('HTTP_AUTHORIZATION'))
  37. response.lang = tko.lang
  38. user_id = tko.userID
  39. if tko.code != 0:
  40. return response.json(tko.code)
  41. if operation == 'vodUpload': # 云存上传云盘
  42. return self.vod_upload(request_dict, response, user_id)
  43. elif operation == 'getUploadUrl': # 获取s3上传地址
  44. return self.get_upload_url(request_dict, response, user_id)
  45. elif operation == 'localUpload': # 本地上传云盘
  46. return self.local_upload(request_dict, response, user_id)
  47. elif operation == 'deleteIcloudRecord': # 删除云盘记录
  48. return self.delete_icloud_record(request_dict, response, user_id)
  49. elif operation == 'getIcloudList': # 获取云盘播放列表
  50. return self.get_icloud_list(request_dict, response, user_id)
  51. else:
  52. return response.json(404)
  53. @staticmethod
  54. def vod_upload(request_dict, response, user_id):
  55. """
  56. 云存上传云盘
  57. @param request_dict: 请求参数
  58. @param user_id: 用户id
  59. @request_dict uid: 设备uid
  60. @request_dict channel: 设备通道
  61. @request_dict start_time: 播放时间戳
  62. @param response: 响应对象
  63. @return: response
  64. """
  65. LOGGER.info('开始云存转移到云盘')
  66. uid = request_dict.get('uid', None)
  67. channel = request_dict.get('channel', None)
  68. time_stamp = request_dict.get('time_stamp', None)
  69. if not all([uid, channel, time_stamp]):
  70. return response.json(444, {'error param': 'uid,channel,start_time'})
  71. device_qs = Device_Info.objects.filter(UID=uid, userID=user_id).values('NickName')
  72. if not device_qs.exists():
  73. return response.json(173)
  74. # 查询是否开通云盘
  75. use_details_qs = IcloudUseDetails.objects.filter(user_id=user_id).values('id', 'bucket_id', 'use_size')
  76. if not use_details_qs.exists():
  77. return response.json(173)
  78. use_details_id = use_details_qs[0]['id']
  79. target_bucket_id = use_details_qs[0]['bucket_id']
  80. use_size = float(use_details_qs[0]['use_size'])
  81. now_time = int(time.time())
  82. nickname = device_qs[0]['NickName']
  83. # 判断云盘是否还有容量
  84. all_size = IcloudService.objects.filter(Q(use_details_id=use_details_id), Q(use_status=0),
  85. Q(end_time__gt=now_time) | Q(end_time=0)).values(
  86. 'size').aggregate(total_size=Sum('size'))['total_size']
  87. all_size = all_size * 1024 if all_size else 0
  88. if use_size > all_size:
  89. LOGGER.info('{}用户套餐总容量为:{},已使用容量为:{}'.format(uid, all_size, use_size))
  90. return response.json(910)
  91. split_vod_hls_obj = SplitVodHlsObject()
  92. vod_hls = split_vod_hls_obj.get_vod_hls_data(uid=uid, channel=channel, start_time=time_stamp).values(
  93. 'bucket_id', 'fg', 'sec')
  94. if not vod_hls.exists():
  95. LOGGER.info('{}用户查无此云存:{}'.format(uid, time_stamp))
  96. return response.json(173)
  97. source_bucket_id = vod_hls[0]['bucket_id']
  98. fg = int(vod_hls[0]['fg'])
  99. sec = vod_hls[0]['sec']
  100. target_bucket_qs = VodBucketModel.objects.filter(id=target_bucket_id).values('bucket')
  101. if not target_bucket_qs.exists():
  102. return response.json(173)
  103. target_bucket_name = target_bucket_qs[0]['bucket']
  104. source_bucket_qs = VodBucketModel.objects.filter(id=source_bucket_id).values('bucket', 'region', 'mold')
  105. if not source_bucket_qs.exists():
  106. return response.json(173)
  107. bucket_region = source_bucket_qs[0]['region']
  108. source_bucket_name = source_bucket_qs[0]['bucket']
  109. mold = source_bucket_qs[0]["mold"]
  110. ts_list = []
  111. ts_size = 0
  112. try:
  113. s3_obj = AmazonS3Util(
  114. AWS_ACCESS_KEY_ID[mold],
  115. AWS_SECRET_ACCESS_KEY[mold],
  116. bucket_region
  117. )
  118. # 获取视频封面
  119. thumbs_png = '{uid}/vod{channel}/{time}/Thumb.jpeg'.format(uid=uid, channel=channel, time=time_stamp)
  120. thumbs_png_size = s3_obj.get_object_size(source_bucket_name, thumbs_png)
  121. if thumbs_png_size:
  122. ts_size += thumbs_png_size
  123. ts_list.append(thumbs_png)
  124. # 获取s3对象,并计算总字节
  125. for i in range(15):
  126. shift = (i + 1) * 4
  127. duration = (fg >> shift) & 0xf
  128. if duration > 0:
  129. ts_file = '{uid}/vod{channel}/{time}/ts{i}.ts'.format(uid=uid, channel=channel, time=time_stamp,
  130. i=i)
  131. ts_list.append(ts_file)
  132. ts_size += s3_obj.get_object_size(source_bucket_name, ts_file) # 获取存储对象字节,单位B
  133. ts_size = round(ts_size / 1024 / 1024, 2) # 字节转换为MB单位
  134. temp_size = ts_size + use_size
  135. if temp_size > all_size:
  136. LOGGER.info('{}用户无空间上传,套餐容量为:{},使用容量为:{}'.format(uid, all_size, temp_size))
  137. return response.json(910)
  138. time_stamp = int(time_stamp) * 1000 # 转换单位为毫秒
  139. icloud_record_qs = IcloudStorageRecord.objects.filter(user_id=user_id, uid=uid, channel=channel,
  140. time_stamp=time_stamp,
  141. size=ts_size, file_type=1)
  142. if not icloud_record_qs.exists(): # 转移云盘,并记录上传记录,更新使用容量
  143. for source_key in ts_list:
  144. ts_name = source_key.split('/')[-1]
  145. target_key = '{user_id}/ts_file/{uid}/channel{channel}/{time}/{ts_name}'.format(user_id=user_id,
  146. uid=uid,
  147. channel=channel,
  148. time=time_stamp,
  149. ts_name=ts_name)
  150. s3_obj.copy_single_obj(source_bucket_name, source_key, target_bucket_name, target_key)
  151. IcloudStorageRecord.objects.create(user_id=user_id, uid=uid, channel=channel, time_stamp=time_stamp,
  152. nickname=nickname,
  153. sec=sec, bucket_id=target_bucket_id, fg=fg, size=ts_size,
  154. file_type=1)
  155. use_details_qs.update(use_size=temp_size)
  156. return response.json(0)
  157. else:
  158. return response.json(174)
  159. except Exception as e:
  160. LOGGER.info('云存转移云盘异常:{}'.format(repr(e)))
  161. return response.json(500)
  162. @staticmethod
  163. def local_upload(request_dict, response, user_id):
  164. """
  165. 本地上传云盘
  166. @param request_dict: 请求参数
  167. @param user_id: 用户id
  168. @request_dict uid: 设备uid
  169. @request_dict channel: 设备通道
  170. @request_dict time_stamp: 时间戳
  171. @request_dict size: 文件大小
  172. @request_dict file_type: 文件类型
  173. @request_dict sec: 时长
  174. @param response: 响应对象
  175. @return: response
  176. """
  177. LOGGER.info('开始本地转移到云盘')
  178. uid = request_dict.get('uid', None)
  179. channel = request_dict.get('channel', None)
  180. time_stamp = request_dict.get('time_stamp', None)
  181. size = request_dict.get('size', None)
  182. file_type = request_dict.get('file_type', None)
  183. sec = request_dict.get('sec', 0)
  184. if not all([uid, channel, time_stamp, size]):
  185. return response.json(444, {'error param': 'uid,channel,time_stamp,size'})
  186. device_qs = Device_Info.objects.filter(UID=uid, userID=user_id).values('NickName')
  187. if not device_qs.exists():
  188. return response.json(173)
  189. # 查询是否开通云盘
  190. use_details_qs = IcloudUseDetails.objects.filter(user_id=user_id).values('id', 'use_size', 'bucket_id')
  191. if not use_details_qs.exists():
  192. return response.json(173)
  193. size = float(size) # 单位是MB
  194. use_details_id = use_details_qs[0]['id']
  195. bucket_id = use_details_qs[0]['bucket_id']
  196. use_size = float(use_details_qs[0]['use_size'])
  197. now_time = int(time.time())
  198. nickname = device_qs[0]['NickName']
  199. # 判断云盘是否还有容量
  200. all_size = IcloudService.objects.filter(Q(use_details_id=use_details_id), Q(use_status=0),
  201. Q(end_time__gt=now_time) | Q(end_time=0)).values(
  202. 'size').aggregate(total_size=Sum('size'))['total_size']
  203. all_size = all_size * 1024 if all_size else 0
  204. if use_size > all_size:
  205. LOGGER.info('{}用户套餐总容量为:{},已使用容量为:{}'.format(uid, all_size, use_size))
  206. return response.json(910)
  207. try:
  208. new_size = round(use_size + size, 2)
  209. icloud_record_qs = IcloudStorageRecord.objects.filter(user_id=user_id, uid=uid, channel=channel,
  210. time_stamp=time_stamp,
  211. size=size, file_type=file_type)
  212. if not icloud_record_qs.exists():
  213. IcloudStorageRecord.objects.create(user_id=user_id, uid=uid, channel=channel, time_stamp=time_stamp,
  214. nickname=nickname,
  215. sec=sec, bucket_id=bucket_id, size=size,
  216. file_type=file_type)
  217. use_details_qs.update(use_size=new_size)
  218. return response.json(0)
  219. else:
  220. return response.json(174)
  221. except Exception as e:
  222. LOGGER.info('本地转移云盘异常:{}'.format(repr(e)))
  223. return response.json(500)
  224. @staticmethod
  225. def get_upload_url(request_dict, response, user_id):
  226. """
  227. 获取s3上传地址
  228. @param request_dict: 请求参数
  229. @param user_id: 用户id
  230. @request_dict time_stamp: 时间戳
  231. @request_dict file_type: 文件类型
  232. @request_dict uid: 设备uid
  233. @request_dict channel: 设备通道
  234. @param response: 响应对象
  235. @return: response
  236. """
  237. LOGGER.info('获取s3上传地址')
  238. time_stamp = request_dict.get('time_stamp', None)
  239. file_type = request_dict.get('file_type', None)
  240. uid = request_dict.get('uid', None)
  241. channel = request_dict.get('channel', None)
  242. if not all([time_stamp, uid, channel]):
  243. return response.json(444, {'error param': 'time_stamp,channel,uid'})
  244. device_qs = Device_Info.objects.filter(UID=uid, userID=user_id)
  245. if not device_qs.exists():
  246. return response.json(173)
  247. # 查询是否开通云盘
  248. use_details_qs = IcloudUseDetails.objects.filter(user_id=user_id).values('id', 'use_size', 'bucket_id')
  249. if not use_details_qs.exists():
  250. return response.json(173)
  251. use_details_id = use_details_qs[0]['id']
  252. use_size = use_details_qs[0]['use_size']
  253. bucket_id = use_details_qs[0]['bucket_id']
  254. now_time = int(time.time())
  255. # 判断云盘是否还有容量
  256. all_size = IcloudService.objects.filter(Q(use_details_id=use_details_id), Q(use_status=0),
  257. Q(end_time__gt=now_time) | Q(end_time=0)).values(
  258. 'size').aggregate(total_size=Sum('size'))['total_size']
  259. all_size = all_size * 1024 if all_size else 0
  260. excess_size = all_size - use_size
  261. if excess_size < 0:
  262. LOGGER.info('{}用户套餐总容量为:{},已使用容量为:{}'.format(user_id, all_size, use_size))
  263. return response.json(910)
  264. bucket_qs = VodBucketModel.objects.filter(id=bucket_id).values('bucket', 'region', 'mold')
  265. if not bucket_qs.exists():
  266. return response.json(173)
  267. bucket_name = bucket_qs[0]['bucket']
  268. bucket_region = bucket_qs[0]['region']
  269. mold = bucket_qs[0]['mold']
  270. try:
  271. s3_obj = AmazonS3Util(
  272. AWS_ACCESS_KEY_ID[mold],
  273. AWS_SECRET_ACCESS_KEY[mold],
  274. bucket_region
  275. )
  276. if file_type == '0': # 图片
  277. key_name = '{user_id}/image_file/{uid}/channel{channel}/{time}/{time}.jpeg'.format(user_id=user_id,
  278. uid=uid,
  279. channel=channel,
  280. time=time_stamp)
  281. elif file_type == '2':
  282. key_name = '{user_id}/mp4_file/{uid}/channel{channel}/{time}/{time}.mp4'.format(user_id=user_id,
  283. uid=uid,
  284. channel=channel,
  285. time=time_stamp)
  286. else:
  287. return response.json(444, {'error param': 'file_type'})
  288. upload_url = s3_obj.generate_put_obj_url(bucket_name, key_name)
  289. return response.json(0, {'uploadUrl': upload_url, 'excessSize': excess_size})
  290. except Exception as e:
  291. LOGGER.info('获取s3上传地址异常:{}'.format(repr(e)))
  292. return response.json(500)
  293. @staticmethod
  294. def delete_icloud_record(request_dict, response, user_id):
  295. """
  296. 删除云盘记录
  297. @param request_dict: 请求参数
  298. @param user_id: 用户id
  299. @request_dict time_stamp: 时间戳
  300. @request_dict uid: 设备uid
  301. @request_dict channel: 设备通道
  302. @request_dict file_type: 文件类型
  303. @param response: 响应对象
  304. @return: response
  305. """
  306. LOGGER.info('删除云盘记录')
  307. time_stamp = request_dict.get('time_stamp', None)
  308. file_type = request_dict.get('file_type', None)
  309. uid = request_dict.get('uid', None)
  310. channel = request_dict.get('channel', None)
  311. if not all([time_stamp, uid, channel]):
  312. return response.json(444, {'error param': 'time_stamp,channel,uid'})
  313. # 查询是否开通云盘
  314. use_details_qs = IcloudUseDetails.objects.filter(user_id=user_id).values('use_size', 'bucket_id')
  315. if not use_details_qs.exists():
  316. return response.json(173)
  317. use_size = float(use_details_qs[0]['use_size'])
  318. bucket_id = use_details_qs[0]['bucket_id']
  319. icloud_record_qs = IcloudStorageRecord.objects.filter(user_id=user_id, uid=uid, channel=channel,
  320. time_stamp=time_stamp,
  321. bucket_id=bucket_id, file_type=file_type)
  322. if not icloud_record_qs.exists():
  323. return response.json(173)
  324. record_qs = icloud_record_qs.values('size')
  325. size = float(record_qs[0]['size'])
  326. bucket_qs = VodBucketModel.objects.filter(id=bucket_id).values('bucket', 'region', 'mold')
  327. if not bucket_qs.exists():
  328. return response.json(173)
  329. bucket_name = bucket_qs[0]['bucket']
  330. bucket_region = bucket_qs[0]['region']
  331. mold = bucket_qs[0]['mold']
  332. try:
  333. s3_obj = AmazonS3Util(
  334. AWS_ACCESS_KEY_ID[mold],
  335. AWS_SECRET_ACCESS_KEY[mold],
  336. bucket_region
  337. )
  338. new_size = round(use_size - size, 2)
  339. new_size = new_size if new_size > 0 else 0
  340. if file_type == '0': # 图片
  341. key_name = '{user_id}/image_file/{uid}/vod{channel}/{time}'.format(user_id=user_id, uid=uid,
  342. channel=channel,
  343. time=time_stamp)
  344. object_list = s3_obj.get_object_list(bucket_name, key_name)
  345. elif file_type == '1': # ts文件
  346. key_name = '{user_id}/ts_file/{uid}/vod{channel}/{time}'.format(user_id=user_id, uid=uid,
  347. channel=channel,
  348. time=time_stamp)
  349. object_list = s3_obj.get_object_list(bucket_name, key_name)
  350. elif file_type == '2': # mp4文件
  351. key_name = '{user_id}/mp4_file/{uid}/vod{channel}/{time}'.format(user_id=user_id, uid=uid,
  352. channel=channel,
  353. time=time_stamp)
  354. object_list = s3_obj.get_object_list(bucket_name, key_name)
  355. else:
  356. return response.json(444, {'error param': 'file_type'})
  357. for item in object_list:
  358. s3_obj.delete_obj(bucket_name, item['Key'])
  359. icloud_record_qs.delete()
  360. use_details_qs.update(use_size=new_size)
  361. return response.json(0)
  362. except Exception as e:
  363. LOGGER.info('删除云盘记录异常:{}'.format(repr(e)))
  364. return response.json(500)
  365. @staticmethod
  366. def get_icloud_list(request_dict, response, user_id):
  367. """
  368. 获取云盘播放列表
  369. @param request_dict: 请求参数
  370. @param user_id: 用户id
  371. @request_dict time_stamp: 时间戳
  372. @request_dict uid: 设备uid
  373. @request_dict channel: 设备通道
  374. @param response: 响应对象
  375. @return: response
  376. """
  377. LOGGER.info('删除云盘记录')
  378. start_time = request_dict.get('start_time', None)
  379. end_time = request_dict.get('end_time', None)
  380. uid = request_dict.get('uid', None)
  381. file_type = request_dict.get('file_type', None)
  382. page = request_dict.get('page', None)
  383. line = request_dict.get('line', None)
  384. if not all([page, line]):
  385. return response.json(444, {'error param': 'page, line'})
  386. page = int(page)
  387. line = int(line)
  388. now_time = int(time.time())
  389. icloud_list = []
  390. use_details_qs = IcloudUseDetails.objects.filter(user_id=user_id).values('id')
  391. if not use_details_qs.exists():
  392. return response.json(911)
  393. use_details_id = use_details_qs[0]['id']
  394. all_size = IcloudService.objects.filter(Q(use_details_id=use_details_id), Q(use_status=0),
  395. Q(end_time__gt=now_time) | Q(end_time=0)).values(
  396. 'size').aggregate(total_size=Sum('size'))['total_size']
  397. all_size = float(all_size * 1024) if all_size else 0 # 转换单位为MB
  398. icloud_record_qs = IcloudStorageRecord.objects.filter(user_id=user_id)
  399. if start_time and end_time: # 按时间段查询
  400. icloud_record_qs = icloud_record_qs.filter(time_stamp__gte=start_time, time_stamp__lt=end_time)
  401. if uid: # 按设备昵称查询
  402. uid_list = uid.split(',')
  403. icloud_record_qs = icloud_record_qs.filter(uid__in=uid_list)
  404. if file_type: # 按文件类型查询
  405. file_type_list = file_type.split(',')
  406. icloud_record_qs = icloud_record_qs.filter(file_type__in=file_type_list)
  407. icloud_record_qs = icloud_record_qs.values('bucket_id', 'sec', 'fg', 'nickname', 'uid', 'channel', 'size', 'id',
  408. 'time_stamp', 'file_type').order_by('-time_stamp')[
  409. (page - 1) * line:page * line]
  410. if not icloud_record_qs.exists():
  411. return response.json(0, {'icloudList': icloud_list, 'notShow': 0})
  412. bucket_id = icloud_record_qs[0]['bucket_id']
  413. bucket_qs = VodBucketModel.objects.filter(id=bucket_id).values('bucket', 'region', 'mold')
  414. if not bucket_qs.exists():
  415. return response.json(173)
  416. bucket_name = bucket_qs[0]['bucket']
  417. bucket_region = bucket_qs[0]['region']
  418. mold = bucket_qs[0]['mold']
  419. total_size = 0
  420. try:
  421. s3_obj = AmazonS3Util(
  422. AWS_ACCESS_KEY_ID[mold],
  423. AWS_SECRET_ACCESS_KEY[mold],
  424. bucket_region
  425. )
  426. for item in icloud_record_qs:
  427. uid = item['uid']
  428. channel = item['channel']
  429. time_stamp = item['time_stamp']
  430. file_type = item['file_type']
  431. total_size += float(item['size'])
  432. uid_token = TokenObject().encryption(data={'uid': uid, 'user_id': user_id})
  433. res_data = {
  434. 'is_show': 1 if total_size < all_size else 0, # 只展示剩余容量的数据,超过的部分不展示,1:展示;0:不展示
  435. 'time_stamp': time_stamp,
  436. 'icloud_id': item['id'],
  437. 'file_type': file_type,
  438. 'nick_name': item['nickname'],
  439. 'uid': uid
  440. }
  441. if file_type == 0: # 图片,只返回图片地址
  442. thumbs_png = '{user_id}/image_file/{uid}/channel{channel}/{time}/{time}.jpeg'.format(
  443. user_id=user_id,
  444. uid=uid,
  445. channel=channel,
  446. time=time_stamp)
  447. response_url = s3_obj.generate_file_obj_url(bucket_name, thumbs_png)
  448. res_data['thumb'] = response_url
  449. icloud_list.append(res_data)
  450. elif file_type == 1: # ts文件,返回播放地址
  451. thumbs_png = '{user_id}/image_file/{uid}/channel{channel}/{time}/{time}.jpeg'.format(
  452. user_id=user_id,
  453. uid=uid,
  454. channel=channel,
  455. time=time_stamp)
  456. response_url = s3_obj.generate_file_obj_url(bucket_name, thumbs_png)
  457. ts_num = int(item['fg']) & 0xf
  458. res_data['thumb'] = response_url
  459. icloud_url = '{server_domain}icloud/service/icloudPlay?' \
  460. 'uid={uid}&channel={channel}&time_stamp={time_stamp}&file_type=1&sign=tktktktk'. \
  461. format(server_domain=SERVER_DOMAIN_SSL, uid=uid_token, channel=channel, time_stamp=time_stamp)
  462. res_data['ts_num'] = ts_num
  463. res_data['icloud_url'] = icloud_url
  464. res_data['sec'] = item['sec']
  465. icloud_list.append(res_data)
  466. elif file_type == 2: # mp4文件,返回播放地址
  467. mp4_file = '{user_id}/mp4_file/{uid}/channel{channel}/{time}/{time}.mp4'.format(
  468. user_id=user_id,
  469. uid=uid,
  470. channel=channel,
  471. time=time_stamp)
  472. response_url = s3_obj.generate_file_obj_url(bucket_name, mp4_file)
  473. res_data['icloud_url'] = response_url
  474. res_data['sec'] = item['sec']
  475. icloud_list.append(res_data)
  476. not_show = total_size - all_size if total_size > all_size else 0
  477. return response.json(0, {'icloudList': icloud_list, 'notShow': not_show})
  478. except Exception as e:
  479. LOGGER.info('获取云盘播放列表异常:{}'.format(repr(e)))
  480. return response.json(500)
  481. @staticmethod
  482. def icloud_play(request_dict, response):
  483. """
  484. 云盘播放
  485. @param request_dict: 请求参数
  486. @request_dict time_stamp: 时间戳
  487. @request_dict uid: 设备uid
  488. @request_dict channel: 设备通道
  489. @param response: 响应对象
  490. @return: response
  491. """
  492. LOGGER.info('播放云盘记录')
  493. time_stamp = request_dict.get('time_stamp', None)
  494. uid_token = request_dict.get('uid', None)
  495. channel = request_dict.get('channel', None)
  496. file_type = request_dict.get('file_type', None)
  497. if not all([time_stamp, uid_token, channel, file_type]):
  498. return response.json(444, {'error param': 'time_stamp, uid_token, channel, file_type'})
  499. if file_type != '1':
  500. return response.json(444, {'error param': 'file_type'})
  501. token = jwt.decode(uid_token, OAUTH_ACCESS_TOKEN_SECRET, algorithms='HS256')
  502. uid = token.get('uid', '')
  503. user_id = token.get('user_id', '')
  504. icloud_record_qs = IcloudStorageRecord.objects.filter(user_id=user_id, uid=uid, channel=channel,
  505. file_type=file_type,
  506. time_stamp=time_stamp).values('bucket_id', 'fg')
  507. if not icloud_record_qs.exists():
  508. return response.json(173)
  509. bucket_id = icloud_record_qs[0]['bucket_id']
  510. fg = int(icloud_record_qs[0]['fg'])
  511. bucket_qs = VodBucketModel.objects.filter(id=bucket_id).values('bucket', 'region', 'mold')
  512. if not bucket_qs.exists():
  513. return response.json(173)
  514. bucket_name = bucket_qs[0]['bucket']
  515. bucket_region = bucket_qs[0]['region']
  516. mold = bucket_qs[0]['mold']
  517. try:
  518. s3_obj = AmazonS3Util(
  519. AWS_ACCESS_KEY_ID[mold],
  520. AWS_SECRET_ACCESS_KEY[mold],
  521. bucket_region
  522. )
  523. playlist_entries = []
  524. for i in range(15):
  525. shift = (i + 1) * 4
  526. duration = (fg >> shift) & 0xf
  527. if duration > 0:
  528. ts_file = '{user_id}/ts_file/{uid}/channel{channel}/{time_stamp}/ts{i}.ts'.format(
  529. user_id=user_id, uid=uid, channel=channel, time_stamp=time_stamp, i=i)
  530. response_url = s3_obj.generate_file_obj_url(bucket_name, ts_file)
  531. playlist_entries.append({
  532. 'name': response_url,
  533. 'duration': duration,
  534. })
  535. playlist = PlaylistGenerator(playlist_entries).generate()
  536. response = HttpResponse(playlist)
  537. response['Content-Type'] = 'application/octet-stream'
  538. response['Content-Disposition'] = 'attachment;filename="play.m3u8"'
  539. return response
  540. except Exception as e:
  541. LOGGER.info('云盘播放异常:{}'.format(repr(e)))
  542. return response.json(500)