IcloudService.py 29 KB

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