import time import boto3 import botocore import oss2 from botocore import client from django.core.paginator import Paginator from django.db.models import Q from django.views.generic.base import View from Ansjer.config import OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET, AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, \ SERVER_TYPE from Model.models import Device_Info, VodBucketModel, DeviceTypeModel from Object.ResponseObject import ResponseObject from Object.TokenObject import TokenObject from Model import models from Object.utils import LocalDateTimeUtil from Service.EquipmentInfoService import EquipmentInfoService from Service.VodHlsService import SplitVodHlsObject class MassageView(View): def get(self, request, *args, **kwargs): request.encoding = 'utf-8' operation = kwargs.get('operation') return self.validation(request, request.GET, operation) def post(self, request, *args, **kwargs): request.encoding = 'utf-8' operation = kwargs.get('operation') return self.validation(request, request.POST, operation) def validation(self, request, request_dict, operation): language = request_dict.get('language', 'en') response = ResponseObject(language, 'pc') if operation == 'XXX': return 0 else: tko = TokenObject(request.META.get('HTTP_AUTHORIZATION'), returntpye='pc') if tko.code != 0: return response.json(tko.code) if operation == 'queryInfoList': return self.query_info_list(request_dict, response) else: return response.json(414) def query_info_list(self, request_dict, response): """ 查询推送数据 @param request_dict: 请求参数 @request_dict uids: 设备id @request_dict eventType: 事件类型 @request_dict eventType: userID @request_dict startTime: 开始时间戳 @request_dict endTime: 结束时间戳 @request_dict pageNo: 开始时间戳 @request_dict pageSize: 结束时间戳 @param response: @return: """ uids = request_dict.get('uids', None) event_type = request_dict.get('eventType', None) user_id = request_dict.get('userID', None) # 时间 start_time = request_dict.get('startTime', None) end_time = request_dict.get('endTime', None) # 分页 page = int(request_dict.get('pageNo', None)) size = int(request_dict.get('pageSize', None)) # 区域 if SERVER_TYPE == 'Ansjer.cn_config.formal_settings' or SERVER_TYPE == 'Ansjer.cn_config.test_settings': region = 2 else: region = 1 query = Q() if not start_time and not end_time: # 默认查询近七天内数据 end_time = int(time.time()) start_time = LocalDateTimeUtil.get_before_days_timestamp(end_time, 7) query &= Q(event_time__range=(start_time, end_time)) elif start_time and end_time: query &= Q(event_time__range=(start_time, end_time)) else: response.json(10, "需要给出一个时间段") if user_id is None and uids is None: return response.json(0, {"list": [], "total": 0}) # 过滤条件 if user_id is not None and user_id != "": query &= Q(device_user_id=user_id) if uids is not None and uids != "": uid_list = uids.split(',') query &= Q(device_uid__in=uid_list) if event_type is not None: event_type_list = EquipmentInfoService.get_comb_event_type(event_type) event_type_list = list(set(event_type_list)) tags = EquipmentInfoService.get_event_tag(event_type) if event_type_list: query &= Q(event_type__in=event_type_list) tags = '' query &= Q(event_tag__regex=tags) elif tags: query &= Q(event_tag__regex=tags) # 联表查询 querysets = [] for i in range(1, 21): table_name = f'EquipmentInfo{i}' model_class = getattr(models, table_name) annotated_queryset = model_class.objects.filter(query) querysets.append(annotated_queryset) equipment_info_combined_qs = querysets[0].union(*querysets[1:], all=True) # 创建分页对象 equipment_info_combined_qs = equipment_info_combined_qs.order_by("-event_time") paginator = Paginator(equipment_info_combined_qs, size) # 获取请求页的数据 packages_page = paginator.page(page) # 连接存储桶 auth = oss2.Auth(OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET) oss_img_bucket = oss2.Bucket(auth, 'oss-cn-shenzhen.aliyuncs.com', 'apg') aws_s3 = boto3.client( 's3', aws_access_key_id=AWS_ACCESS_KEY_ID[1], aws_secret_access_key=AWS_SECRET_ACCESS_KEY[1], config=botocore.client.Config(signature_version='s3v4'), region_name='us-east-1' ) aws_s3_cn = boto3.client( 's3', aws_access_key_id=AWS_ACCESS_KEY_ID[0], aws_secret_access_key=AWS_SECRET_ACCESS_KEY[0], config=botocore.client.Config(signature_version='s3v4'), region_name='cn-northwest-1' ) # ai消息标识所有组合标签 ai_all_event_type = EquipmentInfoService.get_all_comb_event_type() # 遍历调整返回数据 equipment_info_list = [] for equipment_info in packages_page: uid = equipment_info.device_uid channel = equipment_info.channel event_time = equipment_info.event_time storage_location = equipment_info.storage_location border_coords = equipment_info.border_coords event_tag = equipment_info.event_tag img_url = "" if equipment_info.is_st == 1: thumbspng = '{}/{}/{}.jpeg'.format(uid, channel, event_time) if storage_location == 1: # 阿里云oss img_url = oss_img_bucket.sign_url('GET', thumbspng, 300) else: params = {'Key': thumbspng} if region == 1: # AWS国外 params['Bucket'] = 'foreignpush' img_url = aws_s3.generate_presigned_url( 'get_object', Params=params, ExpiresIn=300) else: # AWS国内 params['Bucket'] = 'push' img_url = aws_s3_cn.generate_presigned_url( 'get_object', Params=params, ExpiresIn=300) img_url = img_url img_list = [img_url] elif equipment_info.is_st == 2: # 列表装载回放时间戳标记 split_vod_hls_obj = SplitVodHlsObject() vodqs = split_vod_hls_obj.get_vod_hls_data( uid=uid, channel=channel, start_time=int(event_time)).values('bucket_id') if not vodqs.exists(): return response.json(173) vod_bucket_qs = VodBucketModel.objects.filter(id=vodqs[0]['bucket_id']).values('bucket', 'endpoint') if not vod_bucket_qs.exists(): return response.json(173) bucket_name = vod_bucket_qs[0]['bucket'] endpoint = vod_bucket_qs[0]['endpoint'] bucket = oss2.Bucket(auth, endpoint, bucket_name) ts = '{}/vod{}/{}/ts0.ts'.format(uid, channel, event_time) if storage_location == 1: # 阿里云oss thumb0 = bucket.sign_url('GET', ts, 3600, params={'x-oss-process': 'video/snapshot,t_0000,w_700'}) thumb1 = bucket.sign_url('GET', ts, 3600, params={'x-oss-process': 'video/snapshot,t_1000,w_700'}) thumb2 = bucket.sign_url('GET', ts, 3600, params={'x-oss-process': 'video/snapshot,t_2000,w_700'}) img_url = "" img_list = [thumb0, thumb1, thumb2] else: params = {'Key': ts} if region == 1: # AWS国外 params['Bucket'] = 'foreignpush' img_url = aws_s3.generate_presigned_url( 'get_object', Params=params, ExpiresIn=300) else: # AWS国内 params['Bucket'] = 'push' img_url = aws_s3_cn.generate_presigned_url( 'get_object', Params=params, ExpiresIn=300) img_list = [img_url] elif equipment_info.is_st == 3 or equipment_info.is_st == 4: # 列表装载回放时间戳标记 img_list = [] for i in range(equipment_info.is_st): thumbspng = '{}/{}/{}_{}.jpeg'.format(uid, channel, event_time, i) if storage_location == 1: # 阿里云oss img_url = oss_img_bucket.sign_url('GET', thumbspng, 300) else: params = {'Key': thumbspng} if region == 1: # 国外AWS params['Bucket'] = 'foreignpush' img_url = aws_s3.generate_presigned_url( 'get_object', Params=params, ExpiresIn=300) else: # 国内AWS params['Bucket'] = 'push' img_url = aws_s3_cn.generate_presigned_url( 'get_object', Params=params, ExpiresIn=300) img_list.append(img_url) else: img_list = [] uid_type = Device_Info.objects.filter(UID=uid).values('Type').first() device_type = DeviceTypeModel.objects.filter(type=uid_type['Type']).values('name').first() border_coords = '' if border_coords == '' else eval(border_coords) ai_event_type_list = [] # 如果是ai消息类型，则分解eventType, 如：123 -> [1,2,3] if border_coords and event_type in ai_all_event_type: ai_event_type_list = list(map(int, str(event_type))) if EquipmentInfoService.is_combo_tag(event_type, event_tag): ai_event_type_list += EquipmentInfoService.get_combo_types(event_type, event_tag) equipment_info_data = { "uid": uid, "status": equipment_info.status, "answerStatus": equipment_info.answer_status, "alarm": equipment_info.alarm, "isSt": equipment_info.is_st, "storageLocation": storage_location, "devNickName": equipment_info.device_nick_name, "channel": channel, "eventType": equipment_info.event_type, "eventTime": int(event_time), "addTime": equipment_info.add_time, "borderCoords": border_coords, "eventTag": event_tag, "img": img_url, "imgList": img_list, "uidType": device_type["name"], "aiEventTypeList": str(ai_event_type_list), } equipment_info_list.append(equipment_info_data) data = { "list": equipment_info_list, "total": paginator.count } return response.json(0, data)