SmartSceneController.py 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395
  1. # -*- coding: utf-8 -*-
  2. """
  3. @Author : Rocky
  4. @Time : 2022/6/29 9:31
  5. @File :SmartSceneController.py
  6. """
  7. import time
  8. from django.core.exceptions import ObjectDoesNotExist
  9. from django.db.models import F, Q
  10. from django.views import View
  11. from Model.models import FamilyRoomDevice, GatewaySubDevice, FamilyRoom, SmartScene, EffectiveTime, Device_Info
  12. from Service.CommonService import CommonService
  13. class SmartSceneView(View):
  14. def get(self, request, *args, **kwargs):
  15. request.encoding = 'utf-8'
  16. operation = kwargs.get('operation')
  17. return self.validation(request.GET, request, operation)
  18. def post(self, request, *args, **kwargs):
  19. request.encoding = 'utf-8'
  20. operation = kwargs.get('operation')
  21. return self.validation(request.POST, request, operation)
  22. def validation(self, request_dict, request, operation):
  23. token_code, user_id, response = CommonService.verify_token_get_user_id(request_dict, request)
  24. if token_code != 0:
  25. return response.json(token_code)
  26. if operation == 'condition-devices': # 添加条件-查询设备
  27. return self.condition_devices(request_dict, response)
  28. elif operation == 'task-devices': # 添加任务-查询设备
  29. return self.task_devices(request_dict, response)
  30. elif operation == 'create': # 创建智能场景
  31. return self.create_smart_scene(request_dict, user_id, response)
  32. elif operation == 'scene-list': # 查询智能场景列表
  33. return self.scene_list(request_dict, user_id, response)
  34. elif operation == 'update-status': # 更新智能场景状态
  35. return self.update_status(request_dict, response)
  36. elif operation == 'detail': # 查询智能场景详情
  37. return self.scene_detail(request_dict, response)
  38. elif operation == 'edit': # 编辑智能场景
  39. return self.edit_smart_scene(request_dict, response)
  40. elif operation == 'delete': # 删除智能场景
  41. return self.delete_smart_scene(request_dict, response)
  42. else:
  43. return response.json(414)
  44. @classmethod
  45. def condition_devices(cls, request_dict, response):
  46. """
  47. 添加条件-查询设备
  48. @param request_dict: 请求参数
  49. @request_dict deviceId: 网关设备id
  50. @request_dict subDeviceId: 子设备id
  51. @param response: 响应对象
  52. @return: response
  53. """
  54. device_id = request_dict.get('deviceId', None)
  55. sub_device_id = request_dict.get('subDeviceId', None)
  56. if not any([device_id, sub_device_id]):
  57. return response.json(444, {'error param': 'deviceId or subDeviceId'})
  58. try:
  59. if sub_device_id:
  60. device_id = GatewaySubDevice.objects.get(id=sub_device_id).device_id
  61. gateway_sub_device_qs = GatewaySubDevice.objects.filter(device_id=device_id)
  62. if not gateway_sub_device_qs.exists():
  63. return response.json(173)
  64. res = cls.get_sub_device_room_name(gateway_sub_device_qs)
  65. return response.json(0, res)
  66. except Exception as e:
  67. return response.json(500, repr(e))
  68. @classmethod
  69. def task_devices(cls, request_dict, response):
  70. """
  71. 添加任务-查询设备
  72. @param request_dict: 请求参数
  73. @request_dict deviceId: 网关设备id
  74. @param response: 响应对象
  75. @return: response
  76. """
  77. sub_device_id = request_dict.get('subDeviceId', None)
  78. device_id = request_dict.get('deviceId', None)
  79. if not any([device_id, sub_device_id]):
  80. return response.json(444, {'error param': 'deviceId or subDeviceId'})
  81. try:
  82. if device_id:
  83. res = cls.get_gateway_data(device_id)
  84. else:
  85. sub_device_qs = GatewaySubDevice.objects.filter(id=sub_device_id).values('device_id', 'device_type')
  86. device_id = sub_device_qs[0]['device_id']
  87. device_type = sub_device_qs[0]['device_type']
  88. if device_type != 216:
  89. res = cls.get_gateway_data(device_id)
  90. else:
  91. gateway_data = cls.get_gateway_data(device_id)
  92. sub_device_qs = GatewaySubDevice.objects.filter(
  93. Q(device_id=device_id) & Q(device_type=215) | Q(device_type=219)).values('id', 'nickname',
  94. 'status',
  95. 'device_type')
  96. if not sub_device_qs.exists():
  97. return response.json(173)
  98. res = cls.get_sub_device_room_name(sub_device_qs, gateway_data)
  99. return response.json(0, res)
  100. except Exception as e:
  101. return response.json(500, repr(e))
  102. @staticmethod
  103. def get_sub_device_room_name(sub_device_qs, gateway_data=None):
  104. """
  105. 获取房间名称
  106. @param sub_device_qs: 子设备信息
  107. @param gateway_data: 网关参数
  108. @return: sub_device_list
  109. """
  110. sub_device_list = []
  111. if gateway_data:
  112. sub_device_list.append(gateway_data)
  113. sub_device_qs = sub_device_qs.annotate(gatewaySubId=F('id'),
  114. deviceType=F('device_type'),
  115. deviceNickName=F('nickname')). \
  116. values('gatewaySubId', 'deviceType', 'deviceNickName', 'status')
  117. for sub_device in sub_device_qs:
  118. family_room_device_qs = FamilyRoomDevice.objects.filter(sub_device=sub_device['gatewaySubId']). \
  119. values('room_id')
  120. if not family_room_device_qs.exists():
  121. sub_device['roomName'] = ''
  122. else:
  123. room_id = family_room_device_qs[0]['room_id']
  124. try:
  125. sub_device['roomName'] = FamilyRoom.objects.get(id=room_id).name
  126. except ObjectDoesNotExist:
  127. sub_device['roomName'] = ''
  128. sub_device_list.append(sub_device)
  129. return sub_device_list
  130. @staticmethod
  131. def get_gateway_data(device_id):
  132. """
  133. 获取网关数据
  134. @param device_id: 网关设备id
  135. @return: res
  136. """
  137. device_info_qs = Device_Info.objects.filter(id=device_id).values('NickName', 'Type')
  138. nickname = device_info_qs[0]['NickName']
  139. device_type = device_info_qs[0]['Type']
  140. room_id = FamilyRoomDevice.objects.filter(device_id=device_id).values('room_id')[0]['room_id']
  141. room_id_qs = FamilyRoom.objects.filter(id=room_id).values('name')
  142. room_name = room_id_qs.first()['name'] if room_id_qs.exists() else ''
  143. res = {
  144. 'deviceNickName': nickname,
  145. 'deviceType': device_type,
  146. 'roomName': room_name,
  147. 'status': 1,
  148. }
  149. return res
  150. @staticmethod
  151. def create_smart_scene(request_dict, user_id, response):
  152. """
  153. 创建智能场景
  154. @param request_dict: 请求参数
  155. @param user_id: 用户id
  156. @request_dict deviceId: 网关设备id
  157. @request_dict subDeviceId: 子设备id
  158. @request_dict sceneName: 场景名称
  159. @request_dict conditions: 条件
  160. @request_dict tasks: 任务
  161. @request_dict isAllDay: 是否全天执行
  162. @request_dict startTime: 开始时间
  163. @request_dict endTime: 结束时间
  164. @request_dict repeat: 重复周期
  165. @param response: 响应对象
  166. @return: response
  167. """
  168. device_id = request_dict.get('deviceId', None)
  169. sub_device_id = request_dict.get('subDeviceId', None)
  170. scene_name = request_dict.get('sceneName', None)
  171. conditions = request_dict.get('conditions', None)
  172. tasks = request_dict.get('tasks', None)
  173. is_all_day = request_dict.get('isAllDay', None)
  174. if not any([device_id, sub_device_id]):
  175. return response.json(444, {'error param': 'deviceId or subDeviceId'})
  176. if not all([conditions, tasks]):
  177. return response.json(444, {'error param': 'conditions and tasks'})
  178. now_time = int(time.time())
  179. try:
  180. # 判断是否已存在该场景名
  181. smart_scene_qs = SmartScene.objects.filter(user_id=user_id, scene_name=scene_name)
  182. if smart_scene_qs.exists():
  183. return response.json(174)
  184. smart_scene_dict = {
  185. 'user_id': user_id,
  186. 'scene_name': scene_name,
  187. 'conditions': conditions,
  188. 'tasks': tasks,
  189. 'created_time': now_time,
  190. 'updated_time': now_time,
  191. }
  192. # 处理传网关设备id和子设备id的情况
  193. if device_id:
  194. smart_scene_dict['device_id'] = device_id
  195. device_info_qs = Device_Info.objects.filter(id=device_id).values('serial_number')
  196. if not device_info_qs.exists():
  197. return response.json(173)
  198. serial_number = device_info_qs[0]['serial_number']
  199. else:
  200. smart_scene_dict['sub_device_id'] = sub_device_id
  201. sub_device_qs = GatewaySubDevice.objects.filter(id=sub_device_id).values('device__serial_number')
  202. if not sub_device_qs.exists():
  203. return response.json(173)
  204. serial_number = sub_device_qs[0]['device__serial_number']
  205. if not is_all_day: # 没传时间
  206. SmartScene.objects.create(**smart_scene_dict)
  207. else:
  208. if is_all_day == '0':
  209. effective_time_qs = EffectiveTime.objects.filter(is_all_day=True).values('id')
  210. if effective_time_qs.exists():
  211. effective_time_id = effective_time_qs[0]['id']
  212. else:
  213. effective_time_id = EffectiveTime.objects.create(is_all_day=True).id
  214. smart_scene_dict['effective_time_id'] = effective_time_id
  215. SmartScene.objects.create(**smart_scene_dict)
  216. else:
  217. start_time = int(request_dict.get('startTime', None))
  218. end_time = int(request_dict.get('endTime', None))
  219. repeat = int(request_dict.get('repeat', None))
  220. if not all([start_time, end_time, repeat]):
  221. return response.json(444, {'error param': 'startTime and endTime and repeat'})
  222. effective_time_qs = EffectiveTime.objects.filter(start_time=start_time, end_time=end_time,
  223. repeat=repeat).values('id')
  224. if effective_time_qs.exists():
  225. effective_time_id = effective_time_qs[0]['id']
  226. else:
  227. effective_time_id = EffectiveTime.objects.create(start_time=start_time, end_time=end_time,
  228. repeat=repeat).id
  229. smart_scene_dict['effective_time_id'] = effective_time_id
  230. SmartScene.objects.create(**smart_scene_dict)
  231. # 发布MQTT消息通知网关设备
  232. thing_name = serial_number
  233. topic_name = 'loocam/gateway_sensor/{}/smart_scene'.format(serial_number)
  234. msg = ''
  235. success = CommonService.req_publish_mqtt_msg(thing_name, topic_name, msg)
  236. if not success:
  237. return response.json(10044)
  238. return response.json(0)
  239. except Exception as e:
  240. return response.json(500, repr(e))
  241. @staticmethod
  242. def scene_list(request_dict, user_id, response):
  243. """
  244. 查询智能场景列表
  245. @param request_dict: 请求参数
  246. @param user_id: 用户id
  247. @request_dict deviceId: 网关设备id
  248. @request_dict subDeviceId: 子设备id
  249. @param response: 响应对象
  250. @return: response
  251. """
  252. device_id = request_dict.get('deviceId', None)
  253. sub_device_id = request_dict.get('subDeviceId', None)
  254. if not any([device_id, sub_device_id]):
  255. return response.json(444, {'error param': 'deviceId or subDeviceId'})
  256. try:
  257. if device_id:
  258. smart_scene_qs = SmartScene.objects.filter(user_id=user_id, device_id=device_id)
  259. else:
  260. smart_scene_qs = SmartScene.objects.filter(user_id=user_id, sub_device_id=sub_device_id)
  261. if not smart_scene_qs.exists():
  262. return response.json(173)
  263. smart_scene_qs = smart_scene_qs.values('id', 'scene_name', 'is_enable')
  264. return response.json(0, list(smart_scene_qs))
  265. except Exception as e:
  266. return response.json(500, repr(e))
  267. @staticmethod
  268. def update_status(request_dict, response):
  269. """
  270. 更新智能场景状态
  271. @param request_dict: 请求参数
  272. @request_dict smartSceneId: 智能场景id
  273. @request_dict isEnable: 状态,True or False
  274. @param response: 响应对象
  275. @return: response
  276. """
  277. smart_scene_id = request_dict.get('smartSceneId', None)
  278. is_enable = request_dict.get('isEnable', None)
  279. if not all([smart_scene_id, is_enable]):
  280. return response.json(444, {'error param': 'smartSceneId and status'})
  281. try:
  282. SmartScene.objects.filter(id=smart_scene_id).update(is_enable=is_enable)
  283. return response.json(0)
  284. except Exception as e:
  285. return response.json(500, repr(e))
  286. @staticmethod
  287. def scene_detail(request_dict, response):
  288. """
  289. 查询智能场景详情
  290. @param request_dict: 请求参数
  291. @request_dict smartSceneId: 智能场景id
  292. @param response: 响应对象
  293. @return: response
  294. """
  295. smart_scene_id = request_dict.get('smartSceneId', None)
  296. if not smart_scene_id:
  297. return response.json(444, {'error param': 'smartSceneId'})
  298. try:
  299. smart_scene_qs = SmartScene.objects.filter(id=smart_scene_id).values('id', 'scene_name', 'conditions',
  300. 'tasks', 'effective_time_id')
  301. if not smart_scene_qs.exists():
  302. return response.json(173)
  303. res = {
  304. 'scene_name': smart_scene_qs[0]['scene_name'],
  305. 'condition': eval(smart_scene_qs[0]['conditions']),
  306. 'task': eval(smart_scene_qs[0]['tasks']),
  307. }
  308. # 如果存在关联的时间数据,组织时间数据
  309. try:
  310. effective_time_qs = EffectiveTime.objects.get(id=smart_scene_qs[0]['effective_time_id'])
  311. except ObjectDoesNotExist:
  312. return response.json(0, res)
  313. if effective_time_qs.is_all_day: # 全天
  314. time_dict = {'all_day': 1}
  315. else:
  316. time_dict = {
  317. 'start_time': effective_time_qs.start_time,
  318. 'end_time': effective_time_qs.end_time,
  319. 'repeat': effective_time_qs.repeat,
  320. }
  321. res['time'] = time_dict
  322. return response.json(0, res)
  323. except Exception as e:
  324. return response.json(500, repr(e))
  325. @staticmethod
  326. def edit_smart_scene(request_dict, response):
  327. """
  328. 编辑智能场景
  329. @param request_dict: 请求参数
  330. @request_dict smartSceneId: 智能场景id
  331. @param response: 响应对象
  332. @return: response
  333. """
  334. smart_scene_id = request_dict.get('smartSceneId', None)
  335. if not smart_scene_id:
  336. return response.json(444, {'error param': 'smartSceneId'})
  337. try:
  338. smart_scene_qs = SmartScene.objects.filter(id=smart_scene_id).values('id', 'scene_name', 'conditions',
  339. 'tasks', 'effective_time_id')
  340. if not smart_scene_qs.exists():
  341. return response.json(173)
  342. return response.json(0, list(smart_scene_qs))
  343. except Exception as e:
  344. return response.json(500, repr(e))
  345. @staticmethod
  346. def delete_smart_scene(request_dict, response):
  347. """
  348. 删除智能场景
  349. @param request_dict: 请求参数
  350. @request_dict smartSceneId: 智能场景id
  351. @param response: 响应对象
  352. @return: response
  353. """
  354. smart_scene_id = request_dict.get('smartSceneId', None)
  355. if not smart_scene_id:
  356. return response.json(444, {'error param': 'smartSceneId'})
  357. try:
  358. SmartScene.objects.filter(id=smart_scene_id).delete()
  359. return response.json(0)
  360. except Exception as e:
  361. return response.json(500, repr(e))