EquipmentInfoService.py 31 KB

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