IcloudService.py 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555
  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. nick_name = request_dict.get('nick_name', 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 nick_name: # 按设备昵称查询
  402. icloud_record_qs = icloud_record_qs.filter(nickname=nick_name)
  403. if file_type: # 按文件类型查询
  404. icloud_record_qs = icloud_record_qs.filter(file_type=file_type)
  405. icloud_record_qs = icloud_record_qs.values('bucket_id', 'sec', 'fg', 'nickname', 'uid', 'channel', 'size', 'id',
  406. 'time_stamp', 'file_type').order_by('-time_stamp')[
  407. (page - 1) * line:page * line]
  408. if not icloud_record_qs.exists():
  409. return response.json(0, {'icloudList': icloud_list, 'notShow': 0})
  410. bucket_id = icloud_record_qs[0]['bucket_id']
  411. bucket_qs = VodBucketModel.objects.filter(id=bucket_id).values('bucket', 'region', 'mold')
  412. if not bucket_qs.exists():
  413. return response.json(173)
  414. bucket_name = bucket_qs[0]['bucket']
  415. bucket_region = bucket_qs[0]['region']
  416. mold = bucket_qs[0]['mold']
  417. total_size = 0
  418. try:
  419. s3_obj = AmazonS3Util(
  420. AWS_ACCESS_KEY_ID[mold],
  421. AWS_SECRET_ACCESS_KEY[mold],
  422. bucket_region
  423. )
  424. for item in icloud_record_qs:
  425. uid = item['uid']
  426. channel = item['channel']
  427. time_stamp = item['time_stamp']
  428. file_type = item['file_type']
  429. total_size += float(item['size'])
  430. uid_token = TokenObject().encryption(data={'uid': uid, 'user_id': user_id})
  431. res_data = {
  432. 'is_show': 1 if total_size < all_size else 0, # 只展示剩余容量的数据,超过的部分不展示
  433. 'time_stamp': time_stamp,
  434. 'icloud_id': item['id'],
  435. 'file_type': file_type,
  436. 'nick_name': item['nickname']
  437. }
  438. if file_type == 0: # 图片,只返回图片地址
  439. thumbs_png = '{user_id}/image_file/{uid}/channel{channel}/{time}/{time}.jpeg'.format(
  440. user_id=user_id,
  441. uid=uid,
  442. channel=channel,
  443. time=time_stamp)
  444. response_url = s3_obj.generate_file_obj_url(bucket_name, thumbs_png)
  445. res_data['thumb'] = response_url
  446. icloud_list.append(res_data)
  447. elif file_type == 1: # ts文件,返回播放地址
  448. thumbs_png = '{user_id}/image_file/{uid}/channel{channel}/{time}/{time}.jpeg'.format(
  449. user_id=user_id,
  450. uid=uid,
  451. channel=channel,
  452. time=time_stamp)
  453. response_url = s3_obj.generate_file_obj_url(bucket_name, thumbs_png)
  454. ts_num = int(item['fg']) & 0xf
  455. res_data['thumb'] = response_url
  456. icloud_url = '{server_domain}icloud/service/icloudPlay?' \
  457. 'uid={uid}&channel={channel}&time_stamp={time_stamp}&file_type=1&sign=tktktktk'. \
  458. format(server_domain=SERVER_DOMAIN_SSL, uid=uid_token, channel=channel, time_stamp=time_stamp)
  459. res_data['ts_num'] = ts_num
  460. res_data['icloud_url'] = icloud_url
  461. res_data['sec'] = item['sec']
  462. icloud_list.append(res_data)
  463. elif file_type == 2: # mp4文件,返回播放地址
  464. mp4_file = '{user_id}/mp4_file/{uid}/channel{channel}/{time}/{time}.mp4'.format(
  465. user_id=user_id,
  466. uid=uid,
  467. channel=channel,
  468. time=time_stamp)
  469. response_url = s3_obj.generate_file_obj_url(bucket_name, mp4_file)
  470. res_data['icloud_url'] = response_url
  471. res_data['sec'] = item['sec']
  472. icloud_list.append(res_data)
  473. not_show = total_size - all_size if total_size > all_size else 0
  474. return response.json(0, {'icloudList': icloud_list, 'notShow': not_show})
  475. except Exception as e:
  476. LOGGER.info('获取云盘播放列表异常:{}'.format(repr(e)))
  477. return response.json(500)
  478. @staticmethod
  479. def icloud_play(request_dict, response):
  480. """
  481. 云盘播放
  482. @param request_dict: 请求参数
  483. @request_dict time_stamp: 时间戳
  484. @request_dict uid: 设备uid
  485. @request_dict channel: 设备通道
  486. @param response: 响应对象
  487. @return: response
  488. """
  489. LOGGER.info('播放云盘记录')
  490. time_stamp = request_dict.get('time_stamp', None)
  491. uid_token = request_dict.get('uid', None)
  492. channel = request_dict.get('channel', None)
  493. file_type = request_dict.get('file_type', None)
  494. if not all([time_stamp, uid_token, channel, file_type]):
  495. return response.json(444, {'error param': 'time_stamp, uid_token, channel, file_type'})
  496. if file_type != '1':
  497. return response.json(444, {'error param': 'file_type'})
  498. token = jwt.decode(uid_token, OAUTH_ACCESS_TOKEN_SECRET, algorithms='HS256')
  499. uid = token.get('uid', '')
  500. user_id = token.get('user_id', '')
  501. icloud_record_qs = IcloudStorageRecord.objects.filter(user_id=user_id, uid=uid, channel=channel,
  502. file_type=file_type,
  503. time_stamp=time_stamp).values('bucket_id', 'fg')
  504. if not icloud_record_qs.exists():
  505. return response.json(173)
  506. bucket_id = icloud_record_qs[0]['bucket_id']
  507. fg = int(icloud_record_qs[0]['fg'])
  508. bucket_qs = VodBucketModel.objects.filter(id=bucket_id).values('bucket', 'region', 'mold')
  509. if not bucket_qs.exists():
  510. return response.json(173)
  511. bucket_name = bucket_qs[0]['bucket']
  512. bucket_region = bucket_qs[0]['region']
  513. mold = bucket_qs[0]['mold']
  514. try:
  515. s3_obj = AmazonS3Util(
  516. AWS_ACCESS_KEY_ID[mold],
  517. AWS_SECRET_ACCESS_KEY[mold],
  518. bucket_region
  519. )
  520. playlist_entries = []
  521. for i in range(15):
  522. shift = (i + 1) * 4
  523. duration = (fg >> shift) & 0xf
  524. if duration > 0:
  525. ts_file = '{user_id}/ts_file/{uid}/channel{channel}/{time_stamp}/ts{i}.ts'.format(
  526. user_id=user_id, uid=uid, channel=channel, time_stamp=time_stamp, i=i)
  527. response_url = s3_obj.generate_file_obj_url(bucket_name, ts_file)
  528. playlist_entries.append({
  529. 'name': response_url,
  530. 'duration': duration,
  531. })
  532. playlist = PlaylistGenerator(playlist_entries).generate()
  533. response = HttpResponse(playlist)
  534. response['Content-Type'] = 'application/octet-stream'
  535. response['Content-Disposition'] = 'attachment;filename="play.m3u8"'
  536. return response
  537. except Exception as e:
  538. LOGGER.info('云盘播放异常:{}'.format(repr(e)))
  539. return response.json(500)