EquipmentInfoService.py 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631
  1. # -*- encoding: utf-8 -*-
  2. """
  3. @File : EquipmentInfoService.py
  4. @Time : 2022/4/14 17:28
  5. @Author : stephen
  6. @Email : zhangdongming@asj6.wecom.work
  7. @Software: PyCharm
  8. """
  9. import datetime
  10. import itertools
  11. import time
  12. from django.db.models import Value, CharField, Q
  13. from Model.models import EquipmentInfoMonday, EquipmentInfoTuesday, EquipmentInfoWednesday, EquipmentInfoThursday, \
  14. EquipmentInfoFriday, EquipmentInfoSaturday, EquipmentInfoSunday, EquipmentInfo1, EquipmentInfo2, EquipmentInfo3, \
  15. EquipmentInfo4, EquipmentInfo5, EquipmentInfo6, EquipmentInfo7, EquipmentInfo8, EquipmentInfo9, EquipmentInfo10, \
  16. EquipmentInfo11, EquipmentInfo12, EquipmentInfo13, EquipmentInfo14, EquipmentInfo15, EquipmentInfo16, \
  17. EquipmentInfo17, EquipmentInfo18, EquipmentInfo19, EquipmentInfo20
  18. from Object.utils import LocalDateTimeUtil
  19. EQUIPMENT_INFO_MODEL_LIST = [EquipmentInfo1, EquipmentInfo2, EquipmentInfo3, EquipmentInfo4, EquipmentInfo5,
  20. EquipmentInfo6, EquipmentInfo7, EquipmentInfo8, EquipmentInfo9, EquipmentInfo10,
  21. EquipmentInfo11, EquipmentInfo12, EquipmentInfo13, EquipmentInfo14, EquipmentInfo15,
  22. EquipmentInfo16, EquipmentInfo17, EquipmentInfo18, EquipmentInfo19, EquipmentInfo20]
  23. """
  24. 设备分表查询Service
  25. 因数据量不断增加,单表保留近七天数据进行分表优化设计
  26. 进行拆分为七张表星期一至星期天进行分表存储。分担单表存储读写压力。
  27. """
  28. class EquipmentInfoService:
  29. @staticmethod
  30. def get_equipment_info_model(dt, val=0):
  31. """
  32. 根据日期判断是星期几,返回相应的Model对象
  33. @param val: 1-7代表week
  34. @param dt: 日期 例:2022-03-03
  35. @return: 星期一至星期天equipment_info对象实例
  36. """
  37. week = 1
  38. if dt:
  39. week = LocalDateTimeUtil.date_to_week(dt)
  40. if 0 < val < 8:
  41. week = val
  42. equipment_info = None
  43. if week == 1:
  44. equipment_info = EquipmentInfoMonday.objects.all().annotate(tab_val=Value('1', output_field=CharField()))
  45. elif week == 2:
  46. equipment_info = EquipmentInfoTuesday.objects.all().annotate(tab_val=Value('2', output_field=CharField()))
  47. elif week == 3:
  48. equipment_info = EquipmentInfoWednesday.objects.all().annotate(tab_val=Value('3', output_field=CharField()))
  49. elif week == 4:
  50. equipment_info = EquipmentInfoThursday.objects.all().annotate(tab_val=Value('4', output_field=CharField()))
  51. elif week == 5:
  52. equipment_info = EquipmentInfoFriday.objects.all().annotate(tab_val=Value('5', output_field=CharField()))
  53. elif week == 6:
  54. equipment_info = EquipmentInfoSaturday.objects.all().annotate(tab_val=Value('6', output_field=CharField()))
  55. elif week == 7:
  56. equipment_info = EquipmentInfoSunday.objects.all().annotate(tab_val=Value('7', output_field=CharField()))
  57. return equipment_info
  58. @classmethod
  59. def find_by_start_time_equipment_info(cls, page, size, user_id, start_time, end_time, event_type,
  60. uid_list):
  61. """
  62. 通过start_time查找指定日期当天设备消息推送
  63. @param page: 页数
  64. @param size: 每页条数
  65. @param user_id: 设备用户id
  66. @param start_time: 开始时间
  67. @param end_time: 结束时间
  68. @param event_type: 事件类型
  69. @param uid_list: 设备uid列表
  70. @return: result 查询结果
  71. """
  72. if start_time and end_time:
  73. start_date = datetime.datetime.fromtimestamp(int(start_time))
  74. # 根据开始日期,获取设备信息查询对象
  75. qs = EquipmentInfoService.get_equipment_info_model(str(start_date.date()), 0)
  76. # 调用查询方法
  77. qs = cls.query_equipment_info(qs, user_id, start_time, end_time, event_type,
  78. uid_list)
  79. # 时区问题
  80. week = LocalDateTimeUtil.date_to_week(str(start_date.date()))
  81. if week > 0:
  82. # 根据筛选日期 查找昨天数据
  83. yesterday = 7 if week == 1 else week - 1
  84. yesterday_info = EquipmentInfoService.get_equipment_info_model('', yesterday)
  85. yesterday_info = cls.query_equipment_info(yesterday_info, user_id, start_time, end_time, event_type,
  86. uid_list)
  87. # 根据筛选日期 查找明天数据
  88. tomorrow = 1 if week == 7 else week + 1
  89. tomorrow_info = EquipmentInfoService.get_equipment_info_model('', tomorrow)
  90. tomorrow_info = cls.query_equipment_info(tomorrow_info, user_id, start_time, end_time, event_type,
  91. uid_list)
  92. qs = qs.union(yesterday_info, tomorrow_info, all=True)
  93. if qs.exists():
  94. count = qs.count()
  95. qs_page = cls.get_equipment_info_page(qs, page, size)
  96. return qs_page, count
  97. return None, 0
  98. @classmethod
  99. def get_equipment_info_week_all(cls, page, size, user_id, start_time, end_time, event_type,
  100. uid_list):
  101. """
  102. 分表查询近七天设备消息推送
  103. @param page: 页数
  104. @param size: 分页大小
  105. @param user_id: 设备用户id
  106. @param start_time: 事件开始时间
  107. @param end_time: 事件结束时间
  108. @param event_type: 事件类型
  109. @param uid_list: uid列表
  110. @return: qs_page, count 结果集
  111. """
  112. # 星期一设备信息查询
  113. monday_qs = EquipmentInfoService.get_equipment_info_model('', 1)
  114. monday_qs = cls.query_equipment_info(monday_qs, user_id, start_time, end_time, event_type,
  115. uid_list)
  116. # 星期二设备信息查询
  117. tuesday_qs = EquipmentInfoService.get_equipment_info_model('', 2)
  118. tuesday_qs = cls.query_equipment_info(tuesday_qs, user_id, start_time, end_time, event_type,
  119. uid_list)
  120. # 星期三设备信息查询
  121. wednesday_qs = EquipmentInfoService.get_equipment_info_model('', 3)
  122. wednesday_qs = cls.query_equipment_info(wednesday_qs, user_id, start_time, end_time, event_type,
  123. uid_list)
  124. # 星期四设备信息查询
  125. thursday_qs = EquipmentInfoService.get_equipment_info_model('', 4)
  126. thursday_qs = cls.query_equipment_info(thursday_qs, user_id, start_time, end_time, event_type,
  127. uid_list)
  128. # 星期五设备信息查询
  129. friday_qs = EquipmentInfoService.get_equipment_info_model('', 5)
  130. friday_qs = cls.query_equipment_info(friday_qs, user_id, start_time, end_time, event_type,
  131. uid_list)
  132. # 星期六设备信息查询
  133. saturday_qs = EquipmentInfoService.get_equipment_info_model('', 6)
  134. saturday_qs = cls.query_equipment_info(saturday_qs, user_id, start_time, end_time, event_type,
  135. uid_list)
  136. # 星期天设备信息查询
  137. sunday_qs = EquipmentInfoService.get_equipment_info_model('', 7)
  138. sunday_qs = cls.query_equipment_info(sunday_qs, user_id, start_time, end_time, event_type,
  139. uid_list)
  140. result = monday_qs.union(tuesday_qs, wednesday_qs, thursday_qs, friday_qs, saturday_qs, sunday_qs, all=True)
  141. count = result.count()
  142. qs_page = cls.get_equipment_info_page(result, page, size)
  143. return qs_page, count
  144. @classmethod
  145. def query_equipment_info(cls, qs, user_id, start_time, end_time, event_type,
  146. uid_list):
  147. """
  148. 设备信息条件查询,根据分表设计,默认条件event_time大于七天前时间
  149. @param qs: 设备信息查询对象
  150. @param user_id: 设备用户id
  151. @param start_time: 开始时间
  152. @param end_time: 结束时间
  153. @param event_type: 事件类型
  154. @param uid_list: 设备uid列表
  155. @return: result 设备信息结果集
  156. """
  157. now_time = int(time.time())
  158. # 获取七天前时间戳
  159. seven_days_before_time = LocalDateTimeUtil.get_before_days_timestamp(now_time, 7)
  160. # 默认查询当前表event_time大于七天前时间
  161. qs = qs.filter(event_time__gt=seven_days_before_time)
  162. if user_id:
  163. qs = qs.filter(device_user_id=user_id)
  164. if event_type:
  165. # 多类型查询
  166. eventTypeList = cls.get_comb_event_type(event_type)
  167. # eventTypeList += cls.get_combo_type_bins(event_type)
  168. eventTypeList = list(set(eventTypeList))
  169. tags = cls.get_event_tag(event_type)
  170. if eventTypeList:
  171. qs = qs.filter(Q(event_type__in=eventTypeList, event_tag='') | Q(event_tag__regex=tags))
  172. elif tags:
  173. qs = qs.filter(event_tag__regex=tags)
  174. if start_time and end_time:
  175. qs = qs.filter(event_time__range=(start_time, end_time))
  176. else:
  177. qs = qs.filter(event_time__range=(start_time, now_time))
  178. if uid_list:
  179. uid_list = uid_list.split(',')
  180. qs = qs.filter(device_uid__in=uid_list)
  181. return qs
  182. @classmethod
  183. def get_equipment_info_page(cls, equipment_info_qs, page, size):
  184. """
  185. 获取查询结果集进行排序、分页,遍历重命名字典key(主要针对原函数返回结果集)
  186. @param equipment_info_qs: 设备信息结果集
  187. @param page: 页数
  188. @param size: 分页大小
  189. @return: qs_page 遍历后的设备信息结果集
  190. """
  191. equipment_info_qs = equipment_info_qs.values('id', 'device_uid', 'device_nick_name', 'channel', 'event_type',
  192. 'status', 'answer_status', 'alarm',
  193. 'event_time', 'receive_time', 'is_st', 'add_time',
  194. 'storage_location', 'border_coords', 'tab_val', 'event_tag')
  195. equipment_info_qs = equipment_info_qs.order_by('-event_time')
  196. qs_page = equipment_info_qs[(page - 1) * size:page * size]
  197. if not qs_page or not qs_page.exists() or qs_page.count == 0:
  198. return qs_page
  199. for item in qs_page:
  200. # 星期表值
  201. tab_val = item['tab_val']
  202. # id = 星期表值+id
  203. item['id'] = int(tab_val + str(item['id']))
  204. item['devUid'] = item['device_uid']
  205. item['devNickName'] = item['device_nick_name']
  206. item['Channel'] = item['channel']
  207. item['eventType'] = item['event_type']
  208. item['eventTime'] = item['event_time']
  209. item['receiveTime'] = item['receive_time']
  210. item['addTime'] = item['add_time']
  211. item['borderCoords'] = item['border_coords']
  212. item['eventTag'] = item['event_tag']
  213. item.pop('device_uid')
  214. item.pop('device_nick_name')
  215. item.pop('channel')
  216. item.pop('event_type')
  217. item.pop('event_time')
  218. item.pop('receive_time')
  219. item.pop('add_time')
  220. item.pop('border_coords')
  221. item.pop('tab_val')
  222. item.pop('event_tag')
  223. return qs_page
  224. @classmethod
  225. def get_comb_event_type(cls, event_type):
  226. """
  227. 重新组合ai消息类型查询,使其支持ai多标签查询
  228. @param event_type: 消息类型
  229. @return: event_type_list 消息类型数组
  230. """
  231. if ',' in event_type:
  232. event_type_list = event_type.split(',')
  233. event_type_list = [int(i.strip()) for i in event_type_list]
  234. else:
  235. event_type_list = [int(event_type)]
  236. ai_event_type_list = []
  237. normal_event_type_list = []
  238. a_type = [60, 61, 62, 63, 64, 65, 66]
  239. for val in event_type_list:
  240. if val <= 4: # 分离出ai类型,以便后续组合ai标签,目前只存在4个ai类型1,2,3,4
  241. ai_event_type_list.append(val)
  242. else:
  243. if val not in a_type:
  244. normal_event_type_list.append(val)
  245. if len(ai_event_type_list) < 1:
  246. return normal_event_type_list
  247. ai_event_type_list.sort()
  248. ai_type = [1, 2, 3, 4] # AI目前所有的标签,1人,2车,3宠物,4包裹,后续有新类型需要这里加, 后续会优化,存在表里,包裹存对应的aws标签
  249. comb_ai_event_type = []
  250. seen = set()
  251. for i in range(1, len(ai_type) + 1): # 计算所有组合,如[1, 2, 3, 4], 4取1,4取2,4取3,4取4
  252. for s in itertools.combinations(ai_type, i):
  253. if s not in seen: # 去除重复项, 如a=[1,2,3,4,4],会有两个[1,2,3,4,4],[1,2,3,4,4]的组合
  254. seen.add(s)
  255. s_list = list(s)
  256. for ai_event_type in ai_event_type_list:
  257. if ai_event_type in s_list: # 排除没有选择的标签组合
  258. if s_list not in comb_ai_event_type:
  259. s_list = [str(v) for v in s_list]
  260. comb_ai_event_type.append(s_list)
  261. regroup_list = []
  262. for val in comb_ai_event_type: # 组合ai类型组合,如[[2,3],[1,3]] -> [23, 13]
  263. val = ''.join(val)
  264. regroup_list.append(int(val))
  265. group_list = regroup_list + normal_event_type_list # 加上普通移动消息类型
  266. return group_list
  267. @classmethod
  268. def get_all_comb_event_type(cls):
  269. """
  270. 计算ai消息类型全组合
  271. @return: event_type_list ai所有消息类型数组
  272. """
  273. ai_type = [1, 2, 3, 4] # AI目前所有的标签,1人,2车,3宠物,4包裹,后续有新类型需要这里加, 后续会优化,存在表里,包裹存对应的aws标签
  274. comb_ai_event_type = []
  275. for i in range(1, len(ai_type) + 1): # 计算所有组合,如[1, 2, 3, 4], 4取1,4取2,4取3,4取4
  276. for s in itertools.combinations(ai_type, i):
  277. s_list = list(s)
  278. s_list = [str(v) for v in s_list]
  279. comb_ai_event_type.append(s_list)
  280. regroup_list = []
  281. for val in comb_ai_event_type: # 组合ai类型组合,如[[2,3],[1,3]] -> [23, 13]
  282. val = ''.join(val)
  283. regroup_list.append(int(val))
  284. return regroup_list
  285. @classmethod
  286. def get_combo_types(cls, event_type, event_tag):
  287. """
  288. 获取设备算法组合类型
  289. 51:移动侦测,52:传感器报警,53:影像遗失,54:PIR,55:门磁报警,56:外部发报,57:人型报警(提示:有人出现),58:车型,59:宠物,60:人脸,61:异响,
  290. 62:区域闯入,63:区域闯出,64:长时间无人检测,65:长时间无人检测
  291. 0:代表空字符,702:摄像头休眠,703:摄像头唤醒,704:电量过低
  292. AWS AI识别 1:人形,2:车型,3:宠物,4:包裹。云端AI类型
  293. @param event_tag:
  294. @param event_type:
  295. @return:
  296. """
  297. try:
  298. types = []
  299. if event_tag:
  300. res = event_tag.split(',')
  301. types = [int(var) for var in res if var]
  302. return types
  303. res_type = cls.is_type_exist(event_type)
  304. if res_type == 0:
  305. return types
  306. combo_types = [51, 57, 58, 60, 59, 61, 62, 63, 64, 65]
  307. event_type = str(event_type)
  308. len_type = len(event_type)
  309. for i in range(0, len_type):
  310. e_type = int(event_type[len_type - 1 - i])
  311. if e_type == 1:
  312. types.append(combo_types[i])
  313. return types
  314. except Exception as e:
  315. print('推送错误异常,errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  316. return event_type
  317. @staticmethod
  318. def is_combo_tag(event_type, event_tag):
  319. """
  320. 是否为多标签
  321. """
  322. if not event_tag:
  323. return False
  324. res = event_tag.split(',')
  325. types = [int(var) for var in res if var]
  326. if event_type in types and len(types) == 1:
  327. return False
  328. return True
  329. @classmethod
  330. def is_type_exist(cls, event_type):
  331. """
  332. 判断类型是否存在列表
  333. @param event_type:
  334. @return: 0 or event_type
  335. """
  336. combo_types = cls.combo_type_all()
  337. if not combo_types:
  338. return 0
  339. if event_type in combo_types:
  340. return event_type
  341. return 0
  342. @staticmethod
  343. def combo_type_all():
  344. """
  345. 获取所有组合类型
  346. @return:
  347. """
  348. arr_list = []
  349. event_arr = []
  350. resource_list = [1, 2, 4, 8, 16, 32, 64, 128, 256, 512]
  351. for i in range(2, len(resource_list) + 1):
  352. arr_list += list(itertools.combinations(resource_list, i)) # 表示从 [1,2,3,4] 中选出 3个元素的组合情况
  353. for i in arr_list:
  354. val = 0
  355. for item in i:
  356. val += item
  357. event_arr.append(int(EquipmentInfoService.dec_to_bin(val)))
  358. return event_arr
  359. @staticmethod
  360. def dec_to_bin(num):
  361. """
  362. 十进制转二进制
  363. @param num:
  364. @return:
  365. """
  366. result = ""
  367. while num != 0:
  368. ret = num % 2
  369. num //= 2
  370. result = str(ret) + result
  371. return result
  372. @staticmethod
  373. def get_combo_type_bins(event_type):
  374. """
  375. 获取组合类型二进制列表
  376. @param event_type: 标签类型
  377. @return:
  378. """
  379. res_list = []
  380. try:
  381. if ',' in event_type:
  382. res_list = event_type.split(',')
  383. res_list = [int(i.strip()) for i in res_list]
  384. else:
  385. res_list = [int(event_type)]
  386. combo_types = [51, 57, 58, 60, 59, 61]
  387. for e_item in res_list:
  388. bins = EquipmentInfoService.combo_type_all()
  389. if e_item in combo_types:
  390. event_label = combo_types.index(e_item) + 1
  391. for item in bins:
  392. val = str(item)
  393. val_len = len(val)
  394. if val_len >= event_label and int(val[val_len - event_label]) == 1:
  395. res_list.append(int(item))
  396. return res_list
  397. except Exception as e:
  398. print('推送错误异常,errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  399. return res_list
  400. @staticmethod
  401. def get_event_tag(event_type):
  402. """
  403. 获取标签用于筛选推送消息
  404. """
  405. if ',' in event_type:
  406. tags = ''
  407. res_list = event_type.split(',')
  408. tag_size = len(res_list)
  409. for i, item in enumerate(res_list):
  410. tags += ',' + str(item) + ','
  411. if i < (tag_size - 1):
  412. tags += '|'
  413. return tags
  414. else:
  415. return ',' + str(event_type) + ','
  416. @classmethod
  417. def union_equipment_info(cls, user_id, uid_list, event_type, start_time, end_time, page, line):
  418. """
  419. 联表查询推送数据
  420. @param user_id:
  421. @param uid_list:
  422. @param event_type:
  423. @param start_time:
  424. @param end_time:
  425. @param page:
  426. @param line:
  427. @return:
  428. """
  429. try:
  430. kwargs = {
  431. 'device_user_id': user_id,
  432. 'event_time__range': (start_time, end_time),
  433. }
  434. event_type_kwargs = {
  435. 'device_user_id': user_id,
  436. 'event_time__range': (start_time, end_time),
  437. }
  438. if uid_list:
  439. kwargs['device_uid__in'] = uid_list
  440. event_type_kwargs['device_uid__in'] = uid_list
  441. event_type_list = []
  442. if event_type:
  443. # 多类型查询
  444. event_type_list = cls.get_comb_event_type(event_type)
  445. event_type_list = list(set(event_type_list))
  446. tags = cls.get_event_tag(event_type)
  447. if event_type_list:
  448. event_type_kwargs['event_type__in'] = event_type_list
  449. event_type_kwargs['event_tag'] = ''
  450. kwargs['event_tag__regex'] = tags
  451. elif tags:
  452. kwargs['event_tag__regex'] = tags
  453. if not event_type_list:
  454. # 新增tab_val字段记录表的标识:表编号+10
  455. equipment_info_1 = EquipmentInfo1.objects.filter(**kwargs).annotate(tab_val=Value('11', output_field=CharField()))
  456. equipment_info_2 = EquipmentInfo2.objects.filter(**kwargs).annotate(tab_val=Value('12', output_field=CharField()))
  457. equipment_info_3 = EquipmentInfo3.objects.filter(**kwargs).annotate(tab_val=Value('13', output_field=CharField()))
  458. equipment_info_4 = EquipmentInfo4.objects.filter(**kwargs).annotate(tab_val=Value('14', output_field=CharField()))
  459. equipment_info_5 = EquipmentInfo5.objects.filter(**kwargs).annotate(tab_val=Value('15', output_field=CharField()))
  460. equipment_info_6 = EquipmentInfo6.objects.filter(**kwargs).annotate(tab_val=Value('16', output_field=CharField()))
  461. equipment_info_7 = EquipmentInfo7.objects.filter(**kwargs).annotate(tab_val=Value('17', output_field=CharField()))
  462. equipment_info_8 = EquipmentInfo8.objects.filter(**kwargs).annotate(tab_val=Value('18', output_field=CharField()))
  463. equipment_info_9 = EquipmentInfo9.objects.filter(**kwargs).annotate(tab_val=Value('19', output_field=CharField()))
  464. equipment_info_10 = EquipmentInfo10.objects.filter(**kwargs).annotate(tab_val=Value('20', output_field=CharField()))
  465. equipment_info_11 = EquipmentInfo11.objects.filter(**kwargs).annotate(tab_val=Value('21', output_field=CharField()))
  466. equipment_info_12 = EquipmentInfo12.objects.filter(**kwargs).annotate(tab_val=Value('22', output_field=CharField()))
  467. equipment_info_13 = EquipmentInfo13.objects.filter(**kwargs).annotate(tab_val=Value('23', output_field=CharField()))
  468. equipment_info_14 = EquipmentInfo14.objects.filter(**kwargs).annotate(tab_val=Value('24', output_field=CharField()))
  469. equipment_info_15 = EquipmentInfo15.objects.filter(**kwargs).annotate(tab_val=Value('25', output_field=CharField()))
  470. equipment_info_16 = EquipmentInfo16.objects.filter(**kwargs).annotate(tab_val=Value('26', output_field=CharField()))
  471. equipment_info_17 = EquipmentInfo17.objects.filter(**kwargs).annotate(tab_val=Value('27', output_field=CharField()))
  472. equipment_info_18 = EquipmentInfo18.objects.filter(**kwargs).annotate(tab_val=Value('28', output_field=CharField()))
  473. equipment_info_19 = EquipmentInfo19.objects.filter(**kwargs).annotate(tab_val=Value('29', output_field=CharField()))
  474. equipment_info_20 = EquipmentInfo20.objects.filter(**kwargs).annotate(tab_val=Value('30', output_field=CharField()))
  475. else:
  476. equipment_info_1 = EquipmentInfo1.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('11', output_field=CharField()))
  477. equipment_info_2 = EquipmentInfo2.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('12', output_field=CharField()))
  478. equipment_info_3 = EquipmentInfo3.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('13', output_field=CharField()))
  479. equipment_info_4 = EquipmentInfo4.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('14', output_field=CharField()))
  480. equipment_info_5 = EquipmentInfo5.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('15', output_field=CharField()))
  481. equipment_info_6 = EquipmentInfo6.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('16', output_field=CharField()))
  482. equipment_info_7 = EquipmentInfo7.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('17', output_field=CharField()))
  483. equipment_info_8 = EquipmentInfo8.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('18', output_field=CharField()))
  484. equipment_info_9 = EquipmentInfo9.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('19', output_field=CharField()))
  485. equipment_info_10 = EquipmentInfo10.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('20', output_field=CharField()))
  486. equipment_info_11 = EquipmentInfo11.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('21', output_field=CharField()))
  487. equipment_info_12 = EquipmentInfo12.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('22', output_field=CharField()))
  488. equipment_info_13 = EquipmentInfo13.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('23', output_field=CharField()))
  489. equipment_info_14 = EquipmentInfo14.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('24', output_field=CharField()))
  490. equipment_info_15 = EquipmentInfo15.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('25', output_field=CharField()))
  491. equipment_info_16 = EquipmentInfo16.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('26', output_field=CharField()))
  492. equipment_info_17 = EquipmentInfo17.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('27', output_field=CharField()))
  493. equipment_info_18 = EquipmentInfo18.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('28', output_field=CharField()))
  494. equipment_info_19 = EquipmentInfo19.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('29', output_field=CharField()))
  495. equipment_info_20 = EquipmentInfo20.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(tab_val=Value('30', output_field=CharField()))
  496. equipment_info_union = equipment_info_1.union(
  497. equipment_info_2, equipment_info_3, equipment_info_4, equipment_info_5, equipment_info_6, equipment_info_7,
  498. equipment_info_8, equipment_info_9, equipment_info_10,
  499. equipment_info_11, equipment_info_12, equipment_info_13, equipment_info_14, equipment_info_15,
  500. equipment_info_16, equipment_info_17, equipment_info_18, equipment_info_19, equipment_info_20)
  501. count = equipment_info_union.count()
  502. if count == 0:
  503. return None, 0
  504. equipment_info_union = cls.get_equipment_info_union_page(equipment_info_union, page, line)
  505. return equipment_info_union, count
  506. except Exception as e:
  507. print('error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  508. @staticmethod
  509. def get_equipment_info_union_page(equipment_info_union, page, line):
  510. """
  511. 获取查询结果集进行排序、分页,遍历重命名字典key(主要针对原函数返回结果集)
  512. @param equipment_info_union: 设备信息结果集
  513. @param page: 页数
  514. @param line: 分页大小
  515. @return: qs_page 遍历后的设备信息结果集
  516. """
  517. equipment_info_qs = equipment_info_union.values('id', 'device_uid', 'device_nick_name', 'channel', 'event_type',
  518. 'status', 'answer_status', 'alarm', 'event_time', 'is_st',
  519. 'add_time', 'storage_location', 'border_coords', 'tab_val',
  520. 'event_tag')
  521. equipment_info_qs = equipment_info_qs.order_by('-event_time')
  522. equipment_info_qs = equipment_info_qs[(page - 1) * line:page * line]
  523. for equipment_info in equipment_info_qs:
  524. # id: 表标识+id
  525. equipment_info['id'] = equipment_info['tab_val'] + str(equipment_info['id'])
  526. equipment_info['devUid'] = equipment_info['device_uid']
  527. equipment_info['devNickName'] = equipment_info['device_nick_name']
  528. equipment_info['Channel'] = equipment_info['channel']
  529. equipment_info['eventType'] = equipment_info['event_type']
  530. equipment_info['eventTime'] = equipment_info['event_time']
  531. equipment_info['receiveTime'] = equipment_info['event_time']
  532. equipment_info['addTime'] = equipment_info['add_time']
  533. equipment_info['borderCoords'] = equipment_info['border_coords']
  534. equipment_info['eventTag'] = equipment_info['event_tag']
  535. equipment_info.pop('device_uid')
  536. equipment_info.pop('device_nick_name')
  537. equipment_info.pop('channel')
  538. equipment_info.pop('event_type')
  539. equipment_info.pop('event_time')
  540. equipment_info.pop('add_time')
  541. equipment_info.pop('border_coords')
  542. equipment_info.pop('tab_val')
  543. equipment_info.pop('event_tag')
  544. return equipment_info_qs
  545. @staticmethod
  546. def all_read_equipment_info(**kwargs):
  547. """
  548. 全部已读推送消息
  549. @param kwargs:
  550. @return:
  551. """
  552. for equipment_info_model in EQUIPMENT_INFO_MODEL_LIST:
  553. equipment_info_model.objects.filter(**kwargs).update(status=1)
  554. @staticmethod
  555. def update_equipment_info_answer_status(**kwargs):
  556. """
  557. 已接听一键通话消息
  558. @param kwargs:
  559. @return:
  560. """
  561. for equipment_info_model in EQUIPMENT_INFO_MODEL_LIST:
  562. equipment_info_model.objects.filter(**kwargs).update(answer_status=1)
  563. @staticmethod
  564. def delete_all_equipment_info(**kwargs):
  565. """
  566. 删除全部推送消息
  567. @param kwargs:
  568. @return:
  569. """
  570. for equipment_info_model in EQUIPMENT_INFO_MODEL_LIST:
  571. equipment_info_model.objects.filter(**kwargs).delete()
  572. @staticmethod
  573. def get_equipment_info_model_with_full_id(full_id):
  574. """
  575. 根据拼接id获取推送表
  576. @param full_id: 拼接id,表标识+id
  577. @return:
  578. """
  579. tab_val = int(full_id[:2]) - 10
  580. if tab_val > 20:
  581. return None
  582. tab_index = tab_val - 1
  583. return EQUIPMENT_INFO_MODEL_LIST[tab_index]
  584. @staticmethod
  585. def count_equipment_info(**kwargs):
  586. """
  587. 查询推送数据数量
  588. @param kwargs:
  589. @return: count
  590. """
  591. count = 0
  592. for equipment_info_model in EQUIPMENT_INFO_MODEL_LIST:
  593. count += equipment_info_model.objects.filter(**kwargs).count()
  594. return count