ASJServer/Controller/Cron/CronCloudPhotoController.py

# -*- encoding: utf-8 -*-
"""
@File    : CronCloudPhotoController.py
@Time    : 2022/10/24 15:48
@Author  : stephen
@Email   : zhangdongming@asj6.wecom.work
@Software: PyCharm
"""
import datetime
import logging
import os
import time
import traceback

import cv2
from django.db import transaction
from django.views import View

from Ansjer.config import PUSH_CLOUD_PHOTO, ACCESS_KEY_ID, SECRET_ACCESS_KEY, REGION_NAME, PUSH_BUCKET
from Model.models import UidSetModel, DeviceCloudPhotoInfo, DevicePicturePushInfo
from Object.AWS.AmazonS3Util import AmazonS3Util
from Object.RedisObject import RedisObject
from Object.ResponseObject import ResponseObject
from Object.utils import LocalDateTimeUtil
from Service.EquipmentInfoService import EquipmentInfoService

LOGGER = logging.getLogger('info')
UID_KEY = 'ANSJER:UID:SET:INFO'


class CronCloudPhotoView(View):

    def get(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        operation = kwargs.get('operation')
        return self.validation(request.GET, request, operation)

    def post(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        operation = kwargs.get('operation')
        return self.validation(request.POST, request, operation)

    def validation(self, request_dict, request, operation):
        response = ResponseObject()
        if operation == 'get-photo':
            return self.get_photo(response)
        elif operation == 'cache-uid-set':
            return self.cache_photo_uid_set(response)
        elif operation == 'generate-video':
            return self.generate_video(response)
        else:
            return response.json(404)

    @classmethod
    def get_photo(cls, response):
        """
        定时获取设备消息推送图
        """
        try:
            redis = RedisObject().CONN
            uid_set_cache = redis.lrange(UID_KEY, 0, -1)
            if not uid_set_cache:
                return response.json(0)
            uid_set_cache = list(uid_set_cache)
            today = datetime.date.today()
            time_stamp = cls.get_month_stamp()
            now_time = int(time.time())
            s3 = AmazonS3Util(ACCESS_KEY_ID, SECRET_ACCESS_KEY, REGION_NAME)
            for item in uid_set_cache:
                try:
                    eq_qs = EquipmentInfoService.get_equipment_info_model(str(today))
                    item = str(item, encoding="utf-8")
                    eq_qs = eq_qs.filter(event_time__gt=time_stamp, is_st=1, device_uid=item) \
                        .values('device_uid', 'channel', 'event_time', 'device_nick_name')
                    count = eq_qs.count()
                    page = int(count / 2) if count > 1 else count
                    redis.lrem(UID_KEY, item, 0)
                    if page == 0:
                        continue
                    eq_qs = eq_qs[(page - 1) * 1:page * 1]
                    eq_vo = eq_qs[0]
                    file_path = '{uid}/{channel}/{event_time}.jpeg'.format(uid=eq_vo['device_uid'],
                                                                           channel=eq_vo['channel'],
                                                                           event_time=eq_vo['event_time'])
                    s3.copy_obj(PUSH_BUCKET, PUSH_CLOUD_PHOTO, file_path)
                    push_data = {'type': 0, 'uid': eq_vo['device_uid'], 'channel': eq_vo['channel'],
                                 'event_time': eq_vo['event_time'], 'updated_time': now_time, 'created_time': now_time,
                                 'device_nick_name': eq_vo['device_nick_name']}
                    DevicePicturePushInfo.objects.create(**push_data)
                except Exception as e:
                    LOGGER.info('异常详情,errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
                    continue
            return response.json(0)
        except Exception as e:
            print(e)
            LOGGER.info('--->抽取推送图片异常:{}'.format(traceback.format_exc()))
            return response.json(177, repr(e))

    @classmethod
    def cache_photo_uid_set(cls, response):
        """
        缓存uid_set信息
        """
        try:
            photo_qs = DeviceCloudPhotoInfo.objects.filter(status=1).values('uid')
            if not photo_qs.exists():
                return response.json(0)
            redis = RedisObject()
            for item in photo_qs:
                uid_set_qs = UidSetModel.objects.filter(uid=item['uid'], detect_status=1)
                if not uid_set_qs:
                    continue
                redis.CONN.rpush(UID_KEY, item['uid'])
            return response.json(0)
        except Exception as e:
            print(e)
            LOGGER.info('--->获取uid_set信息异常:{}'.format(traceback.format_exc()))
            return response.json(177, repr(e))

    @classmethod
    def generate_video(cls, response):
        """
        生成视频并存库
        """
        try:
            photo_qs = DeviceCloudPhotoInfo.objects.filter(status=1).values('uid', 'user_id')
            if not photo_qs.exists():
                return response.json(0)
            # 获取当前项目路径
            poj_path = os.path.abspath(os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
            time_stamp = cls.get_month_stamp()
            s3 = AmazonS3Util(ACCESS_KEY_ID, SECRET_ACCESS_KEY, REGION_NAME)
            now_time = int(time.time())
            for item in photo_qs:
                try:
                    with transaction.atomic():
                        picture_qs = DevicePicturePushInfo.objects.filter(uid=item['uid'], event_time__gt=time_stamp,
                                                                          type=0) \
                            .values('uid', 'channel', 'event_time', 'device_nick_name')
                        if not picture_qs.exists():
                            continue
                        device_nickname = picture_qs[0]['device_nick_name']
                        pic_list = cls.download_push_picture(s3, item['uid'], poj_path, picture_qs)
                        video_path = poj_path + r'\Ansjer\file\video.mp4'  # 输出视频的保存路径
                        cls.picture_synthesis_video(video_path, poj_path, pic_list)  # 图片合成视频
                        video_name = datetime.datetime.now().strftime('%Y%m')
                        data = open(video_path, 'rb')
                        key = '{}/video/{}.mp4'.format(item['uid'], video_name)
                        s3.upload_file_obj(PUSH_CLOUD_PHOTO, key, data)  # 将视频资源上传至S3保存
                        os.remove(video_path)  # 上传完成删除本地资源
                        push_data = {'type': 1, 'uid': item['uid'], 'channel': 1,
                                     'event_time': int(video_name), 'updated_time': now_time, 'created_time': now_time,
                                     'device_nick_name': device_nickname, 'user_id': item['user_id']}
                        DevicePicturePushInfo.objects.create(**push_data)
                except Exception as e:
                    LOGGER.info('异常详情,errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
                    continue
            return response.json(0)
        except Exception as e:
            print(e)
            LOGGER.info('--->图片合成视频异常:{}'.format(traceback.format_exc()))
            return response.json(177)

    @staticmethod
    def download_push_picture(s3, uid, dir_url, picture_qs):
        """
        下载推送图片，讲下载资源保存到项目本地
        @param s3: s3对象
        @param uid: 设备uid
        @param dir_url: 项目文件路径
        @param picture_qs: 消息推送图片对象
        @return: pic_list 图片文件名集合
        """
        pic_list = []
        for pic in picture_qs:
            path = dir_url + r'\Ansjer\file\{}.jpeg'.format(pic['event_time'])
            s3_key = '{}/{}/{}.jpeg'.format(uid, pic['channel'], pic['event_time'])
            s3.download_object(PUSH_CLOUD_PHOTO, s3_key, path)
            pic_list.append('{}.jpeg'.format(pic['event_time']))
        return pic_list

    @staticmethod
    def picture_synthesis_video(video_path, poj_path, pic_list):
        """
        图片合成视频
        @param video_path: 输出视频的保存路径
        @param poj_path: 项目路径
        @param pic_list: 图片文件名集合
        @return:
        """
        if not pic_list:
            raise Exception('this pic_list is null.')
        fps = 1  # 帧率
        img_path = poj_path + r'\Ansjer\file\{}'.format(pic_list[0])
        image = cv2.imread(img_path)
        size = image.shape
        w = size[1]  # 宽度
        h = size[0]  # 高度
        img_size = (w, h)  # 图片尺寸
        fourcc = cv2.VideoWriter_fourcc(*"mp4v")
        video_writer = cv2.VideoWriter(video_path, fourcc, fps, img_size)
        for item in pic_list:
            img_path = poj_path + r'\Ansjer\file\{}'.format(item)
            frame = cv2.imread(img_path)
            if frame is None:
                continue
            frame = cv2.resize(frame, img_size)  # 生成视频   图片尺寸和设定尺寸相同
            video_writer.write(frame)  # 将图片写进视频里
            os.remove(img_path)
        video_writer.release()  # 释放资源

    @staticmethod
    def get_month_stamp():
        """
        获取当月开始时间戳
        """
        now_month = LocalDateTimeUtil.get_cur_month_start()
        start_time = "{} 00:00:00".format(str(now_month))
        time_array = time.strptime(start_time, "%Y-%m-%d %H:%M:%S")
        # 转换为时间戳
        return int(time.mktime(time_array))