# -*- encoding: utf-8 -*- """ @File : EquipmentInfoService.py @Time : 2022/4/14 17:28 @Author : stephen @Email : zhangdongming@asj6.wecom.work @Software: PyCharm """ import itertools 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 EQUIPMENT_INFO_MODEL_LIST = [EquipmentInfo1, EquipmentInfo2, EquipmentInfo3, EquipmentInfo4, EquipmentInfo5, EquipmentInfo6, EquipmentInfo7, EquipmentInfo8, EquipmentInfo9, EquipmentInfo10, EquipmentInfo11, EquipmentInfo12, EquipmentInfo13, EquipmentInfo14, EquipmentInfo15, EquipmentInfo16, EquipmentInfo17, EquipmentInfo18, EquipmentInfo19, EquipmentInfo20] 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): """ 联表查询推送数据 @param user_id: @param uid_list: @param event_type: @param start_time: @param end_time: @param page: @param line: @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 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())) 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_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) count = equipment_info_union.count() if count == 0: return None, 0 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'] 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 > 20: return None tab_index = tab_val - 1 return EQUIPMENT_INFO_MODEL_LIST[tab_index] @staticmethod def count_equipment_info(**kwargs): """ 查询推送数据数量 @param kwargs: @return: count """ count = 0 for equipment_info_model in EQUIPMENT_INFO_MODEL_LIST: count += equipment_info_model.objects.filter(**kwargs).count() return count