UserDeviceShareController.py 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470
  1. # -*- encoding: utf-8 -*-
  2. """
  3. @File : UserDeviceShareController.py
  4. @Time : 2023/1/7 15:05
  5. @Author : stephen
  6. @Email : zhangdongming@asj6.wecom.work
  7. @Software: PyCharm
  8. """
  9. import logging
  10. import time
  11. import boto3
  12. import botocore
  13. from botocore import client
  14. from django.db import transaction
  15. from django.db.models import Q
  16. from django.views import View
  17. from Model.models import DeviceSharePermission, DeviceChannelUserSet, DeviceChannelUserPermission, UidChannelSetModel, \
  18. Device_Info
  19. from Object.ResponseObject import ResponseObject
  20. from Object.TokenObject import TokenObject
  21. from Service.UserDeviceService import UserDeviceService
  22. from Ansjer.config import CONFIG_CN, CONFIG_INFO, CONFIG_TEST, AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, \
  23. AWS_SES_ACCESS_REGION, AWS_IOT_SES_ACCESS_CHINA_REGION
  24. from Model.models import DeviceWallpaper
  25. from Object.AWS.AmazonS3Util import AmazonS3Util
  26. LOGGER = logging.getLogger('info')
  27. class UserDeviceShareView(View):
  28. def get(self, request, *args, **kwargs):
  29. request.encoding = 'utf-8'
  30. operation = kwargs.get('operation')
  31. return self.validation(request.GET, request, operation)
  32. def post(self, request, *args, **kwargs):
  33. request.encoding = 'utf-8'
  34. operation = kwargs.get('operation')
  35. return self.validation(request.POST, request, operation)
  36. def validation(self, request_dict, request, operation):
  37. token = TokenObject(request.META.get('HTTP_AUTHORIZATION'))
  38. lang = request_dict.get('lang', token.lang)
  39. response = ResponseObject(lang)
  40. if token.code != 0:
  41. return response.json(token.code)
  42. if operation == 'user-permissions':
  43. return self.get_user_share_permission(request_dict, response)
  44. elif operation == 'permissions-save':
  45. return self.user_channel_permission_save(request_dict, response)
  46. elif operation == 'permissions-test':
  47. return self.synch_share_device_permission(response)
  48. elif operation == 'getWallpaperList':
  49. return self.get_wallpaper_list(request_dict, response)
  50. elif operation == 'getUploadWallpaper':
  51. return self.get_upload_wallpaper(request_dict, response)
  52. elif operation == 'notifyUploadWallpaper':
  53. return self.notify_upload_wallpaper(request_dict, response)
  54. elif operation == 'delWallpaper':
  55. return self.del_wallpaper(request_dict, response)
  56. elif operation == 'selectWallpaper':
  57. return self.select_wallpaper(request_dict, response)
  58. else:
  59. return response.json(404)
  60. @classmethod
  61. def get_user_share_permission(cls, request_dict, response):
  62. """
  63. 获取用户分享权限
  64. @param request_dict: 设备uid
  65. @param response: 响应对象
  66. @return: permission List
  67. """
  68. try:
  69. uid = request_dict.get('uid', None)
  70. user_id = request_dict.get('userId', None)
  71. channel_count = request_dict.get('channelCount', None)
  72. if not all([user_id, uid, channel_count]):
  73. return response(444, 'uid, userId and channelCount is required')
  74. user_permission_qs = DeviceChannelUserSet.objects.filter(user_id=user_id, uid=uid) \
  75. .values('id', 'channels')
  76. if not user_permission_qs.exists():
  77. return response.json(0, {})
  78. up_id = user_permission_qs[0]['id']
  79. channel_permission_qs = DeviceChannelUserPermission.objects.filter(channel_user_id=up_id) \
  80. .values('permission_id', 'channel_user_id')
  81. if not channel_permission_qs.exists():
  82. return response.json(0, {})
  83. channel_list = list(range(1, int(channel_count) + 1))
  84. share_channel_list = user_permission_qs[0]['channels'].split(',')
  85. c_list = []
  86. for channel in channel_list:
  87. is_select = 1 if str(channel) in share_channel_list else 0
  88. c_list.append({'channelIndex': channel, 'isSelect': is_select})
  89. p_list = []
  90. permission_qs = DeviceSharePermission.objects.all()
  91. share_permission_list = [item['permission_id'] for item in channel_permission_qs]
  92. for item in permission_qs:
  93. is_select = 1 if item.id in share_permission_list else 0
  94. p_list.append({'permissionId': item.id, 'code': item.code, 'isSelect': is_select})
  95. data = {'channels': c_list, 'permissions': p_list}
  96. return response.json(0, data)
  97. except Exception as e:
  98. LOGGER.info('异常详情,errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  99. return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  100. @classmethod
  101. def user_channel_permission_save(cls, request_dict, response):
  102. """
  103. 主用户分享设备时设置通道权限保存
  104. """
  105. try:
  106. uid = request_dict.get('uid', None)
  107. channels = request_dict.get('channels', None) # 通道集合,多个','隔开
  108. user_id = request_dict.get('userId', None)
  109. permission_ids = request_dict.get('permissionIds', None) # 权限集合,多个','隔开
  110. if not all([user_id, uid, channels, permission_ids]):
  111. return response.json(444)
  112. p_ids = []
  113. device_user_set = DeviceChannelUserSet.objects.filter(user_id=user_id, uid=uid)
  114. now_time = int(time.time())
  115. with transaction.atomic():
  116. is_delete = False
  117. if not device_user_set.exists():
  118. device_set = {'uid': uid, 'user_id': user_id, 'channels': channels,
  119. 'created_time': now_time, 'updated_time': now_time}
  120. device_user_set = DeviceChannelUserSet.objects.create(**device_set)
  121. channel_user_id = device_user_set.id
  122. else:
  123. DeviceChannelUserSet.objects.update(channels=channels, updated_time=now_time)
  124. channel_user_id = device_user_set.first().id
  125. is_delete = True
  126. if ',' in permission_ids:
  127. p_ids = [int(val) for val in permission_ids.split(',')]
  128. if is_delete:
  129. DeviceChannelUserPermission.objects.filter(
  130. channel_user_id=channel_user_id).delete()
  131. if not p_ids:
  132. channel_permission = {'permission_id': int(permission_ids),
  133. 'channel_user_id': channel_user_id,
  134. 'created_time': now_time}
  135. DeviceChannelUserPermission.objects.create(**channel_permission)
  136. else:
  137. channel_permission_list = []
  138. for item in p_ids:
  139. channel_permission_list.append(DeviceChannelUserPermission(
  140. permission_id=int(item),
  141. channel_user_id=channel_user_id,
  142. created_time=now_time))
  143. DeviceChannelUserPermission.objects.bulk_create(channel_permission_list)
  144. return response.json(0)
  145. except Exception as e:
  146. LOGGER.info('异常详情,errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  147. return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  148. @staticmethod
  149. def qrcode_share_channel_permission_save(user_id, uid):
  150. """
  151. 二维码分享保存通道权限
  152. @param user_id: 用户id
  153. @param uid: 用户设备ID
  154. @return: True | False
  155. """
  156. try:
  157. if not all([user_id, uid]):
  158. return False
  159. with transaction.atomic():
  160. ds_qs = DeviceChannelUserSet.objects.filter(user_id=user_id, uid=uid)
  161. if ds_qs.exists():
  162. return True
  163. UserDeviceService.update_device_channel(uid)
  164. channel_qs = UidChannelSetModel.objects.filter(uid__uid=uid).values('channel')
  165. if not channel_qs.exists():
  166. return False
  167. channel_list = [str(val['channel']) for val in channel_qs]
  168. channel_str = ','.join(channel_list)
  169. now_time = int(time.time())
  170. device_set = {'uid': uid, 'user_id': user_id, 'channels': channel_str,
  171. 'created_time': now_time, 'updated_time': now_time}
  172. device_user_set = DeviceChannelUserSet.objects.create(**device_set)
  173. channel_permission_qs = DeviceSharePermission.objects \
  174. .all().values('id', 'code').order_by('sort')
  175. user_set_id = device_user_set.id
  176. channel_permission_list = []
  177. for item in channel_permission_qs:
  178. channel_permission_list.append(DeviceChannelUserPermission(
  179. permission_id=item['id'],
  180. channel_user_id=user_set_id,
  181. created_time=now_time))
  182. DeviceChannelUserPermission.objects.bulk_create(channel_permission_list)
  183. return True
  184. except Exception as e:
  185. LOGGER.info('异常详情,errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  186. return False
  187. @staticmethod
  188. def synch_share_device_permission(response):
  189. """
  190. 同步分析设备权限
  191. @param response: 响应结果
  192. """
  193. device_info_qs = Device_Info.objects \
  194. .filter(~Q(Type__in=[1, 2, 3, 4, 10001]), ~Q(primaryUserID=''), isShare=1) \
  195. .values('userID_id', 'UID').order_by('-data_joined')
  196. if not device_info_qs.exists():
  197. return response.json(0)
  198. for item in device_info_qs:
  199. UserDeviceShareView.qrcode_share_channel_permission_save(item['userID_id'], item['UID'])
  200. return response.json(0)
  201. @classmethod
  202. def get_wallpaper_list(cls, request_dict, response):
  203. """
  204. 获取设备壁纸列表
  205. @param request_dict:
  206. @param response:
  207. @return:
  208. """
  209. try:
  210. device_type = int(request_dict.get('deviceType', None))
  211. uid = request_dict.get('uid', None)
  212. channel = int(request_dict.get('channel', 1))
  213. if not all([device_type, uid]):
  214. return response.json(444)
  215. # 查询用户自定义壁纸
  216. user_wallpaper_qs = DeviceWallpaper.objects.filter(channel=channel, uid=uid,
  217. device_type=device_type, parent_id=0, status=1)
  218. # 查询系统默认壁纸
  219. def_wallpaper_qs = DeviceWallpaper.objects.filter(classification=1, device_type=device_type, status=1)
  220. # 查询用户选中的壁纸
  221. user_checked_qs = DeviceWallpaper.objects.filter(channel=channel, uid=uid,
  222. device_type=device_type, parent_id__gt=0, status=1)
  223. checked_id = user_checked_qs[0].parent_id if user_checked_qs.exists() else def_wallpaper_qs.first().id
  224. wallpaper_list = []
  225. if def_wallpaper_qs.exists() or user_wallpaper_qs.exists():
  226. # 初始化存储桶客户端
  227. if CONFIG_CN == CONFIG_INFO or CONFIG_TEST == CONFIG_INFO:
  228. s3 = boto3.client(
  229. 's3',
  230. aws_access_key_id=AWS_ACCESS_KEY_ID[0],
  231. aws_secret_access_key=AWS_SECRET_ACCESS_KEY[0],
  232. config=botocore.client.Config(signature_version='s3v4'),
  233. region_name='cn-northwest-1'
  234. )
  235. else:
  236. s3 = boto3.client(
  237. 's3',
  238. aws_access_key_id=AWS_ACCESS_KEY_ID[1],
  239. aws_secret_access_key=AWS_SECRET_ACCESS_KEY[1],
  240. config=botocore.client.Config(signature_version='s3v4'),
  241. region_name='us-east-1'
  242. )
  243. # 处理系统默认壁纸和用户自定义壁纸
  244. all_wallpapers_qs = def_wallpaper_qs.union(user_wallpaper_qs)
  245. for item in all_wallpapers_qs:
  246. obj_key = item.obj_prefix + item.obj_name
  247. params = {
  248. 'Key': obj_key,
  249. 'Bucket': 'ansjerfilemanager',
  250. }
  251. response_url = s3.generate_presigned_url(
  252. 'get_object', Params=params, ExpiresIn=3600)
  253. wallpaper = {
  254. 'id': item.id,
  255. 'url': response_url,
  256. 'state': 1 if checked_id == item.id else 0,
  257. 'classification': item.classification
  258. }
  259. wallpaper_list.append(wallpaper)
  260. return response.json(0, {'wallpapers': wallpaper_list})
  261. except Exception as e:
  262. LOGGER.error('查询设备壁纸异常:errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  263. return response.json(5)
  264. @classmethod
  265. def get_upload_wallpaper(cls, request_dict, response):
  266. """
  267. 获取设备壁纸上传链接
  268. @param request_dict:
  269. @param response:
  270. @return:
  271. """
  272. try:
  273. uid = request_dict.get('uid', None)
  274. channel = int(request_dict.get('channel', 1))
  275. if not uid:
  276. return response.json(444)
  277. # 初始化存储桶客户端
  278. if CONFIG_CN == CONFIG_INFO or CONFIG_TEST == CONFIG_INFO:
  279. s3 = boto3.client(
  280. 's3',
  281. aws_access_key_id=AWS_ACCESS_KEY_ID[0],
  282. aws_secret_access_key=AWS_SECRET_ACCESS_KEY[0],
  283. config=botocore.client.Config(signature_version='s3v4'),
  284. region_name='cn-northwest-1'
  285. )
  286. else:
  287. s3 = boto3.client(
  288. 's3',
  289. aws_access_key_id=AWS_ACCESS_KEY_ID[1],
  290. aws_secret_access_key=AWS_SECRET_ACCESS_KEY[1],
  291. config=botocore.client.Config(signature_version='s3v4'),
  292. region_name='us-east-1'
  293. )
  294. # 生成唯一的文件名
  295. file_name = f"{int(time.time())}.png"
  296. obj_key = f"app/static/device-wallpaper/{uid}/{file_name}"
  297. # 生成预签名的 URL
  298. presigned_url = s3.generate_presigned_url(
  299. 'put_object',
  300. Params={
  301. 'Bucket': 'ansjerfilemanager',
  302. 'Key': obj_key,
  303. },
  304. ExpiresIn=600
  305. )
  306. # 返回预签名的URL和字段
  307. return response.json(0, {
  308. 'uploadUrl': presigned_url,
  309. 'fileName': file_name,
  310. })
  311. except Exception as e:
  312. LOGGER.error('获取上传链接异常:errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  313. return response.json(5)
  314. @classmethod
  315. def notify_upload_wallpaper(cls, request_dict, response):
  316. """
  317. 确认壁纸上传成功,创建DeviceWallpaper
  318. @param request_dict:
  319. @param response:
  320. @return:
  321. """
  322. try:
  323. uid = request_dict.get('uid', None)
  324. file_name = request_dict.get('fileName', None)
  325. channel = int(request_dict.get('channel', 1))
  326. if not all([uid, file_name]):
  327. return response.json(444)
  328. device_info_qs = Device_Info.objects.filter(UID=uid).values('id', 'Type')
  329. device_type = device_info_qs[0]['Type'] if device_info_qs.exists() else ''
  330. DeviceWallpaper.objects.create(
  331. device_type=device_type,
  332. storage_type=1,
  333. channel=channel,
  334. uid=uid,
  335. obj_prefix=f'app/static/device-wallpaper/{uid}/',
  336. obj_name=file_name,
  337. classification=2, # 自定义壁纸分类
  338. status=1,
  339. parent_id=0,
  340. created_time=int(time.time())
  341. )
  342. return response.json(0)
  343. except Exception as e:
  344. LOGGER.error('壁纸创建失败:errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  345. return response.json(5)
  346. @classmethod
  347. def del_wallpaper(cls, request_dict, response):
  348. """
  349. 删除设备壁纸
  350. """
  351. wallpaper_id = request_dict.get('wallpaperId', None)
  352. if not all([wallpaper_id]):
  353. return response.json(444)
  354. try:
  355. wallpaper_id = int(wallpaper_id)
  356. if DeviceWallpaper.objects.filter(id=wallpaper_id, classification=1).exists():
  357. return response.json(176, "系统图片不能删除")
  358. device_wallpaper_qs = DeviceWallpaper.objects.filter(id=wallpaper_id)
  359. device_wallpaper = device_wallpaper_qs.values("obj_prefix", "obj_name", "uid", "parent_id")
  360. uid = device_wallpaper[0]['uid']
  361. obj_name = device_wallpaper[0]['obj_name']
  362. # 使用中壁纸被删除
  363. use_device_wallpaper = DeviceWallpaper.objects.filter(uid=uid, parent_id__gt=0).first()
  364. if use_device_wallpaper and use_device_wallpaper.parent_id == wallpaper_id:
  365. system_device_wallpaper = DeviceWallpaper.objects.filter(id=1).first()
  366. use_device_wallpaper.parent_id = 1
  367. use_device_wallpaper.storage_type = system_device_wallpaper.storage_type
  368. use_device_wallpaper.obj_name = system_device_wallpaper.obj_name
  369. use_device_wallpaper.obj_prefix = system_device_wallpaper.obj_prefix
  370. use_device_wallpaper.classification = system_device_wallpaper.classification
  371. use_device_wallpaper.channel = system_device_wallpaper.channel
  372. use_device_wallpaper.save()
  373. if CONFIG_CN == CONFIG_INFO or CONFIG_TEST == CONFIG_INFO:
  374. s3 = AmazonS3Util(
  375. AWS_ACCESS_KEY_ID[0], AWS_SECRET_ACCESS_KEY[0], 'cn-northwest-1'
  376. )
  377. else:
  378. s3 = AmazonS3Util(
  379. AWS_ACCESS_KEY_ID[1], AWS_SECRET_ACCESS_KEY[1], 'us-east-1'
  380. )
  381. s3.delete_obj("ansjerfilemanager", f"app/static/device-wallpaper/{uid}/{obj_name}")
  382. device_wallpaper_qs.delete()
  383. return response.json(0)
  384. except Exception as e:
  385. LOGGER.info('删除壁纸异常:{}'.format(e))
  386. return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  387. @classmethod
  388. def select_wallpaper(cls, request_dict, response):
  389. """
  390. 用户选取壁纸
  391. @param request_dict:
  392. @param response:
  393. @return:
  394. """
  395. try:
  396. wallpaper_id = request_dict.get('wallpaperId', None)
  397. uid = request_dict.get('uid', None)
  398. if not all([wallpaper_id, uid]):
  399. return response.json(444)
  400. device_wallpaper = DeviceWallpaper.objects.filter(id=wallpaper_id).values("device_type", "storage_type",
  401. "obj_prefix",
  402. "obj_name",
  403. "classification",
  404. "channel").first()
  405. device_wallpaper_qs = DeviceWallpaper.objects.filter(uid=uid, parent_id__gt=0)
  406. if device_wallpaper_qs.exists():
  407. device_wallpaper_qs.update(
  408. storage_type=device_wallpaper["storage_type"],
  409. classification=device_wallpaper["classification"],
  410. obj_prefix=device_wallpaper["obj_prefix"],
  411. obj_name=device_wallpaper["obj_name"],
  412. parent_id=wallpaper_id
  413. )
  414. else:
  415. DeviceWallpaper.objects.create(
  416. device_type=device_wallpaper["device_type"],
  417. storage_type=device_wallpaper["storage_type"],
  418. obj_name=device_wallpaper["obj_name"],
  419. obj_prefix=device_wallpaper["obj_prefix"],
  420. classification=device_wallpaper["classification"],
  421. uid=uid,
  422. channel=device_wallpaper["channel"],
  423. status=1,
  424. parent_id=wallpaper_id,
  425. create_time=int(time.time())
  426. )
  427. return response.json(0)
  428. except Exception as e:
  429. LOGGER.error('用户选取壁纸异常:errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  430. return response.json(5)