IcloudService.py 29 KB

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