123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599 |
- # -*- encoding: utf-8 -*-
- """
- @File : EquipmentInfoService.py
- @Time : 2022/4/14 17:28
- @Author : stephen
- @Email : zhangdongming@asj6.wecom.work
- @Software: PyCharm
- """
- import itertools
- from Ansjer.config import CONFIG_INFO, CONFIG_EUR
- from django.db.models import Value, CharField, Q
- from Model.models import EquipmentInfo1, EquipmentInfo2, EquipmentInfo3, EquipmentInfo4, EquipmentInfo5, \
- EquipmentInfo6, EquipmentInfo7, EquipmentInfo8, EquipmentInfo9, EquipmentInfo10, \
- EquipmentInfo11, EquipmentInfo12, EquipmentInfo13, EquipmentInfo14, EquipmentInfo15, EquipmentInfo16, \
- EquipmentInfo17, EquipmentInfo18, EquipmentInfo19, EquipmentInfo20, \
- EquipmentInfo21, EquipmentInfo22, EquipmentInfo23, EquipmentInfo24, EquipmentInfo25, EquipmentInfo26, \
- EquipmentInfo27, EquipmentInfo28, EquipmentInfo29, EquipmentInfo30, \
- EquipmentInfo31, EquipmentInfo32, EquipmentInfo33, EquipmentInfo34, EquipmentInfo35, EquipmentInfo36, \
- EquipmentInfo37, EquipmentInfo38, EquipmentInfo39, EquipmentInfo40
- from Object.RedisObject import RedisObject
- EQUIPMENT_INFO_MODEL_LIST = [
- EquipmentInfo1, EquipmentInfo2, EquipmentInfo3, EquipmentInfo4, EquipmentInfo5, EquipmentInfo6,
- EquipmentInfo7, EquipmentInfo8, EquipmentInfo9, EquipmentInfo10,
- EquipmentInfo11, EquipmentInfo12, EquipmentInfo13, EquipmentInfo14, EquipmentInfo15, EquipmentInfo16,
- EquipmentInfo17, EquipmentInfo18, EquipmentInfo19, EquipmentInfo20,
- EquipmentInfo21, EquipmentInfo22, EquipmentInfo23, EquipmentInfo24, EquipmentInfo25, EquipmentInfo26,
- EquipmentInfo27, EquipmentInfo28, EquipmentInfo29, EquipmentInfo30,
- EquipmentInfo31, EquipmentInfo32, EquipmentInfo33, EquipmentInfo34, EquipmentInfo35, EquipmentInfo36,
- EquipmentInfo37, EquipmentInfo38, EquipmentInfo39, EquipmentInfo40
- ]
- class EquipmentInfoService:
- """
- 推送数据分表,封装常用函数
- """
- @classmethod
- def get_comb_event_type(cls, event_type):
- """
- 重新组合ai消息类型查询,使其支持ai多标签查询
- @param event_type: 消息类型
- @return: event_type_list 消息类型数组
- """
- if ',' in event_type:
- event_type_list = event_type.split(',')
- event_type_list = [int(i.strip()) for i in event_type_list]
- else:
- event_type_list = [int(event_type)]
- ai_event_type_list = []
- normal_event_type_list = []
- a_type = [60, 61, 62, 63, 64, 65, 66]
- for val in event_type_list:
- if val <= 4: # 分离出ai类型,以便后续组合ai标签,目前只存在4个ai类型1,2,3,4
- ai_event_type_list.append(val)
- else:
- if val not in a_type:
- normal_event_type_list.append(val)
- if len(ai_event_type_list) < 1:
- return normal_event_type_list
- ai_event_type_list.sort()
- ai_type = [1, 2, 3, 4] # AI目前所有的标签,1人,2车,3宠物,4包裹,后续有新类型需要这里加, 后续会优化,存在表里,包裹存对应的aws标签
- comb_ai_event_type = []
- seen = set()
- for i in range(1, len(ai_type) + 1): # 计算所有组合,如[1, 2, 3, 4], 4取1,4取2,4取3,4取4
- for s in itertools.combinations(ai_type, i):
- if s not in seen: # 去除重复项, 如a=[1,2,3,4,4],会有两个[1,2,3,4,4],[1,2,3,4,4]的组合
- seen.add(s)
- s_list = list(s)
- for ai_event_type in ai_event_type_list:
- if ai_event_type in s_list: # 排除没有选择的标签组合
- if s_list not in comb_ai_event_type:
- s_list = [str(v) for v in s_list]
- comb_ai_event_type.append(s_list)
- regroup_list = []
- for val in comb_ai_event_type: # 组合ai类型组合,如[[2,3],[1,3]] -> [23, 13]
- val = ''.join(val)
- regroup_list.append(int(val))
- group_list = regroup_list + normal_event_type_list # 加上普通移动消息类型
- return group_list
- @classmethod
- def get_all_comb_event_type(cls):
- """
- 计算ai消息类型全组合
- @return: event_type_list ai所有消息类型数组
- """
- ai_type = [1, 2, 3, 4] # AI目前所有的标签,1人,2车,3宠物,4包裹,后续有新类型需要这里加, 后续会优化,存在表里,包裹存对应的aws标签
- comb_ai_event_type = []
- for i in range(1, len(ai_type) + 1): # 计算所有组合,如[1, 2, 3, 4], 4取1,4取2,4取3,4取4
- for s in itertools.combinations(ai_type, i):
- s_list = list(s)
- s_list = [str(v) for v in s_list]
- comb_ai_event_type.append(s_list)
- regroup_list = []
- for val in comb_ai_event_type: # 组合ai类型组合,如[[2,3],[1,3]] -> [23, 13]
- val = ''.join(val)
- regroup_list.append(int(val))
- return regroup_list
- @classmethod
- def get_combo_types(cls, event_type, event_tag):
- """
- 获取设备算法组合类型
- 51:移动侦测,52:传感器报警,53:影像遗失,54:PIR,55:门磁报警,56:外部发报,57:人型报警(提示:有人出现),58:车型,59:宠物,60:人脸,61:异响,
- 62:区域闯入,63:区域闯出,64:长时间无人检测,65:长时间无人检测
- 0:代表空字符,702:摄像头休眠,703:摄像头唤醒,704:电量过低
- AWS AI识别 1:人形,2:车型,3:宠物,4:包裹。云端AI类型
- @param event_tag:
- @param event_type:
- @return:
- """
- try:
- types = []
- if event_tag:
- res = event_tag.split(',')
- types = [int(var) for var in res if var]
- return types
- res_type = cls.is_type_exist(event_type)
- if res_type == 0:
- return types
- combo_types = [51, 57, 58, 60, 59, 61, 62, 63, 64, 65]
- event_type = str(event_type)
- len_type = len(event_type)
- for i in range(0, len_type):
- e_type = int(event_type[len_type - 1 - i])
- if e_type == 1:
- types.append(combo_types[i])
- return types
- except Exception as e:
- print('推送错误异常,errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
- return event_type
- @staticmethod
- def is_combo_tag(event_type, event_tag):
- """
- 是否为多标签
- """
- if not event_tag:
- return False
- res = event_tag.split(',')
- types = [int(var) for var in res if var]
- if event_type in types and len(types) == 1:
- return False
- return True
- @classmethod
- def is_type_exist(cls, event_type):
- """
- 判断类型是否存在列表
- @param event_type:
- @return: 0 or event_type
- """
- combo_types = cls.combo_type_all()
- if not combo_types:
- return 0
- if event_type in combo_types:
- return event_type
- return 0
- @staticmethod
- def combo_type_all():
- """
- 获取所有组合类型
- @return:
- """
- arr_list = []
- event_arr = []
- resource_list = [1, 2, 4, 8, 16, 32, 64, 128, 256, 512]
- for i in range(2, len(resource_list) + 1):
- arr_list += list(itertools.combinations(resource_list, i)) # 表示从 [1,2,3,4] 中选出 3个元素的组合情况
- for i in arr_list:
- val = 0
- for item in i:
- val += item
- event_arr.append(int(EquipmentInfoService.dec_to_bin(val)))
- return event_arr
- @staticmethod
- def dec_to_bin(num):
- """
- 十进制转二进制
- @param num:
- @return:
- """
- result = ""
- while num != 0:
- ret = num % 2
- num //= 2
- result = str(ret) + result
- return result
- @staticmethod
- def get_combo_type_bins(event_type):
- """
- 获取组合类型二进制列表
- @param event_type: 标签类型
- @return:
- """
- res_list = []
- try:
- if ',' in event_type:
- res_list = event_type.split(',')
- res_list = [int(i.strip()) for i in res_list]
- else:
- res_list = [int(event_type)]
- combo_types = [51, 57, 58, 60, 59, 61]
- for e_item in res_list:
- bins = EquipmentInfoService.combo_type_all()
- if e_item in combo_types:
- event_label = combo_types.index(e_item) + 1
- for item in bins:
- val = str(item)
- val_len = len(val)
- if val_len >= event_label and int(val[val_len - event_label]) == 1:
- res_list.append(int(item))
- return res_list
- except Exception as e:
- print('推送错误异常,errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
- return res_list
- @staticmethod
- def get_event_tag(event_type):
- """
- 获取标签用于筛选推送消息
- """
- if ',' in event_type:
- tags = ''
- res_list = event_type.split(',')
- tag_size = len(res_list)
- for i, item in enumerate(res_list):
- tags += ',' + str(item) + ','
- if i < (tag_size - 1):
- tags += '|'
- return tags
- else:
- return ',' + str(event_type) + ','
- @classmethod
- def union_equipment_info(cls, user_id, uid_list, event_type, start_time, end_time, page, line, event_time=None, channel=None):
- """
- 联表查询推送数据
- @param user_id:
- @param uid_list:
- @param event_type:
- @param start_time:
- @param end_time:
- @param page:
- @param line:
- @param event_time:
- @return:
- """
- try:
- kwargs = {
- 'device_user_id': user_id,
- 'event_time__range': (start_time, end_time),
- }
- event_type_kwargs = {
- 'device_user_id': user_id,
- 'event_time__range': (start_time, end_time),
- }
- if uid_list:
- kwargs['device_uid__in'] = uid_list
- event_type_kwargs['device_uid__in'] = uid_list
- event_type_list = []
- if event_type:
- # 多类型查询
- event_type_list = cls.get_comb_event_type(event_type)
- event_type_list = list(set(event_type_list))
- tags = cls.get_event_tag(event_type)
- if event_type_list:
- event_type_kwargs['event_type__in'] = event_type_list
- event_type_kwargs['event_tag'] = ''
- kwargs['event_tag__regex'] = tags
- elif tags:
- kwargs['event_tag__regex'] = tags
- if event_time:
- kwargs['event_time'] = event_time
- event_type_kwargs['event_time'] = event_time
- del kwargs['event_tag__regex']
- del event_type_kwargs['event_tag']
- if channel:
- kwargs['channel'] = channel
- event_type_kwargs['channel'] = channel
- if not event_type_list:
- # 新增tab_val字段记录表的标识:表编号+10
- equipment_info_1 = EquipmentInfo1.objects.filter(**kwargs).annotate(
- tab_val=Value('11', output_field=CharField()))
- equipment_info_2 = EquipmentInfo2.objects.filter(**kwargs).annotate(
- tab_val=Value('12', output_field=CharField()))
- equipment_info_3 = EquipmentInfo3.objects.filter(**kwargs).annotate(
- tab_val=Value('13', output_field=CharField()))
- equipment_info_4 = EquipmentInfo4.objects.filter(**kwargs).annotate(
- tab_val=Value('14', output_field=CharField()))
- equipment_info_5 = EquipmentInfo5.objects.filter(**kwargs).annotate(
- tab_val=Value('15', output_field=CharField()))
- equipment_info_6 = EquipmentInfo6.objects.filter(**kwargs).annotate(
- tab_val=Value('16', output_field=CharField()))
- equipment_info_7 = EquipmentInfo7.objects.filter(**kwargs).annotate(
- tab_val=Value('17', output_field=CharField()))
- equipment_info_8 = EquipmentInfo8.objects.filter(**kwargs).annotate(
- tab_val=Value('18', output_field=CharField()))
- equipment_info_9 = EquipmentInfo9.objects.filter(**kwargs).annotate(
- tab_val=Value('19', output_field=CharField()))
- equipment_info_10 = EquipmentInfo10.objects.filter(**kwargs).annotate(
- tab_val=Value('20', output_field=CharField()))
- equipment_info_11 = EquipmentInfo11.objects.filter(**kwargs).annotate(
- tab_val=Value('21', output_field=CharField()))
- equipment_info_12 = EquipmentInfo12.objects.filter(**kwargs).annotate(
- tab_val=Value('22', output_field=CharField()))
- equipment_info_13 = EquipmentInfo13.objects.filter(**kwargs).annotate(
- tab_val=Value('23', output_field=CharField()))
- equipment_info_14 = EquipmentInfo14.objects.filter(**kwargs).annotate(
- tab_val=Value('24', output_field=CharField()))
- equipment_info_15 = EquipmentInfo15.objects.filter(**kwargs).annotate(
- tab_val=Value('25', output_field=CharField()))
- equipment_info_16 = EquipmentInfo16.objects.filter(**kwargs).annotate(
- tab_val=Value('26', output_field=CharField()))
- equipment_info_17 = EquipmentInfo17.objects.filter(**kwargs).annotate(
- tab_val=Value('27', output_field=CharField()))
- equipment_info_18 = EquipmentInfo18.objects.filter(**kwargs).annotate(
- tab_val=Value('28', output_field=CharField()))
- equipment_info_19 = EquipmentInfo19.objects.filter(**kwargs).annotate(
- tab_val=Value('29', output_field=CharField()))
- equipment_info_20 = EquipmentInfo20.objects.filter(**kwargs).annotate(
- tab_val=Value('30', output_field=CharField()))
- equipment_info_21 = EquipmentInfo21.objects.filter(**kwargs).annotate(
- tab_val=Value('31', output_field=CharField()))
- equipment_info_22 = EquipmentInfo22.objects.filter(**kwargs).annotate(
- tab_val=Value('32', output_field=CharField()))
- equipment_info_23 = EquipmentInfo23.objects.filter(**kwargs).annotate(
- tab_val=Value('33', output_field=CharField()))
- equipment_info_24 = EquipmentInfo24.objects.filter(**kwargs).annotate(
- tab_val=Value('34', output_field=CharField()))
- equipment_info_25 = EquipmentInfo25.objects.filter(**kwargs).annotate(
- tab_val=Value('35', output_field=CharField()))
- equipment_info_26 = EquipmentInfo26.objects.filter(**kwargs).annotate(
- tab_val=Value('36', output_field=CharField()))
- equipment_info_27 = EquipmentInfo27.objects.filter(**kwargs).annotate(
- tab_val=Value('37', output_field=CharField()))
- equipment_info_28 = EquipmentInfo28.objects.filter(**kwargs).annotate(
- tab_val=Value('38', output_field=CharField()))
- equipment_info_29 = EquipmentInfo29.objects.filter(**kwargs).annotate(
- tab_val=Value('39', output_field=CharField()))
- equipment_info_30 = EquipmentInfo30.objects.filter(**kwargs).annotate(
- tab_val=Value('40', output_field=CharField()))
- equipment_info_31 = EquipmentInfo31.objects.filter(**kwargs).annotate(
- tab_val=Value('41', output_field=CharField()))
- equipment_info_32 = EquipmentInfo32.objects.filter(**kwargs).annotate(
- tab_val=Value('42', output_field=CharField()))
- equipment_info_33 = EquipmentInfo33.objects.filter(**kwargs).annotate(
- tab_val=Value('43', output_field=CharField()))
- equipment_info_34 = EquipmentInfo34.objects.filter(**kwargs).annotate(
- tab_val=Value('44', output_field=CharField()))
- equipment_info_35 = EquipmentInfo35.objects.filter(**kwargs).annotate(
- tab_val=Value('45', output_field=CharField()))
- equipment_info_36 = EquipmentInfo36.objects.filter(**kwargs).annotate(
- tab_val=Value('46', output_field=CharField()))
- equipment_info_37 = EquipmentInfo37.objects.filter(**kwargs).annotate(
- tab_val=Value('47', output_field=CharField()))
- equipment_info_38 = EquipmentInfo38.objects.filter(**kwargs).annotate(
- tab_val=Value('48', output_field=CharField()))
- equipment_info_39 = EquipmentInfo39.objects.filter(**kwargs).annotate(
- tab_val=Value('49', output_field=CharField()))
- equipment_info_40 = EquipmentInfo40.objects.filter(**kwargs).annotate(
- tab_val=Value('50', output_field=CharField()))
- else:
- equipment_info_1 = EquipmentInfo1.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('11', output_field=CharField()))
- equipment_info_2 = EquipmentInfo2.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('12', output_field=CharField()))
- equipment_info_3 = EquipmentInfo3.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('13', output_field=CharField()))
- equipment_info_4 = EquipmentInfo4.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('14', output_field=CharField()))
- equipment_info_5 = EquipmentInfo5.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('15', output_field=CharField()))
- equipment_info_6 = EquipmentInfo6.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('16', output_field=CharField()))
- equipment_info_7 = EquipmentInfo7.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('17', output_field=CharField()))
- equipment_info_8 = EquipmentInfo8.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('18', output_field=CharField()))
- equipment_info_9 = EquipmentInfo9.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('19', output_field=CharField()))
- equipment_info_10 = EquipmentInfo10.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('20', output_field=CharField()))
- equipment_info_11 = EquipmentInfo11.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('21', output_field=CharField()))
- equipment_info_12 = EquipmentInfo12.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('22', output_field=CharField()))
- equipment_info_13 = EquipmentInfo13.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('23', output_field=CharField()))
- equipment_info_14 = EquipmentInfo14.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('24', output_field=CharField()))
- equipment_info_15 = EquipmentInfo15.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('25', output_field=CharField()))
- equipment_info_16 = EquipmentInfo16.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('26', output_field=CharField()))
- equipment_info_17 = EquipmentInfo17.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('27', output_field=CharField()))
- equipment_info_18 = EquipmentInfo18.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('28', output_field=CharField()))
- equipment_info_19 = EquipmentInfo19.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('29', output_field=CharField()))
- equipment_info_20 = EquipmentInfo20.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('30', output_field=CharField()))
- equipment_info_21 = EquipmentInfo21.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('31', output_field=CharField()))
- equipment_info_22 = EquipmentInfo22.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('32', output_field=CharField()))
- equipment_info_23 = EquipmentInfo23.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('33', output_field=CharField()))
- equipment_info_24 = EquipmentInfo24.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('34', output_field=CharField()))
- equipment_info_25 = EquipmentInfo25.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('35', output_field=CharField()))
- equipment_info_26 = EquipmentInfo26.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('36', output_field=CharField()))
- equipment_info_27 = EquipmentInfo27.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('37', output_field=CharField()))
- equipment_info_28 = EquipmentInfo28.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('38', output_field=CharField()))
- equipment_info_29 = EquipmentInfo29.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('39', output_field=CharField()))
- equipment_info_30 = EquipmentInfo30.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('40', output_field=CharField()))
- equipment_info_31 = EquipmentInfo31.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('41', output_field=CharField()))
- equipment_info_32 = EquipmentInfo32.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('42', output_field=CharField()))
- equipment_info_33 = EquipmentInfo33.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('43', output_field=CharField()))
- equipment_info_34 = EquipmentInfo34.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('44', output_field=CharField()))
- equipment_info_35 = EquipmentInfo35.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('45', output_field=CharField()))
- equipment_info_36 = EquipmentInfo36.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('46', output_field=CharField()))
- equipment_info_37 = EquipmentInfo37.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('47', output_field=CharField()))
- equipment_info_38 = EquipmentInfo38.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('48', output_field=CharField()))
- equipment_info_39 = EquipmentInfo39.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('49', output_field=CharField()))
- equipment_info_40 = EquipmentInfo40.objects.filter(Q(**kwargs) | Q(**event_type_kwargs)).annotate(
- tab_val=Value('50', output_field=CharField()))
- equipment_info_union = equipment_info_1.union(
- equipment_info_2, equipment_info_3, equipment_info_4, equipment_info_5, equipment_info_6,
- equipment_info_7, equipment_info_8, equipment_info_9, equipment_info_10,
- equipment_info_11, equipment_info_12, equipment_info_13, equipment_info_14, equipment_info_15,
- equipment_info_16, equipment_info_17, equipment_info_18, equipment_info_19, equipment_info_20,
- equipment_info_21, equipment_info_22, equipment_info_23, equipment_info_24, equipment_info_25,
- equipment_info_26, equipment_info_27, equipment_info_28, equipment_info_29, equipment_info_30,
- equipment_info_31, equipment_info_32, equipment_info_33, equipment_info_34, equipment_info_35,
- equipment_info_36, equipment_info_37, equipment_info_38, equipment_info_39, equipment_info_40,
- all=True
- )
- if not equipment_info_union.exists():
- return None, 0
- if CONFIG_EUR == CONFIG_INFO:
- line = 10
- count = 21
- equipment_info_union = cls.get_equipment_info_union_page(equipment_info_union, page, line)
- return equipment_info_union, count
- except Exception as e:
- print('error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
- @staticmethod
- def get_equipment_info_union_page(equipment_info_union, page, line):
- """
- 获取查询结果集进行排序、分页,遍历重命名字典key(主要针对原函数返回结果集)
- @param equipment_info_union: 设备信息结果集
- @param page: 页数
- @param line: 分页大小
- @return: qs_page 遍历后的设备信息结果集
- """
- equipment_info_qs = equipment_info_union.values('id', 'device_uid', 'device_nick_name', 'channel', 'event_type',
- 'status', 'answer_status', 'alarm', 'event_time', 'is_st',
- 'add_time', 'storage_location', 'border_coords', 'tab_val',
- 'event_tag')
- equipment_info_qs = equipment_info_qs.order_by('-event_time')
- equipment_info_qs = equipment_info_qs[(page - 1) * line:page * line]
- for equipment_info in equipment_info_qs:
- # id: 表标识+id
- equipment_info['id'] = equipment_info['tab_val'] + str(equipment_info['id'])
- equipment_info['devUid'] = equipment_info['device_uid']
- equipment_info['devNickName'] = equipment_info['device_nick_name']
- equipment_info['Channel'] = equipment_info['channel']
- equipment_info['eventType'] = equipment_info['event_type']
- equipment_info['eventTime'] = equipment_info['event_time']
- equipment_info['receiveTime'] = equipment_info['event_time']
- equipment_info['addTime'] = equipment_info['add_time']
- equipment_info['borderCoords'] = equipment_info['border_coords']
- equipment_info['eventTag'] = equipment_info['event_tag']
- if equipment_info['alarm'] == '':
- equipment_info['alarm'] = 'Motion \tChannel:{}'.format(equipment_info['channel'])
- equipment_info.pop('device_uid')
- equipment_info.pop('device_nick_name')
- equipment_info.pop('channel')
- equipment_info.pop('event_type')
- equipment_info.pop('event_time')
- equipment_info.pop('add_time')
- equipment_info.pop('border_coords')
- equipment_info.pop('tab_val')
- equipment_info.pop('event_tag')
- return equipment_info_qs
- @staticmethod
- def all_read_equipment_info(**kwargs):
- """
- 全部已读推送消息
- @param kwargs:
- @return:
- """
- for equipment_info_model in EQUIPMENT_INFO_MODEL_LIST:
- equipment_info_model.objects.filter(**kwargs).update(status=1)
- @staticmethod
- def update_equipment_info_answer_status(**kwargs):
- """
- 已接听一键通话消息
- @param kwargs:
- @return:
- """
- for equipment_info_model in EQUIPMENT_INFO_MODEL_LIST:
- equipment_info_model.objects.filter(**kwargs).update(answer_status=1)
- @staticmethod
- def delete_all_equipment_info(**kwargs):
- """
- 删除全部推送消息
- @param kwargs:
- @return:
- """
- for equipment_info_model in EQUIPMENT_INFO_MODEL_LIST:
- equipment_info_model.objects.filter(**kwargs).delete()
- @staticmethod
- def get_equipment_info_model_with_full_id(full_id):
- """
- 根据拼接id获取推送表
- @param full_id: 拼接id,表标识+id
- @return:
- """
- tab_val = int(full_id[:2]) - 10
- if tab_val > len(EQUIPMENT_INFO_MODEL_LIST):
- return None
- tab_index = tab_val - 1
- return EQUIPMENT_INFO_MODEL_LIST[tab_index]
- @staticmethod
- def count_equipment_info(**kwargs):
- """
- 查询推送数据数量
- @param kwargs:
- @return: count
- """
- redis_obj = RedisObject()
- key = 'PUSH:MESSAGE:USER:{}'.format(kwargs['device_user_id'])
- push_data = redis_obj.get_data(key)
- if push_data:
- return int(push_data)
- count = 0
- for equipment_info_model in EQUIPMENT_INFO_MODEL_LIST:
- count += equipment_info_model.objects.filter(**kwargs).count()
- if count == 0:
- return count
- if CONFIG_EUR == CONFIG_INFO and count > 99:
- redis_obj.set_data(key, count, 300)
- else:
- redis_obj.set_data(key, count, 60)
- return count
- @classmethod
- def get_equipment_info(cls, **kwargs):
- """
- 查询推送数据
- @param kwargs:
- @return: qs
- """
- equipment_info_union = EQUIPMENT_INFO_MODEL_LIST[0].objects.filter(**kwargs).annotate(
- tab_val=Value('11', output_field=CharField()))
- for index, equipment_info_model in enumerate(EQUIPMENT_INFO_MODEL_LIST[1:]):
- tab_val = str(index + 2)
- equipment_info_union = equipment_info_union.union(equipment_info_model.objects.filter(**kwargs).annotate(
- tab_val=Value(tab_val, output_field=CharField())))
- equipment_info_union = cls.get_equipment_info_union_page(equipment_info_union, 1, 10)
- count = equipment_info_union.count()
- return equipment_info_union, count
|