IcloudService.py 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586
  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. record_ids = request_dict.get('record_ids', None)
  311. if not all([record_ids]):
  312. return response.json(444, {'error param': 'record_id'})
  313. record_id_list = record_ids.split(',')
  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, id__in=record_id_list)
  321. if not icloud_record_qs.exists():
  322. return response.json(173)
  323. record_qs = icloud_record_qs.values('size', 'uid', 'channel', 'time_stamp', 'file_type')
  324. bucket_qs = VodBucketModel.objects.filter(id=bucket_id).values('bucket', 'region', 'mold')
  325. if not bucket_qs.exists():
  326. return response.json(173)
  327. bucket_name = bucket_qs[0]['bucket']
  328. bucket_region = bucket_qs[0]['region']
  329. mold = bucket_qs[0]['mold']
  330. try:
  331. s3_obj = AmazonS3Util(
  332. AWS_ACCESS_KEY_ID[mold],
  333. AWS_SECRET_ACCESS_KEY[mold],
  334. bucket_region
  335. )
  336. for item in record_qs:
  337. size = float(item['size'])
  338. uid = item['uid']
  339. channel = item['channel']
  340. time_stamp = item['time_stamp']
  341. file_type = item['file_type']
  342. use_size = round(use_size - size, 2)
  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 obj in object_list:
  359. s3_obj.delete_obj(bucket_name, obj['Key'])
  360. use_size = use_size if use_size > 0 else 0
  361. icloud_record_qs.delete()
  362. use_details_qs.update(use_size=use_size)
  363. return response.json(0)
  364. except Exception as e:
  365. LOGGER.info('删除云盘记录异常:{}'.format(repr(e)))
  366. return response.json(500)
  367. @staticmethod
  368. def get_icloud_list(request_dict, response, user_id):
  369. """
  370. 获取云盘播放列表
  371. @param request_dict: 请求参数
  372. @param user_id: 用户id
  373. @request_dict time_stamp: 时间戳
  374. @request_dict uid: 设备uid
  375. @request_dict channel: 设备通道
  376. @param response: 响应对象
  377. @return: response
  378. """
  379. LOGGER.info('删除云盘记录')
  380. start_time = request_dict.get('start_time', None)
  381. end_time = request_dict.get('end_time', None)
  382. uid = request_dict.get('uid', None)
  383. file_type = request_dict.get('file_type', None)
  384. page = request_dict.get('page', None)
  385. line = request_dict.get('line', None)
  386. if not all([page, line]):
  387. return response.json(444, {'error param': 'page, line'})
  388. page = int(page)
  389. line = int(line)
  390. now_time = int(time.time())
  391. icloud_list = []
  392. use_details_qs = IcloudUseDetails.objects.filter(user_id=user_id).values('id')
  393. if not use_details_qs.exists():
  394. return response.json(911)
  395. use_details_id = use_details_qs[0]['id']
  396. all_size = IcloudService.objects.filter(Q(use_details_id=use_details_id), Q(use_status=0),
  397. Q(end_time__gt=now_time) | Q(end_time=0)).values(
  398. 'size').aggregate(total_size=Sum('size'))['total_size']
  399. all_size = float(all_size * 1024) if all_size else 0 # 转换单位为MB
  400. icloud_record_qs = IcloudStorageRecord.objects.filter(user_id=user_id)
  401. if start_time and end_time: # 按时间段查询
  402. icloud_record_qs = icloud_record_qs.filter(time_stamp__gte=start_time, time_stamp__lt=end_time)
  403. if uid: # 按设备昵称查询
  404. uid_list = uid.split(',')
  405. icloud_record_qs = icloud_record_qs.filter(uid__in=uid_list)
  406. if file_type: # 按文件类型查询
  407. file_type_list = file_type.split(',')
  408. icloud_record_qs = icloud_record_qs.filter(file_type__in=file_type_list)
  409. icloud_record_qs = icloud_record_qs.values('bucket_id', 'sec', 'fg', 'nickname', 'uid', 'channel', 'size', 'id',
  410. 'time_stamp', 'file_type').order_by('-time_stamp')[
  411. (page - 1) * line:page * line]
  412. if not icloud_record_qs.exists():
  413. return response.json(0, {'icloudList': icloud_list, 'notShow': 0})
  414. bucket_id = icloud_record_qs[0]['bucket_id']
  415. bucket_qs = VodBucketModel.objects.filter(id=bucket_id).values('bucket', 'region', 'mold')
  416. if not bucket_qs.exists():
  417. return response.json(173)
  418. bucket_name = bucket_qs[0]['bucket']
  419. bucket_region = bucket_qs[0]['region']
  420. mold = bucket_qs[0]['mold']
  421. total_size = 0
  422. try:
  423. s3_obj = AmazonS3Util(
  424. AWS_ACCESS_KEY_ID[mold],
  425. AWS_SECRET_ACCESS_KEY[mold],
  426. bucket_region
  427. )
  428. for item in icloud_record_qs:
  429. uid = item['uid']
  430. channel = item['channel']
  431. time_stamp = item['time_stamp']
  432. file_type = item['file_type']
  433. total_size += float(item['size'])
  434. uid_token = TokenObject().encryption(data={'uid': uid, 'user_id': user_id})
  435. res_data = {
  436. 'is_show': 1 if total_size < all_size else 0, # 只展示剩余容量的数据,超过的部分不展示,1:展示;0:不展示
  437. 'time_stamp': time_stamp,
  438. 'icloud_id': item['id'],
  439. 'file_type': file_type,
  440. 'nick_name': item['nickname'],
  441. 'uid': uid
  442. }
  443. if file_type == 0: # 图片,只返回图片地址
  444. thumbs_png = '{user_id}/image_file/{uid}/channel{channel}/{time}/{time}.jpeg'.format(
  445. user_id=user_id,
  446. uid=uid,
  447. channel=channel,
  448. time=time_stamp)
  449. response_url = s3_obj.generate_file_obj_url(bucket_name, thumbs_png)
  450. res_data['thumb'] = response_url
  451. icloud_list.append(res_data)
  452. elif file_type == 1: # ts文件,返回播放地址
  453. thumbs_png = '{user_id}/ts_file/{uid}/channel{channel}/{time}/Thumb.jpeg'.format(
  454. user_id=user_id,
  455. uid=uid,
  456. channel=channel,
  457. time=time_stamp)
  458. response_url = s3_obj.generate_file_obj_url(bucket_name, thumbs_png)
  459. ts_num = int(item['fg']) & 0xf
  460. res_data['thumb'] = response_url
  461. icloud_url = '{server_domain}icloud/service/icloudPlay?' \
  462. 'uid={uid}&channel={channel}&time_stamp={time_stamp}&file_type=1&sign=tktktktk'. \
  463. format(server_domain=SERVER_DOMAIN_SSL, uid=uid_token, channel=channel, time_stamp=time_stamp)
  464. res_data['ts_num'] = ts_num
  465. res_data['icloud_url'] = icloud_url
  466. res_data['sec'] = item['sec']
  467. icloud_list.append(res_data)
  468. elif file_type == 2: # mp4文件,返回播放地址
  469. mp4_file = '{user_id}/mp4_file/{uid}/channel{channel}/{time}/{time}.mp4'.format(
  470. user_id=user_id,
  471. uid=uid,
  472. channel=channel,
  473. time=time_stamp)
  474. response_url = s3_obj.generate_file_obj_url(bucket_name, mp4_file)
  475. res_data['icloud_url'] = response_url
  476. res_data['sec'] = item['sec']
  477. icloud_list.append(res_data)
  478. not_show = total_size - all_size if total_size > all_size else 0
  479. return response.json(0, {'icloudList': icloud_list, 'notShow': not_show})
  480. except Exception as e:
  481. LOGGER.info('获取云盘播放列表异常:{}'.format(repr(e)))
  482. return response.json(500)
  483. @staticmethod
  484. def icloud_play(request_dict, response):
  485. """
  486. 云盘播放
  487. @param request_dict: 请求参数
  488. @request_dict time_stamp: 时间戳
  489. @request_dict uid: 设备uid
  490. @request_dict channel: 设备通道
  491. @param response: 响应对象
  492. @return: response
  493. """
  494. LOGGER.info('播放云盘记录')
  495. time_stamp = request_dict.get('time_stamp', None)
  496. uid_token = request_dict.get('uid', None)
  497. channel = request_dict.get('channel', None)
  498. file_type = request_dict.get('file_type', None)
  499. if not all([time_stamp, uid_token, channel, file_type]):
  500. return response.json(444, {'error param': 'time_stamp, uid_token, channel, file_type'})
  501. if file_type != '1':
  502. return response.json(444, {'error param': 'file_type'})
  503. token = jwt.decode(uid_token, OAUTH_ACCESS_TOKEN_SECRET, algorithms='HS256')
  504. uid = token.get('uid', '')
  505. user_id = token.get('user_id', '')
  506. icloud_record_qs = IcloudStorageRecord.objects.filter(user_id=user_id, uid=uid, channel=channel,
  507. file_type=file_type,
  508. time_stamp=time_stamp).values('bucket_id', 'fg')
  509. if not icloud_record_qs.exists():
  510. return response.json(173)
  511. bucket_id = icloud_record_qs[0]['bucket_id']
  512. fg = int(icloud_record_qs[0]['fg'])
  513. bucket_qs = VodBucketModel.objects.filter(id=bucket_id).values('bucket', 'region', 'mold')
  514. if not bucket_qs.exists():
  515. return response.json(173)
  516. bucket_name = bucket_qs[0]['bucket']
  517. bucket_region = bucket_qs[0]['region']
  518. mold = bucket_qs[0]['mold']
  519. try:
  520. s3_obj = AmazonS3Util(
  521. AWS_ACCESS_KEY_ID[mold],
  522. AWS_SECRET_ACCESS_KEY[mold],
  523. bucket_region
  524. )
  525. playlist_entries = []
  526. for i in range(15):
  527. shift = (i + 1) * 4
  528. duration = (fg >> shift) & 0xf
  529. if duration > 0:
  530. ts_file = '{user_id}/ts_file/{uid}/channel{channel}/{time_stamp}/ts{i}.ts'.format(
  531. user_id=user_id, uid=uid, channel=channel, time_stamp=time_stamp, i=i)
  532. response_url = s3_obj.generate_file_obj_url(bucket_name, ts_file)
  533. playlist_entries.append({
  534. 'name': response_url,
  535. 'duration': duration,
  536. })
  537. playlist = PlaylistGenerator(playlist_entries).generate()
  538. response = HttpResponse(playlist)
  539. response['Content-Type'] = 'application/octet-stream'
  540. response['Content-Disposition'] = 'attachment;filename="play.m3u8"'
  541. return response
  542. except Exception as e:
  543. LOGGER.info('云盘播放异常:{}'.format(repr(e)))
  544. return response.json(500)
  545. @staticmethod
  546. def do_change_icloud_status(request_dict, response, user_id):
  547. """
  548. 修改云盘开关
  549. @param request_dict: 请求参数
  550. @param user_id: 用户id
  551. @request_dict status: 开关状态, 0: 关闭, 1: 开启
  552. @param response: 响应对象
  553. @return: response
  554. """
  555. status = request_dict.get('status', None)
  556. if not status:
  557. return response.json(444, 'status')
  558. try:
  559. device_user_qs = Device_User.objects.filter(userID=user_id)
  560. if not device_user_qs.exists():
  561. return response.json(173)
  562. icloud_use_details_qs = IcloudUseDetails.objects.filter(user_id=user_id)
  563. icloud_use_details_qs.update(detect_status=status)
  564. return response.json(0)
  565. except Exception as e:
  566. print(e)
  567. return response.json(500)