EquipmentInfoService.py 30 KB

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