SmartSceneController.py 69 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523
  1. # -*- coding: utf-8 -*-
  2. """
  3. @Author : Rocky
  4. @Time : 2022/6/29 9:31
  5. @File :SmartSceneController.py
  6. """
  7. import json
  8. import time
  9. from django.core.exceptions import ObjectDoesNotExist
  10. from django.db import transaction
  11. from django.db.models import F, Q, Count
  12. from django.views import View
  13. from Ansjer.Config.gatewaySensorConfig import SMART_SCENE_TOPIC, SENSOR_TYPE, EVENT_TYPE, SCENE_EVENT_CREATE, \
  14. SCENE_EVENT_EDIT, SCENE_EVENT_DELETE, SCENE_STATUS_ON, SCENE_STATUS_OFF, SCENE_EVENT_EDIT_STATUS, \
  15. VOICE_AUDITION_TOPIC, DEVICE_TYPE, SMART_SCENE_TASK, LINKAGE_DEVICE_TYPE_LIST
  16. from Model.models import FamilyRoomDevice, GatewaySubDevice, FamilyRoom, SmartScene, EffectiveTime, Device_Info, \
  17. SceneLog
  18. from Object.CeleryBeatObject import CeleryBeatObj
  19. from Object.ResponseObject import ResponseObject
  20. from Service.CommonService import CommonService
  21. class SmartSceneView(View):
  22. def get(self, request, *args, **kwargs):
  23. request.encoding = 'utf-8'
  24. operation = kwargs.get('operation')
  25. return self.validation(request.GET, request, operation)
  26. def post(self, request, *args, **kwargs):
  27. request.encoding = 'utf-8'
  28. operation = kwargs.get('operation')
  29. return self.validation(request.POST, request, operation)
  30. def validation(self, request_dict, request, operation):
  31. if operation == 'get-scene': # 设备获取智能场景数据
  32. return self.get_scene_data(request_dict, ResponseObject('cn'))
  33. token_code, user_id, response = CommonService.verify_token_get_user_id(request_dict, request)
  34. if token_code != 0:
  35. return response.json(token_code)
  36. if operation == 'condition-devices': # 添加条件-查询设备
  37. return self.condition_devices(request_dict, response)
  38. elif operation == 'task-devices': # 添加任务-查询设备
  39. return self.task_devices(request_dict, user_id, response)
  40. elif operation == 'create': # 创建智能场景
  41. return self.create_smart_scene(request_dict, user_id, response)
  42. elif operation == 'scene-list': # 查询智能场景列表
  43. return self.scene_list(request_dict, user_id, response)
  44. elif operation == 'smart-button-scene-list': # 查询智能按钮场景列表
  45. return self.smart_button_scene_list(request_dict, user_id, response)
  46. elif operation == 'update-status': # 更新智能场景状态
  47. return self.update_status(request_dict, response)
  48. elif operation == 'detail': # 查询智能场景详情
  49. return self.scene_detail(request_dict, response)
  50. elif operation == 'edit': # 编辑智能场景
  51. return self.edit_smart_scene(request_dict, user_id, response)
  52. elif operation == 'delete': # 删除智能场景
  53. return self.delete_smart_scene(request_dict, response)
  54. elif operation == 'log': # 查询智能场景日志
  55. return self.scene_log(request_dict, response)
  56. elif operation == 'log-date': # 查询智能场景日志日期
  57. return self.scene_log_date(request_dict, response)
  58. elif operation == 'voice-audition': # 智能场景音频试听
  59. return self.voice_audition(request_dict, response)
  60. else:
  61. return response.json(414)
  62. @classmethod
  63. def condition_devices(cls, request_dict, response):
  64. """
  65. 添加条件-查询设备
  66. @param request_dict: 请求参数
  67. @request_dict deviceId: 网关设备id
  68. @request_dict subDeviceId: 子设备id
  69. @param response: 响应对象
  70. @return: response
  71. """
  72. device_id = request_dict.get('deviceId', None)
  73. sub_device_id = request_dict.get('subDeviceId', None)
  74. if not any([device_id, sub_device_id]):
  75. return response.json(444, {'error param': 'deviceId or subDeviceId'})
  76. try:
  77. if sub_device_id:
  78. device_id = GatewaySubDevice.objects.get(id=sub_device_id).device_id
  79. gateway_sub_device_qs = GatewaySubDevice.objects.filter(device_id=device_id)
  80. if not gateway_sub_device_qs.exists():
  81. return response.json(173)
  82. res = cls.get_sub_device_room_name(gateway_sub_device_qs)
  83. return response.json(0, res)
  84. except Exception as e:
  85. return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  86. @classmethod
  87. def task_devices(cls, request_dict, user_id, response):
  88. """
  89. 添加任务-查询设备
  90. @param request_dict: 请求参数
  91. @request_dict deviceId: 网关设备id
  92. @param user_id: 用户id
  93. @param response: 响应对象
  94. @return: response
  95. """
  96. sub_device_id = request_dict.get('subDeviceId', None)
  97. device_id = request_dict.get('deviceId', None)
  98. if not any([device_id, sub_device_id]):
  99. return response.json(444, {'error param': 'deviceId or subDeviceId'})
  100. try:
  101. if device_id:
  102. res = [cls.get_gateway_data(device_id)]
  103. else:
  104. sub_device_qs = GatewaySubDevice.objects.filter(id=sub_device_id).values('device_id', 'device_type')
  105. device_id = sub_device_qs[0]['device_id']
  106. device_type = sub_device_qs[0]['device_type']
  107. if device_type != SENSOR_TYPE['smart_button']: # 非智能按钮只返回网关
  108. res = [cls.get_gateway_data(device_id)]
  109. else: # 智能按钮返回网关,门磁和人体传感器(如果存在)
  110. gateway_data = cls.get_gateway_data(device_id)
  111. sub_device_qs = GatewaySubDevice.objects.filter(
  112. Q(Q(device_id=device_id) & Q(device_type=SENSOR_TYPE['door_magnet'])) |
  113. Q(Q(device_id=device_id) & Q(device_type=SENSOR_TYPE['body_sensor']))
  114. ).values('id', 'nickname', 'status', 'device_type')
  115. if sub_device_qs.exists():
  116. res = cls.get_sub_device_room_name(sub_device_qs, gateway_data)
  117. else:
  118. res = [gateway_data]
  119. # 添加联动设备数据
  120. res = cls.append_linkage_device_data(res, user_id)
  121. return response.json(0, res)
  122. except Exception as e:
  123. return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  124. @staticmethod
  125. def get_sub_device_room_name(sub_device_qs, gateway_data=None):
  126. """
  127. 获取房间名称
  128. @param sub_device_qs: 子设备信息
  129. @param gateway_data: 网关参数
  130. @return: sub_device_list
  131. """
  132. sub_device_list = []
  133. if gateway_data:
  134. sub_device_list.append(gateway_data)
  135. sub_device_qs = sub_device_qs.annotate(gatewaySubId=F('id'),
  136. deviceType=F('device_type'),
  137. deviceNickName=F('nickname')). \
  138. values('gatewaySubId', 'deviceType', 'deviceNickName', 'status')
  139. for sub_device in sub_device_qs:
  140. family_room_device_qs = FamilyRoomDevice.objects.filter(sub_device=sub_device['gatewaySubId']). \
  141. values('room_id')
  142. if not family_room_device_qs.exists():
  143. sub_device['roomName'] = ''
  144. else:
  145. room_id = family_room_device_qs[0]['room_id']
  146. try:
  147. sub_device['roomName'] = FamilyRoom.objects.get(id=room_id).name
  148. except ObjectDoesNotExist:
  149. sub_device['roomName'] = ''
  150. sub_device_list.append(sub_device)
  151. return sub_device_list
  152. @staticmethod
  153. def get_gateway_data(device_id):
  154. """
  155. 获取网关数据
  156. @param device_id: 网关设备id
  157. @return: res
  158. """
  159. device_info_qs = Device_Info.objects.filter(id=device_id).values('NickName', 'Type', 'serial_number')
  160. nickname = device_info_qs[0]['NickName']
  161. device_type = device_info_qs[0]['Type']
  162. serial_number = device_info_qs[0]['serial_number']
  163. room_id = FamilyRoomDevice.objects.filter(device_id=device_id).values('room_id')[0]['room_id']
  164. room_id_qs = FamilyRoom.objects.filter(id=room_id).values('name')
  165. room_name = room_id_qs.first()['name'] if room_id_qs.exists() else ''
  166. res = {
  167. 'serialNumber': serial_number,
  168. 'deviceNickName': nickname,
  169. 'deviceType': device_type,
  170. 'roomName': room_name,
  171. 'status': 1,
  172. }
  173. return res
  174. @staticmethod
  175. def append_linkage_device_data(res, user_id):
  176. """
  177. 添加联动设备数据
  178. @param res: 网关等设备数据
  179. @param user_id: 用户id
  180. @return: res
  181. """
  182. device_info_qs = Device_Info.objects.filter(userID_id=user_id, Type__in=LINKAGE_DEVICE_TYPE_LIST).\
  183. values('id', 'NickName', 'Type', 'serial_number')
  184. if device_info_qs.exists():
  185. for device_info in device_info_qs:
  186. nickname = device_info['NickName']
  187. device_type = device_info['Type']
  188. serial_number = device_info['serial_number']
  189. device_id = device_info['id']
  190. room_id = FamilyRoomDevice.objects.filter(device_id=device_id).values('room_id')[0]['room_id']
  191. room_id_qs = FamilyRoom.objects.filter(id=room_id).values('name')
  192. room_name = room_id_qs.first()['name'] if room_id_qs.exists() else ''
  193. data = {
  194. 'serialNumber': serial_number,
  195. 'deviceNickName': nickname,
  196. 'deviceType': device_type,
  197. 'roomName': room_name,
  198. 'status': 1,
  199. }
  200. res.append(data)
  201. return res
  202. @classmethod
  203. def create_smart_scene(cls, request_dict, user_id, response):
  204. """
  205. 创建智能场景
  206. @param request_dict: 请求参数
  207. @param user_id: 用户id
  208. @request_dict deviceId: 网关设备id
  209. @request_dict subDeviceId: 子设备id
  210. @request_dict sceneName: 场景名称
  211. @request_dict conditions: 条件
  212. @request_dict tasks: 任务
  213. @request_dict isAllDay: 是否全天执行
  214. @request_dict startTime: 开始时间
  215. @request_dict endTime: 结束时间
  216. @request_dict repeat: 重复周期
  217. @param response: 响应对象
  218. @return: response
  219. """
  220. device_id = request_dict.get('deviceId', None)
  221. sub_device_id = request_dict.get('subDeviceId', None)
  222. scene_name = request_dict.get('sceneName', None)
  223. conditions = request_dict.get('conditions', None)
  224. tasks = request_dict.get('tasks', None)
  225. is_all_day = request_dict.get('isAllDay', None)
  226. if not all([scene_name, conditions, tasks]):
  227. return response.json(444, {'error param': 'scene_name and conditions and tasks'})
  228. now_time = int(time.time())
  229. conditions_dict = eval(conditions)
  230. tasks_list = eval(tasks)
  231. try:
  232. # 判断是否已存在该场景名
  233. smart_scene_qs = SmartScene.objects.filter(user_id=user_id, scene_name=scene_name)
  234. if smart_scene_qs.exists():
  235. return response.json(179)
  236. tz = CommonService.get_user_tz(user_id)
  237. smart_scene_dict = {
  238. 'user_id': user_id,
  239. 'scene_name': scene_name,
  240. 'conditions': conditions,
  241. 'tasks': tasks,
  242. 'tz': tz,
  243. 'created_time': now_time,
  244. 'updated_time': now_time,
  245. }
  246. msg = {
  247. 'scene_event': SCENE_EVENT_CREATE,
  248. 'scene_status': SCENE_STATUS_ON
  249. }
  250. # 处理设置时间
  251. if is_all_day is not None:
  252. is_all_day = int(is_all_day)
  253. smart_scene_dict['is_all_day'] = is_all_day
  254. # 判断条件是否为设置时间
  255. is_set_time, minutes, repeat = False, 0, 0
  256. if conditions_dict['type'] == 1:
  257. is_set_time = True
  258. minutes = conditions_dict['time']['minutes']
  259. repeat = conditions_dict['time']['repeat']
  260. # 条件为设置时间
  261. if is_set_time:
  262. if not device_id:
  263. return response.json(444, {'error param': 'deviceId'})
  264. smart_scene_dict['device_id'] = device_id
  265. device_info_qs = Device_Info.objects.filter(id=device_id).values('serial_number')
  266. if not device_info_qs.exists():
  267. return response.json(173)
  268. serial_number = device_info_qs[0]['serial_number']
  269. # 网关数据
  270. msg['sensor_type'] = DEVICE_TYPE['gateway']
  271. msg['sensor_status'] = 2002
  272. msg['sensor_ieee_addr'] = 'FFFFFFFFFFFFFFFF'
  273. # 条件为传感器设备
  274. else:
  275. if not sub_device_id:
  276. return response.json(444, {'error param': 'subDeviceId'})
  277. # 查询数据
  278. sub_device_qs = GatewaySubDevice.objects.filter(id=sub_device_id).\
  279. values('device__serial_number', 'ieee_addr', 'is_tampered')
  280. if not sub_device_qs.exists():
  281. return response.json(173)
  282. if cls.time_conflict(sub_device_id, conditions, is_all_day, request_dict):
  283. return response.json(182)
  284. device_type = int(conditions_dict['sensor']['device_type'])
  285. # 智能按钮不能创建触发条件相同的场景
  286. if device_type == SENSOR_TYPE['smart_button']:
  287. event_type = conditions_dict['sensor']['eventValues'][0]['event_type']
  288. smart_scene_qs = SmartScene.objects.filter(sub_device_id=sub_device_id,
  289. conditions__contains=event_type)
  290. if smart_scene_qs.exists():
  291. return response.json(180)
  292. # 紧急按钮打开时,创建的场景状态默认为关闭
  293. if sub_device_qs[0]['is_tampered'] == 1:
  294. smart_scene_dict['is_enable'] = False
  295. # 温湿度传感器返回温湿度
  296. elif device_type == SENSOR_TYPE['tem_hum_sensor']:
  297. event_values = conditions_dict['sensor']['eventValues'][0]
  298. if '≥' in event_values['value']:
  299. replace_str = '≥ '
  300. msg['sensor_symbol'] = 2
  301. else:
  302. replace_str = '≤ '
  303. msg['sensor_symbol'] = 1
  304. value = event_values['value'].replace(replace_str, '')
  305. msg['sensor_data'] = float(value)
  306. smart_scene_dict['sub_device_id'] = sub_device_id
  307. serial_number = sub_device_qs[0]['device__serial_number']
  308. msg['sensor_type'] = int(conditions_dict['sensor']['device_type'])
  309. msg['sensor_ieee_addr'] = sub_device_qs[0]['ieee_addr']
  310. msg['sensor_status'] = int(conditions_dict['sensor']['eventValues'][0]['event_type'])
  311. with transaction.atomic():
  312. if is_all_day is None: # 不设置时间
  313. smart_scene_qs = SmartScene.objects.create(**smart_scene_dict)
  314. # 设备的time数据,分钟转为秒
  315. time_dict = {
  316. 'start_time': conditions_dict['time']['minutes'] * 60,
  317. 'repeat': conditions_dict['time']['repeat']
  318. }
  319. elif is_all_day == 1: # 全天
  320. smart_scene_qs = SmartScene.objects.create(**smart_scene_dict)
  321. # 设备的time数据
  322. time_dict = {
  323. 'is_all_day': is_all_day
  324. }
  325. elif is_all_day == 2: # 非全天
  326. start_time = int(request_dict.get('startTime', None))
  327. end_time = int(request_dict.get('endTime', None))
  328. repeat = int(request_dict.get('repeat', None))
  329. effective_time_qs = EffectiveTime.objects.filter(start_time=start_time, end_time=end_time,
  330. repeat=repeat).values('id')
  331. if effective_time_qs.exists():
  332. effective_time_id = effective_time_qs[0]['id']
  333. else:
  334. effective_time_id = EffectiveTime.objects.create(start_time=start_time, end_time=end_time,
  335. repeat=repeat).id
  336. smart_scene_dict['effective_time_id'] = effective_time_id
  337. smart_scene_qs = SmartScene.objects.create(**smart_scene_dict)
  338. # 设备的time数据,分钟转为秒
  339. time_dict = {
  340. 'is_all_day': is_all_day,
  341. 'start_time': start_time * 60,
  342. 'end_time': end_time * 60,
  343. 'repeat': repeat
  344. }
  345. else:
  346. return response.json(444, {'error param': 'invalid isAllDay'})
  347. msg['time'] = time_dict
  348. scene_id = smart_scene_qs.id
  349. msg['scene_id'] = scene_id
  350. # 获取设备任务数据
  351. msg['task'], scene_data = cls.get_task_list_and_scene_data(
  352. conditions_dict, is_set_time, minutes, repeat, tz, now_time, tasks_list, scene_id)
  353. smart_scene_qs.device_data = json.dumps(msg)
  354. smart_scene_qs.scene_data = scene_data
  355. smart_scene_qs.save()
  356. # 设备任务列表不为空,发布MQTT消息通知网关设备
  357. if msg['task']:
  358. thing_name = serial_number
  359. topic_name = SMART_SCENE_TOPIC.format(serial_number)
  360. success = CommonService.req_publish_mqtt_msg(thing_name, topic_name, msg)
  361. try:
  362. assert success
  363. except AssertionError:
  364. return response.json(10044)
  365. return response.json(0)
  366. except Exception as e:
  367. return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  368. @staticmethod
  369. def scene_list(request_dict, user_id, response):
  370. """
  371. 查询智能场景列表
  372. @param request_dict: 请求参数
  373. @param user_id: 用户id
  374. @request_dict deviceId: 网关设备id
  375. @request_dict subDeviceId: 子设备id
  376. @param response: 响应对象
  377. @return: response
  378. """
  379. device_id = request_dict.get('deviceId', None)
  380. sub_device_id = request_dict.get('subDeviceId', None)
  381. if not any([device_id, sub_device_id]):
  382. return response.json(444, {'error param': 'deviceId or subDeviceId'})
  383. try:
  384. if device_id:
  385. sub_device_id = GatewaySubDevice.objects.filter(device_id=device_id).values('id')
  386. smart_scene_qs = SmartScene.objects.filter(
  387. Q(user_id=user_id) & Q(device_id=device_id) | Q(sub_device_id__in=sub_device_id))
  388. else:
  389. smart_scene_qs = SmartScene.objects.filter(user_id=user_id, sub_device_id=sub_device_id)
  390. if not smart_scene_qs.exists():
  391. return response.json(173)
  392. smart_scene_qs = smart_scene_qs.values(
  393. 'id', 'scene_name', 'is_enable', 'device_id', 'sub_device_id', 'tasks')
  394. smart_scene_list = []
  395. for item in smart_scene_qs:
  396. smart_scene_dict = {
  397. 'id': item['id'],
  398. 'scene_name': item['scene_name'],
  399. 'is_enable': item['is_enable']
  400. }
  401. if item['device_id']:
  402. data = {
  403. 'userID': user_id
  404. }
  405. # 如果最后一个任务执行的设备是插座,返回插座数据,否则返回网关数据
  406. tasks_list = eval(item['tasks'])
  407. last_task = tasks_list[-1]
  408. if int(last_task['device_type']) == DEVICE_TYPE['socket']:
  409. data['serial_number'] = last_task['serial_number']
  410. else:
  411. data['id'] = item['device_id']
  412. device_qs = Device_Info.objects.filter(**data).values('NickName', 'Type')
  413. smart_scene_dict['device_type'] = device_qs[0]['Type'] if device_qs.exists() else ''
  414. smart_scene_dict['device_nickname'] = device_qs[0]['NickName'] if device_qs.exists() else ''
  415. else:
  416. device_qs = GatewaySubDevice.objects.filter(id=item['sub_device_id']).values('device_type',
  417. 'nickname')
  418. smart_scene_dict['device_type'] = device_qs[0]['device_type'] if device_qs.exists() else ''
  419. smart_scene_dict['device_nickname'] = device_qs[0]['nickname'] if device_qs.exists() else ''
  420. smart_scene_list.append(smart_scene_dict)
  421. return response.json(0, smart_scene_list)
  422. except Exception as e:
  423. return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  424. @staticmethod
  425. def smart_button_scene_list(request_dict, user_id, response):
  426. """
  427. 查询智能按钮场景列表
  428. @param request_dict: 请求参数
  429. @param user_id: 用户id
  430. @request_dict subDeviceId: 子设备id
  431. @param response: 响应对象
  432. @return: response
  433. """
  434. sub_device_id = request_dict.get('subDeviceId', None)
  435. if not sub_device_id:
  436. return response.json(444, {'error param': 'subDeviceId'})
  437. try:
  438. click_scene_qs = SmartScene.objects.filter(user_id=user_id, sub_device_id=sub_device_id,
  439. conditions__contains=str(
  440. EVENT_TYPE['smart_button_click'])).values('id', 'scene_name',
  441. 'is_enable')
  442. double_click_scene_qs = SmartScene.objects.filter(user_id=user_id, sub_device_id=sub_device_id,
  443. conditions__contains=str(
  444. EVENT_TYPE['smart_button_double_click'])). \
  445. values('id', 'scene_name', 'is_enable')
  446. three_click_scene_qs = SmartScene.objects.filter(user_id=user_id, sub_device_id=sub_device_id,
  447. conditions__contains=str(
  448. EVENT_TYPE['smart_button_three_click'])). \
  449. values('id', 'scene_name', 'is_enable')
  450. scene_list = []
  451. if click_scene_qs.exists():
  452. scene_list.append({
  453. 'trigger_type': 1,
  454. 'id': click_scene_qs[0]['id'],
  455. 'scene_name': click_scene_qs[0]['scene_name'],
  456. 'is_enable': click_scene_qs[0]['is_enable']
  457. })
  458. if double_click_scene_qs.exists():
  459. scene_list.append({
  460. 'trigger_type': 2,
  461. 'id': double_click_scene_qs[0]['id'],
  462. 'scene_name': double_click_scene_qs[0]['scene_name'],
  463. 'is_enable': double_click_scene_qs[0]['is_enable']
  464. })
  465. if three_click_scene_qs.exists():
  466. scene_list.append({
  467. 'trigger_type': 3,
  468. 'id': three_click_scene_qs[0]['id'],
  469. 'scene_name': three_click_scene_qs[0]['scene_name'],
  470. 'is_enable': three_click_scene_qs[0]['is_enable']
  471. })
  472. return response.json(0, scene_list)
  473. except Exception as e:
  474. return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  475. @classmethod
  476. def update_status(cls, request_dict, response):
  477. """
  478. 更新智能场景状态
  479. @param request_dict: 请求参数
  480. @request_dict smartSceneId: 智能场景id
  481. @request_dict isEnable: 状态,True or False
  482. @param response: 响应对象
  483. @return: response
  484. """
  485. smart_scene_id = request_dict.get('smartSceneId', None)
  486. is_enable = request_dict.get('isEnable', None)
  487. if not all([smart_scene_id, is_enable]):
  488. return response.json(444, {'error param': 'smartSceneId and status'})
  489. try:
  490. smart_scene_id = int(smart_scene_id)
  491. scene_status = SCENE_STATUS_ON if is_enable == 'True' else SCENE_STATUS_OFF
  492. # 查询序列号
  493. smart_scene_qs = SmartScene.objects.filter(id=smart_scene_id).\
  494. values('device_id', 'sub_device_id', 'scene_data')
  495. device_id = smart_scene_qs[0]['device_id']
  496. if device_id:
  497. serial_number = Device_Info.objects.filter(id=device_id).values('serial_number')[0]['serial_number']
  498. else:
  499. sub_device_id = smart_scene_qs[0]['sub_device_id']
  500. serial_number = GatewaySubDevice.objects.filter(id=sub_device_id).values('device__serial_number')[0][
  501. 'device__serial_number']
  502. smart_scene_data = {
  503. 'is_enable': is_enable
  504. }
  505. no_device_task = False
  506. # 如果存在定时任务,暂停或恢复任务
  507. scene_data = smart_scene_qs[0]['scene_data']
  508. if scene_data:
  509. scene_data_dict = eval(scene_data)
  510. task_list = scene_data_dict.get('task_list')
  511. no_device_task = scene_data_dict.get('no_device_task')
  512. if no_device_task is not None and task_list:
  513. # 无设备任务,需要scene_id上报场景日志
  514. scene_id = smart_scene_id if no_device_task else 0
  515. new_scene_data_dict = cls.pause_or_resume_job(scene_data_dict, scene_status, scene_id)
  516. if new_scene_data_dict:
  517. smart_scene_data['scene_data'] = new_scene_data_dict
  518. with transaction.atomic():
  519. SmartScene.objects.filter(id=smart_scene_id).update(**smart_scene_data)
  520. # 存在设备任务,通过mqtt发送设备数据
  521. if not no_device_task:
  522. topic_name = SMART_SCENE_TOPIC.format(serial_number)
  523. msg = {
  524. 'scene_event': SCENE_EVENT_EDIT_STATUS,
  525. 'scene_id': smart_scene_id,
  526. 'scene_status': scene_status
  527. }
  528. success = CommonService.req_publish_mqtt_msg(serial_number, topic_name, msg)
  529. try:
  530. assert success
  531. except AssertionError:
  532. return response.json(10044)
  533. return response.json(0)
  534. except Exception as e:
  535. return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  536. @staticmethod
  537. def scene_detail(request_dict, response):
  538. """
  539. 查询智能场景详情
  540. @param request_dict: 请求参数
  541. @request_dict smartSceneId: 智能场景id
  542. @param response: 响应对象
  543. @return: response
  544. """
  545. smart_scene_id = request_dict.get('smartSceneId', None)
  546. if not smart_scene_id:
  547. return response.json(444, {'error param': 'smartSceneId'})
  548. try:
  549. smart_scene_qs = SmartScene.objects.filter(id=smart_scene_id).values('id', 'scene_name', 'conditions',
  550. 'tasks', 'effective_time_id',
  551. 'is_all_day')
  552. if not smart_scene_qs.exists():
  553. return response.json(173)
  554. res = {
  555. 'scene_name': smart_scene_qs[0]['scene_name'],
  556. 'condition': eval(smart_scene_qs[0]['conditions']),
  557. 'task': eval(smart_scene_qs[0]['tasks']),
  558. }
  559. # 如果存在关联的时间数据,组织时间数据
  560. is_all_day = smart_scene_qs[0]['is_all_day']
  561. effectiveTime = {}
  562. if is_all_day != 0:
  563. effectiveTime['isAllDay'] = is_all_day
  564. if is_all_day == 2:
  565. try:
  566. effective_time_qs = EffectiveTime.objects.get(id=smart_scene_qs[0]['effective_time_id'])
  567. effectiveTime['startTime'] = effective_time_qs.start_time
  568. effectiveTime['endTime'] = effective_time_qs.end_time
  569. effectiveTime['repeat'] = effective_time_qs.repeat
  570. except ObjectDoesNotExist:
  571. return response.json(0, res)
  572. res['effectiveTime'] = effectiveTime
  573. return response.json(0, res)
  574. except Exception as e:
  575. return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  576. @classmethod
  577. def edit_smart_scene(cls, request_dict, user_id, response):
  578. """
  579. 编辑智能场景
  580. @param request_dict: 请求参数
  581. @param user_id: 用户id
  582. @request_dict smartSceneId: 智能场景id
  583. @param response: 响应对象
  584. @return: response
  585. """
  586. smart_scene_id = int(request_dict.get('smartSceneId', None))
  587. device_id = request_dict.get('deviceId', None)
  588. sub_device_id = request_dict.get('subDeviceId', None)
  589. scene_name = request_dict.get('sceneName', None)
  590. conditions = request_dict.get('conditions', None)
  591. tasks = request_dict.get('tasks', None)
  592. is_all_day = request_dict.get('isAllDay', None)
  593. conditions_dict = eval(conditions)
  594. tasks_list = eval(tasks)
  595. now_time = int(time.time())
  596. smart_scene_qs = SmartScene.objects.filter(user_id=user_id, scene_name=scene_name).filter(~Q(id=smart_scene_id))
  597. if smart_scene_qs.exists():
  598. return response.json(179)
  599. res = {
  600. 'scene_name': scene_name,
  601. 'conditions': conditions_dict,
  602. 'tasks': tasks_list
  603. }
  604. effective_time = {}
  605. if is_all_day:
  606. is_all_day = int(is_all_day)
  607. effective_time['is_all_day'] = is_all_day
  608. if not all([smart_scene_id, scene_name, conditions, tasks]):
  609. return response.json(444, {'error param': 'smartSceneId,sceneName,conditions or tasks'})
  610. try:
  611. smart_scene_qs = SmartScene.objects.filter(id=smart_scene_id)
  612. if not smart_scene_qs.exists():
  613. return response.json(173)
  614. tz = smart_scene_qs[0].tz
  615. scene_data = smart_scene_qs[0].scene_data
  616. scene_data_dict = eval(scene_data) if scene_data else None
  617. scene_status = 1 if smart_scene_qs[0].is_enable else 0
  618. msg = {
  619. 'scene_id': smart_scene_id,
  620. 'scene_event': SCENE_EVENT_EDIT,
  621. 'scene_status': scene_status
  622. }
  623. # 判断条件是否为设置时间
  624. is_set_time, minutes, repeat = False, 0, 0
  625. if conditions_dict['type'] == 1:
  626. is_set_time = True
  627. minutes = conditions_dict['time']['minutes']
  628. repeat = conditions_dict['time']['repeat']
  629. # 条件为设置时间
  630. if is_set_time:
  631. if not device_id:
  632. return response.json(444, {'error param': 'deviceId'})
  633. sub_device_id = 0
  634. device_qs = Device_Info.objects.filter(id=device_id).values('serial_number')
  635. if not device_qs.exists():
  636. return response.json(173)
  637. serial_number = device_qs[0]['serial_number']
  638. # 网关数据
  639. msg['sensor_type'] = DEVICE_TYPE['gateway']
  640. msg['sensor_status'] = 2002
  641. msg['sensor_ieee_addr'] = 'FFFFFFFFFFFFFFFF'
  642. # 条件为选择子设备
  643. else:
  644. if not sub_device_id:
  645. return response.json(444, {'error param': 'subDeviceId'})
  646. if cls.time_conflict(sub_device_id, conditions, is_all_day, request_dict, smart_scene_id):
  647. return response.json(182)
  648. device_type = int(conditions_dict['sensor']['device_type'])
  649. # 智能按钮不能创建触发条件相同的场景
  650. if device_type == SENSOR_TYPE['smart_button']:
  651. event_type = conditions_dict['sensor']['eventValues'][0]['event_type']
  652. smart_scene_temp_qs = SmartScene.objects.filter(Q(sub_device_id=sub_device_id),
  653. ~Q(id=smart_scene_id),
  654. conditions__contains=event_type)
  655. if smart_scene_temp_qs.exists():
  656. return response.json(180)
  657. # 温湿度传感器返回温湿度
  658. elif device_type == SENSOR_TYPE['tem_hum_sensor']:
  659. event_values = conditions_dict['sensor']['eventValues'][0]
  660. if '≥' in event_values['value']:
  661. replace_str = '≥ '
  662. msg['sensor_symbol'] = 2
  663. else:
  664. replace_str = '≤ '
  665. msg['sensor_symbol'] = 1
  666. value = event_values['value'].replace(replace_str, '')
  667. msg['sensor_data'] = float(value)
  668. device_id = ''
  669. sub_device_qs = GatewaySubDevice.objects.filter(id=sub_device_id).values('ieee_addr',
  670. 'device__serial_number')
  671. if not sub_device_qs.exists():
  672. return response.json(173)
  673. serial_number = sub_device_qs[0]['device__serial_number']
  674. msg['sensor_type'] = int(conditions_dict['sensor']['device_type'])
  675. msg['sensor_ieee_addr'] = sub_device_qs[0]['ieee_addr']
  676. msg['sensor_status'] = int(conditions_dict['sensor']['eventValues'][0]['event_type'])
  677. # 获取设备任务数据
  678. msg['task'], scene_data = cls.get_task_list_and_scene_data(
  679. conditions_dict, is_set_time, minutes, repeat, tz, now_time, tasks_list, smart_scene_id, scene_data_dict
  680. )
  681. with transaction.atomic():
  682. smart_scene_qs.update(scene_name=scene_name, conditions=conditions, tasks=tasks, scene_data=scene_data,
  683. device_data=json.dumps(msg), updated_time=now_time, device_id=device_id,
  684. sub_device_id=sub_device_id, is_enable=True)
  685. if is_all_day is None: # 不设置时间或全天
  686. smart_scene_qs.update(effective_time_id=0, is_all_day=0)
  687. time_dict = {
  688. 'start_time': conditions_dict['time']['minutes'] * 60,
  689. 'repeat': conditions_dict['time']['repeat']
  690. }
  691. elif is_all_day == 1:
  692. smart_scene_qs.update(effective_time_id=0, is_all_day=is_all_day)
  693. time_dict = {
  694. 'is_all_day': is_all_day
  695. }
  696. else:
  697. start_time = int(request_dict.get('startTime', None))
  698. end_time = int(request_dict.get('endTime', None))
  699. repeat = int(request_dict.get('repeat', None))
  700. effective_time_qs = EffectiveTime.objects.filter(start_time=start_time, end_time=end_time,
  701. repeat=repeat).values('id')
  702. if effective_time_qs.exists():
  703. effective_time_id = effective_time_qs[0]['id']
  704. else:
  705. effective_time_id = EffectiveTime.objects.create(start_time=start_time, end_time=end_time,
  706. repeat=repeat).id
  707. smart_scene_qs.update(effective_time_id=effective_time_id, is_all_day=is_all_day)
  708. time_dict = {
  709. 'is_all_day': is_all_day,
  710. 'start_time': start_time * 60,
  711. 'end_time': end_time * 60,
  712. 'repeat': repeat
  713. }
  714. effective_time = {
  715. 'isAllDay': is_all_day,
  716. 'startTime': start_time,
  717. 'endTime': end_time,
  718. 'repeat': repeat
  719. }
  720. msg['time'] = time_dict
  721. # 设备任务列表不为空,发布MQTT消息通知网关设备
  722. if msg['task']:
  723. thing_name = serial_number
  724. topic_name = SMART_SCENE_TOPIC.format(serial_number)
  725. success = CommonService.req_publish_mqtt_msg(thing_name, topic_name, msg)
  726. try:
  727. assert success
  728. except AssertionError:
  729. return response.json(10044)
  730. res['effectiveTime'] = effective_time
  731. return response.json(0, res)
  732. except Exception as e:
  733. return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  734. @classmethod
  735. def delete_smart_scene(cls, request_dict, response):
  736. """
  737. 删除智能场景
  738. @param request_dict: 请求参数
  739. @request_dict smartSceneIds: 智能场景id
  740. @param response: 响应对象
  741. @return: response
  742. """
  743. smart_scene_ids = request_dict.get('smartSceneIds', None)
  744. if not smart_scene_ids:
  745. return response.json(444, {'error param': 'smartSceneIds'})
  746. try:
  747. smart_scene_id_list = smart_scene_ids.split(',')
  748. # 获取序列号
  749. smart_scene_id = smart_scene_id_list[0]
  750. smart_scene_qs = SmartScene.objects.filter(id=smart_scene_id).values('device_id', 'sub_device_id')
  751. device_id = smart_scene_qs[0]['device_id']
  752. if device_id:
  753. serial_number = Device_Info.objects.filter(id=device_id).values('serial_number')[0]['serial_number']
  754. else:
  755. serial_number = GatewaySubDevice.objects.filter(id=smart_scene_qs[0]['sub_device_id']). \
  756. values('device__serial_number')[0]['device__serial_number']
  757. topic_name = SMART_SCENE_TOPIC.format(serial_number)
  758. with transaction.atomic():
  759. smart_scene_qs = SmartScene.objects.filter(id__in=smart_scene_id_list)
  760. # 删除定时任务
  761. smart_scene_data = smart_scene_qs.values('scene_data')
  762. for smart_scene in smart_scene_data:
  763. scene_data = smart_scene['scene_data']
  764. if scene_data:
  765. scene_data_dict = eval(scene_data)
  766. cls.del_celery_task(scene_data_dict)
  767. smart_scene_qs.delete()
  768. for smart_scene_id in smart_scene_id_list:
  769. # 通知设备删除场景id
  770. msg = {
  771. 'scene_event': SCENE_EVENT_DELETE,
  772. 'scene_id': int(smart_scene_id)
  773. }
  774. success = CommonService.req_publish_mqtt_msg(serial_number, topic_name, msg)
  775. try:
  776. assert success
  777. except AssertionError:
  778. return response.json(10044)
  779. time.sleep(0.3)
  780. return response.json(0)
  781. except Exception as e:
  782. return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  783. @staticmethod
  784. def scene_log(request_dict, response):
  785. """
  786. 查询场景日志
  787. @param request_dict: 请求参数
  788. @request_dict deviceId: 网关id
  789. @request_dict subDeviceId: 子设备id
  790. @request_dict page: 页数
  791. @request_dict size: 条数
  792. @request_dict startTime: 开始时间
  793. @request_dict endTime: 结束时间
  794. @param response: 响应对象
  795. @return: response
  796. """
  797. family_id = request_dict.get('familyId', None)
  798. device_id = request_dict.get('deviceId', None)
  799. sub_device_id = request_dict.get('subDeviceId', None)
  800. page = request_dict.get('page', None)
  801. size = request_dict.get('size', None)
  802. start_time = request_dict.get('startTime', None)
  803. end_time = request_dict.get('endTime', None)
  804. if not all([family_id, page, size]):
  805. return response.json(444, {'error param': 'familyId or page or size'})
  806. device_list = []
  807. sub_device_list = []
  808. if not device_id and not sub_device_id:
  809. family_room_device_qs = FamilyRoomDevice.objects.filter(family_id=family_id).values('device', 'sub_device')
  810. for device in family_room_device_qs:
  811. if device['device'] not in device_list:
  812. device_list.append(device['device'])
  813. if device['sub_device']:
  814. sub_device_list.append(device['sub_device'])
  815. elif sub_device_id: # 查询子设备
  816. family_room_device_qs = FamilyRoomDevice.objects.filter(family_id=family_id,
  817. sub_device=sub_device_id).values('device_id')
  818. for device in family_room_device_qs:
  819. device_list.append(device['device_id'])
  820. sub_device_list.append(sub_device_id)
  821. else: # 查询网关
  822. family_room_device_qs = FamilyRoomDevice.objects.filter(Q(family_id=family_id) & Q(device=device_id) &
  823. ~Q(sub_device=0)).values(
  824. 'sub_device')
  825. device_list.append(device_id)
  826. for device in family_room_device_qs:
  827. sub_device_list.append(device['sub_device'])
  828. try:
  829. page, size = int(page), int(size)
  830. scene_log_qs = SceneLog.objects.filter(Q(device_id__in=device_list) | Q(sub_device_id__in=sub_device_list))
  831. if start_time and end_time:
  832. scene_log_qs = scene_log_qs.filter(created_time__range=(start_time, end_time)).values(
  833. 'status',
  834. 'created_time',
  835. 'device_id',
  836. 'sub_device_id',
  837. 'scene_name',
  838. 'tasks').order_by(
  839. '-created_time')[(page - 1) * size:page * size]
  840. else:
  841. scene_log_qs = scene_log_qs.values('status', 'created_time', 'device_id', 'sub_device_id', 'scene_name',
  842. 'tasks').order_by(
  843. '-created_time')[(page - 1) * size:page * size]
  844. if not scene_log_qs.exists():
  845. return response.json(0, [])
  846. for item in scene_log_qs:
  847. if not device_id and not sub_device_id:
  848. if not item['sub_device_id']:
  849. device_qs = Device_Info.objects.filter(id=item['device_id']).values('Type')
  850. item['device_type'] = device_qs[0]['Type'] if device_qs.exists() else ''
  851. else:
  852. device_qs = GatewaySubDevice.objects.filter(id=item['sub_device_id']).values('device_type')
  853. item['device_type'] = device_qs[0]['device_type'] if device_qs.exists() else ''
  854. elif sub_device_id:
  855. device_qs = GatewaySubDevice.objects.filter(id=item['sub_device_id']).values('device_type')
  856. item['device_type'] = device_qs[0]['device_type'] if device_qs.exists() else ''
  857. else:
  858. device_qs = Device_Info.objects.filter(id=item['device_id']).values('Type')
  859. item['device_type'] = device_qs[0]['Type'] if device_qs.exists() else ''
  860. if item['tasks'] != '':
  861. item['tasks'] = eval(item['tasks'])
  862. scene_log_list = list(scene_log_qs)
  863. return response.json(0, scene_log_list)
  864. except Exception as e:
  865. print('error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  866. return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  867. @staticmethod
  868. def scene_log_date(request_dict, response):
  869. """
  870. 查询场景日志日期
  871. @param request_dict: 请求参数
  872. @request_dict deviceId: 网关id
  873. @request_dict subDeviceId: 子设备id
  874. @param response: 响应对象
  875. @return: response
  876. """
  877. family_id = request_dict.get('familyId', None)
  878. device_id = request_dict.get('deviceId', None)
  879. sub_device_id = request_dict.get('subDeviceId', None)
  880. if not family_id:
  881. return response.json(444, {'error param': 'familyId'})
  882. device_list = []
  883. sub_device_list = []
  884. if not device_id and not sub_device_id:
  885. family_room_device_qs = FamilyRoomDevice.objects.filter(family_id=family_id).values('device', 'sub_device')
  886. for device in family_room_device_qs:
  887. if device['device'] not in device_list:
  888. device_list.append(device['device'])
  889. if device['sub_device']:
  890. sub_device_list.append(device['sub_device'])
  891. elif sub_device_id:
  892. family_room_device_qs = FamilyRoomDevice.objects.filter(family_id=family_id, sub_device=sub_device_id)
  893. sub_device_list.append(sub_device_id)
  894. else:
  895. family_room_device_qs = FamilyRoomDevice.objects.filter(family_id=family_id, device=device_id).values(
  896. 'sub_device')
  897. device_list.append(device_id)
  898. for device in family_room_device_qs:
  899. sub_device_list.append(device['sub_device'])
  900. if not family_room_device_qs.exists():
  901. return response.json(173)
  902. try:
  903. scene_log_qs = SceneLog.objects.extra(
  904. select={'date': "FROM_UNIXTIME(created_time,'%%Y-%%m-%%d')"}).values('date'). \
  905. filter(Q(device_id__in=device_list) | Q(sub_device_id__in=sub_device_list)). \
  906. annotate(count=Count('created_time')). \
  907. order_by('-date')[:31]
  908. log_date_list = []
  909. for scene_log in scene_log_qs:
  910. log_date_list.append({
  911. 'timestamp': CommonService.str_to_timestamp(scene_log['date'], '%Y-%m-%d'),
  912. 'count': scene_log['count'],
  913. 'format': scene_log['date'],
  914. })
  915. return response.json(0, log_date_list)
  916. except Exception as e:
  917. return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  918. @staticmethod
  919. def get_scene_data(request_dict, response):
  920. """
  921. 设备获取智能场景数据
  922. @param request_dict: 请求参数
  923. @request_dict serial_number: 序列号
  924. @param response: 响应对象
  925. @return: response
  926. """
  927. serial_number = request_dict.get('serial_number', None)
  928. if not serial_number:
  929. return response.json(444, {'error param': 'serial_number'})
  930. try:
  931. device_info_qs = Device_Info.objects.filter(serial_number=serial_number).values('id')
  932. if not device_info_qs.exists():
  933. return response.json(173)
  934. device_id = device_info_qs[0]['id']
  935. sub_device_id_list = GatewaySubDevice.objects.filter(device_id=device_id).values_list('id', flat=True)
  936. if sub_device_id_list:
  937. smart_scene_qs = SmartScene.objects.filter(
  938. Q(device_id=device_id) | Q(sub_device_id__in=sub_device_id_list))
  939. else:
  940. smart_scene_qs = SmartScene.objects.filter(device_id=device_id)
  941. if not smart_scene_qs.exists():
  942. return response.json(173)
  943. # 下发智能场景数据
  944. smart_scene_qs = smart_scene_qs.values('device_data')
  945. topic_name = SMART_SCENE_TOPIC.format(serial_number)
  946. for smart_scene in smart_scene_qs:
  947. msg = eval(smart_scene['device_data'])
  948. success = CommonService.req_publish_mqtt_msg(serial_number, topic_name, msg)
  949. try:
  950. assert success
  951. except AssertionError:
  952. return response.json(10044)
  953. time.sleep(2)
  954. # 下发智能按钮数据
  955. smart_button_qs = GatewaySubDevice.objects.filter(device_id=device_id,
  956. device_type=SENSOR_TYPE['smart_button']).values(
  957. 'ieee_addr', 'is_tampered')
  958. if smart_button_qs.exists():
  959. sos_count = smart_button_qs.count()
  960. for index, smart_button in enumerate(smart_button_qs):
  961. msg = {
  962. 'sos_count': sos_count, # 该网关下的智能按钮数量
  963. 'sos_num': index + 1, # 第几个按钮
  964. 'sensor_ieee_addr': smart_button['ieee_addr'],
  965. 'sos_select': smart_button['is_tampered']
  966. }
  967. success = CommonService.req_publish_mqtt_msg(serial_number, topic_name, msg)
  968. try:
  969. assert success
  970. except AssertionError:
  971. return response.json(10044)
  972. time.sleep(2)
  973. return response.json(0)
  974. except Exception as e:
  975. return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  976. @staticmethod
  977. def voice_audition(request_dict, response):
  978. """
  979. 智能场景音频试听
  980. @param request_dict: 请求参数
  981. @request_dict serial_number: 序列号
  982. @request_dict voiceId: 音频id
  983. @param response: 响应对象
  984. @return: response
  985. """
  986. serial_number = request_dict.get('serial_number', None)
  987. voice_id = request_dict.get('voiceId', None)
  988. if not all([serial_number, voice_id]):
  989. return response.json(444)
  990. try:
  991. topic_name = VOICE_AUDITION_TOPIC.format(serial_number)
  992. msg = {'voice_id': int(voice_id)}
  993. success = CommonService.req_publish_mqtt_msg(serial_number, topic_name, msg)
  994. try:
  995. assert success
  996. except AssertionError:
  997. return response.json(10044)
  998. return response.json(0)
  999. except Exception as e:
  1000. return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  1001. @classmethod
  1002. def get_task_list_and_scene_data(
  1003. cls, conditions_dict, is_set_time, minutes, repeat, tz, now_time, tasks_list, scene_id,
  1004. scene_data_dict=None):
  1005. """
  1006. 获取设备任务数据和场景数据
  1007. @param tasks_list: app任务列表
  1008. @param conditions_dict: 条件
  1009. @param is_set_time: 条件是否为设置时间
  1010. @param minutes: 时间分钟数
  1011. @param repeat: 重复星期周期的十进制
  1012. @param tz: 时区
  1013. @param now_time: 当前时间
  1014. @param scene_id: 场景id
  1015. @param scene_data_dict: 场景数据
  1016. @return: task_list, scene_data
  1017. """
  1018. # 删除旧的定时任务
  1019. cls.del_celery_task(scene_data_dict)
  1020. task_list = []
  1021. scene_task_list = []
  1022. # 不需要设备触发的任务
  1023. no_device_task = True if is_set_time else False
  1024. is_last_task = False
  1025. tasks_len = len(tasks_list)
  1026. # 组织条件数据
  1027. condition = cls.get_condition(conditions_dict, is_set_time, minutes, repeat, now_time, tz)
  1028. for index, task in enumerate(tasks_list):
  1029. # 判断是否为列表的最后一个元素
  1030. if index == tasks_len - 1:
  1031. is_last_task = True
  1032. sensor_type = int(task['device_type'])
  1033. # 处理联动设备数据
  1034. # 不用添加到设备的任务列表,添加到场景数据任务列表
  1035. if sensor_type in LINKAGE_DEVICE_TYPE_LIST:
  1036. serial_number = task['serial_number']
  1037. event_type = int(task['event_type'])
  1038. delay_time = task['delay_time']
  1039. task_temp = {
  1040. 'device_type': sensor_type,
  1041. 'event_type': event_type,
  1042. 'delay_time': delay_time,
  1043. 'serial_number': serial_number
  1044. }
  1045. # 如果条件为设置时间,创建定时任务
  1046. if is_set_time:
  1047. smart_scene_id = 0
  1048. # 无设备任务,且是最后一个任务,需要通过scene_id上报场景日志
  1049. if no_device_task and is_last_task:
  1050. smart_scene_id = scene_id
  1051. task_temp['task_id'] = cls.create_celery_task(
  1052. condition, minutes, delay_time, tz, repeat, sensor_type, event_type, serial_number,
  1053. smart_scene_id, index)
  1054. scene_task_list.append(task_temp)
  1055. else:
  1056. no_device_task = False
  1057. task_temp = {
  1058. 'sensor_type': sensor_type,
  1059. 'sensor_delay': 0
  1060. }
  1061. # 延时
  1062. if 'delay_time' in task and task['delay_time'] != 0:
  1063. task_temp['sensor_delay'] = task['delay_time']
  1064. # 不为-1时需要其他数据
  1065. if sensor_type != -1:
  1066. task_temp['sensor_action'] = int(task['event_type'])
  1067. # 子设备返回长地址
  1068. sub_device_id = task.get('subDeviceId', None)
  1069. if sub_device_id:
  1070. sub_device_qs = GatewaySubDevice.objects.filter(id=sub_device_id).values('ieee_addr').first()
  1071. task_temp['sensor_ieee_addr'] = sub_device_qs['ieee_addr']
  1072. # 网关添加报警类型数据
  1073. else:
  1074. task_temp['voice_type'] = task.get('voice_type')
  1075. task_temp['voice_id'] = task.get('voice_id')
  1076. task_temp['count'] = task.get('count')
  1077. task_temp['delay_time'] = task.get('delay_time')
  1078. task_temp['duration'] = task.get('duration')
  1079. task_temp['value_type'] = task.get('value_type')
  1080. task_list.append(task_temp)
  1081. # 组织完整场景数据
  1082. scene_data = {
  1083. 'condition': condition,
  1084. 'task_list': scene_task_list,
  1085. 'no_device_task': no_device_task
  1086. }
  1087. return task_list, scene_data
  1088. @classmethod
  1089. def get_condition(cls, conditions_dict, is_set_time, minutes, repeat, now_time, tz):
  1090. """
  1091. 获取场景条件
  1092. @param conditions_dict: 条件数据
  1093. @param is_set_time: 条件是否为设置时间
  1094. @param minutes: 时间分钟数
  1095. @param repeat: 重复星期周期的十进制
  1096. @param now_time: 当前时间
  1097. @param tz: 时区
  1098. @return: condition
  1099. """
  1100. condition = {}
  1101. # 条件为设置时间
  1102. if is_set_time:
  1103. hour, minute = divmod(minutes, 60)
  1104. # 一次性任务
  1105. if repeat == 0:
  1106. condition['time'] = 'date'
  1107. # 根据时间戳和时区获取年月日,拼接由分钟转换出来的时间
  1108. time_string = CommonService.get_date_from_timestamp(now_time, tz)
  1109. time_string = time_string[:time_string.index(' ')]
  1110. time_string += ' {:02d}:{:02d}:00'.format(hour, minute)
  1111. time_stamp = CommonService.convert_to_timestamp(tz, time_string)
  1112. if time_stamp < now_time:
  1113. time_stamp += 24 * 60 * 60
  1114. time_dict = {
  1115. 'time_stamp': time_stamp
  1116. }
  1117. else:
  1118. condition['time'] = 'cron'
  1119. weeks = cls.int_to_weeks(repeat)
  1120. time_dict = {
  1121. 'weeks': weeks,
  1122. 'hour': hour,
  1123. 'minute': minute
  1124. }
  1125. condition['time_dict'] = time_dict
  1126. else:
  1127. device_type = int(conditions_dict['sensor']['device_type'])
  1128. event_type = int(conditions_dict['sensor']['eventValues'][0]['event_type'])
  1129. condition['event_type'] = event_type
  1130. # 温湿度传感器,取值
  1131. if device_type == SENSOR_TYPE['tem_hum_sensor']:
  1132. condition['value'] = event_type = conditions_dict['sensor']['eventValues'][0]['value']
  1133. return condition
  1134. @staticmethod
  1135. def time_conflict(sub_device_id, conditions, is_all_day, request_dict, smart_scene_id=None):
  1136. """
  1137. 判断传感器是否创建过条件相同且生效时间冲突的场景
  1138. @param sub_device_id: 传感器设备id
  1139. @param conditions: 场景条件
  1140. @param is_all_day: 全天标识
  1141. @param request_dict:
  1142. @param smart_scene_id: 场景id,编辑场景时传
  1143. @return: bool, True: 冲突, False: 不冲突
  1144. """
  1145. # 不设置时间不会冲突
  1146. if is_all_day is None:
  1147. return False
  1148. # 查询设置过时间的数据
  1149. if smart_scene_id is None: # 创建场景
  1150. smart_scene_qs = SmartScene.objects.filter(
  1151. ~Q(is_all_day=0),
  1152. sub_device_id=sub_device_id,
  1153. conditions=conditions).values('effective_time_id')
  1154. else: # 编辑场景,过滤本身场景数据
  1155. smart_scene_qs = SmartScene.objects.filter(
  1156. ~Q(id=smart_scene_id),
  1157. ~Q(is_all_day=0),
  1158. sub_device_id=sub_device_id,
  1159. conditions=conditions).values('effective_time_id')
  1160. if not smart_scene_qs.exists():
  1161. return False
  1162. # 再设置全天必冲突
  1163. if is_all_day == 1:
  1164. return True
  1165. # 非全天
  1166. elif is_all_day == 2:
  1167. start_time = int(request_dict.get('startTime'))
  1168. end_time = int(request_dict.get('endTime'))
  1169. repeat = int(request_dict.get('repeat'))
  1170. for smart_scene in smart_scene_qs:
  1171. effective_time_id = smart_scene['effective_time_id']
  1172. effective_time_qs = EffectiveTime.objects.filter(id=effective_time_id).\
  1173. values('start_time', 'end_time', 'repeat')
  1174. if effective_time_qs.exists():
  1175. old_start_time = effective_time_qs[0]['start_time']
  1176. old_end_time = effective_time_qs[0]['end_time']
  1177. old_repeat = effective_time_qs[0]['repeat']
  1178. # 每天重复
  1179. if repeat == 127:
  1180. # 判断时间是否在已设置过的时间范围之内
  1181. if old_start_time <= start_time <= old_end_time or \
  1182. old_start_time <= end_time <= old_end_time:
  1183. return True
  1184. else:
  1185. # 有相同的重复天
  1186. if repeat & old_repeat != 0:
  1187. # 判断时间是否在已设置过的时间范围之内
  1188. if old_start_time <= start_time <= old_end_time or \
  1189. old_start_time <= end_time <= old_end_time:
  1190. return True
  1191. return False
  1192. @classmethod
  1193. def create_celery_task(
  1194. cls, condition, minutes, delay_time, tz, repeat, device_type, event_type, serial_number, scene_id, index):
  1195. """
  1196. 创建定时任务
  1197. 返回任务id和时间
  1198. @param condition: 条件数据
  1199. @param minutes: 分钟时间
  1200. @param delay_time: 延迟时间
  1201. @param tz: 时区
  1202. @param repeat: 星期周期的十进制数
  1203. @param device_type: 设备类型
  1204. @param event_type: 事件类型
  1205. @param serial_number: 序列号
  1206. @param scene_id: 场景id
  1207. @param index: 任务下标
  1208. @return: task_id
  1209. """
  1210. celery_beat_obj = CeleryBeatObj()
  1211. # 任务名拼接当前时间戳,防止重复
  1212. name = '{}_{}_{}_'.format(serial_number, str(int(time.time())), str(index))
  1213. kwargs = {
  1214. 'device_type': device_type,
  1215. 'event_type': event_type,
  1216. 'serial_number': serial_number,
  1217. 'scene_id': scene_id
  1218. }
  1219. # 一次性任务
  1220. if repeat == 0:
  1221. time_stamp = condition['time_dict']['time_stamp'] + delay_time
  1222. name += str(time_stamp)
  1223. celery_beat_obj.creat_clocked_task(
  1224. name=name, task=SMART_SCENE_TASK, time_stamp=time_stamp, kwargs=kwargs)
  1225. # 周期任务
  1226. else:
  1227. hour, minute, second, is_next_day = cls.handle_delay_time(minutes, delay_time)
  1228. # 加上延时,如果执行时间超过23:59,隔天执行
  1229. weeks = cls.int_to_weeks(repeat, is_next_day)
  1230. time_str = weeks + '_{:02d}{:02d}{:02d}'.format(hour, minute, second)
  1231. name += time_str
  1232. celery_beat_obj.creat_crontab_task(
  1233. timezone_offset=tz, name=name, task=SMART_SCENE_TASK, minute=minute, hour=hour, day_of_week=weeks,
  1234. kwargs=kwargs)
  1235. return name
  1236. @staticmethod
  1237. def handle_delay_time(minutes, delay_time):
  1238. """
  1239. 处理延迟时间
  1240. @param minutes: 时间分钟数,如1439代表23:59
  1241. @param delay_time: 延迟时间,单位:秒
  1242. @return: hour, minute, second, is_next_day
  1243. """
  1244. is_next_day = False
  1245. hour, minute = divmod(minutes, 60)
  1246. # 延迟时间转为分钟,如果加上时间分钟数大于1439,隔天执行
  1247. minute, second = divmod(delay_time, 60)
  1248. total_minutes = minutes + minute
  1249. ex_min = total_minutes - 1439
  1250. if ex_min <= 0:
  1251. hour, minute = divmod(total_minutes, 60)
  1252. else:
  1253. hour, minute = divmod(ex_min, 60)
  1254. is_next_day = True
  1255. return hour, minute, second, is_next_day
  1256. @staticmethod
  1257. def int_to_weeks(repeat, is_next_day=False):
  1258. """
  1259. 十进制转星期周期
  1260. @param repeat: 星期周期的十进制数,如127 -> 每天 -> 0,1,2,3,4,5,6或*
  1261. @param is_next_day: 是否隔天
  1262. @return: weeks
  1263. """
  1264. if repeat == 127:
  1265. return '*'
  1266. # 十进制转为7位的二进制并倒序
  1267. bin_str = bin(repeat)[2:].zfill(7)[::-1]
  1268. # 生成星期周期字符串
  1269. weeks = ''
  1270. next_day = 1 if is_next_day else 0
  1271. for i, bit in enumerate(bin_str):
  1272. if bit == '1':
  1273. # 7 -> 0
  1274. week = i + next_day
  1275. if week == 7:
  1276. week = 0
  1277. weeks += str(week) + ','
  1278. # 删除最后一个逗号并返回结果
  1279. return weeks[:-1]
  1280. @staticmethod
  1281. def del_celery_task(scene_data_dict):
  1282. """
  1283. 删除定时任务
  1284. @param scene_data_dict: 场景数据
  1285. @return:
  1286. """
  1287. if scene_data_dict is not None:
  1288. if scene_data_dict['condition'].get('time'):
  1289. celery_beat_obj = CeleryBeatObj()
  1290. time_task_list = scene_data_dict['task_list']
  1291. for time_task in time_task_list:
  1292. # 存在任务则删除
  1293. task_id = time_task.get('task_id')
  1294. if task_id:
  1295. try:
  1296. celery_beat_obj.del_task(task_id)
  1297. except Exception:
  1298. continue
  1299. @classmethod
  1300. def pause_or_resume_job(cls, scene_data_dict, scene_status, scene_id):
  1301. """
  1302. 暂停或恢复定时任务
  1303. @param scene_data_dict: 场景数据
  1304. @param scene_status: 场景状态: SCENE_STATUS_ON, SCENE_STATUS_OFF
  1305. @param scene_id: 场景id
  1306. @return: None or scene_data_dict
  1307. """
  1308. # 判断条件是否为设置时间
  1309. time_type = scene_data_dict['condition'].get('time')
  1310. if time_type:
  1311. celery_beat_obj = CeleryBeatObj()
  1312. # 关闭: 暂停任务, 打开: 恢复任务
  1313. if scene_status == SCENE_STATUS_OFF:
  1314. cls.pause_time_job(celery_beat_obj, scene_data_dict)
  1315. else:
  1316. return cls.resume_time_job(celery_beat_obj, time_type, scene_data_dict, scene_id)
  1317. @staticmethod
  1318. def pause_time_job(celery_beat_obj, scene_data_dict):
  1319. """
  1320. 暂停定时任务
  1321. @param celery_beat_obj: celery_beat对象
  1322. @param scene_data_dict: 场景数据
  1323. @return:
  1324. """
  1325. time_task_list = scene_data_dict['task_list']
  1326. for time_task in time_task_list:
  1327. task_id = time_task.get('task_id')
  1328. if task_id:
  1329. try:
  1330. celery_beat_obj.disable_task(task_id)
  1331. except Exception:
  1332. continue
  1333. @classmethod
  1334. def resume_time_job(cls, celery_beat_obj, time_type, scene_data_dict, scene_id):
  1335. """
  1336. 恢复定时任务
  1337. @param celery_beat_obj: celery_beat对象
  1338. @param time_type: 时间类型: date, cron
  1339. @param scene_data_dict: 场景数据
  1340. @param scene_id: 场景id
  1341. @return: None or time_stamp
  1342. """
  1343. time_task_list = scene_data_dict['task_list']
  1344. if time_type == 'cron':
  1345. for time_task in time_task_list:
  1346. task_id = time_task.get('task_id')
  1347. if task_id:
  1348. try:
  1349. celery_beat_obj.enable_task(task_id)
  1350. except Exception:
  1351. continue
  1352. return None
  1353. else:
  1354. now_time = int(time.time())
  1355. time_stamp = scene_data_dict['condition']['time_dict']['time_stamp']
  1356. # 恢复定时任务
  1357. if time_stamp > now_time:
  1358. for time_task in time_task_list:
  1359. task_id = time_task.get('task_id')
  1360. if task_id:
  1361. try:
  1362. celery_beat_obj.enable_task(task_id)
  1363. except Exception:
  1364. continue
  1365. return None
  1366. else:
  1367. # 创建新的定时任务,时间+24小时
  1368. time_stamp += 24 * 60 * 60
  1369. task_list_len = len(time_task_list)
  1370. for index, time_task in enumerate(time_task_list):
  1371. device_type = time_task['device_type']
  1372. event_type = time_task['event_type']
  1373. serial_number = time_task['serial_number']
  1374. task_id = '{}_{}_{}'.format(serial_number, str(now_time), str(time_stamp))
  1375. # 最后一个任务使用传入的scene_id
  1376. smart_scene_id = scene_id if index == task_list_len-1 else 0
  1377. kwargs = {
  1378. 'device_type': device_type,
  1379. 'event_type': event_type,
  1380. 'serial_number': serial_number,
  1381. 'scene_id': smart_scene_id
  1382. }
  1383. celery_beat_obj.creat_clocked_task(
  1384. name=task_id, task=SMART_SCENE_TASK, time_stamp=time_stamp, kwargs=kwargs)
  1385. # 更新task_id
  1386. time_task['task_id'] = task_id
  1387. # 更新场景数据的时间戳
  1388. scene_data_dict['condition']['time_dict']['time_stamp'] = time_stamp
  1389. return scene_data_dict
  1390. #
  1391. # ___====-_ _-====___
  1392. # _--^^^#####// \\#####^^^--_
  1393. # _-^##########// ( ) \\##########^-_
  1394. # -############// |\^^/| \\############-
  1395. # _/############// (@::@) \\############\_
  1396. # /#############(( \\// ))#############\
  1397. # -###############\\ (oo) //###############-
  1398. # -#################\\ / VV \ //#################-
  1399. # -###################\\/ \//###################-
  1400. # _#/|##########/\######( /\ )######/\##########|\#_
  1401. # |/ |#/\#/\#/\/ \#/\##\ | | /##/\#/ \/\#/\#/\#| \|
  1402. # ` |/ V V ` V \#\| | | |/#/ V ' V V \| '
  1403. # ` ` ` ` / | | | | \ ' ' ' '
  1404. # ( | | | | )
  1405. # __\ | | | | /__
  1406. # (vvv(VVV)(VVV)vvv)
  1407. # 神兽保佑
  1408. # 代码无BUG!
  1409. #