#!/usr/bin/env python
# _*_ coding:utf-8 _*_
import cv2
import os
import numpy as np
import math
import requests
import json
import base64
from threading import Timer
import time
from pprint import pprint
from PIL import Image, ImageDraw, ImageFont
import sys
reload(sys)
sys.setdefaultencoding('utf-8')
def curl(url):
headers = {}
response = requests.get(url, headers=headers)
# pprint(response)
return json.loads(response.text)
def getInfo(file_path):
info = {}
filepath, filename = os.path.split(file_path)
info["name"] = filename
if os.path.exists(file_path):
info["size"] = os.path.getsize(file_path)
info["sizeUnit"] = sizeConvert(info["size"])
cap = cv2.VideoCapture(file_path)
if cap.isOpened():
# get方法参数按顺序对应下表（从0开始编号)
rate = cap.get(5) # 帧速率
frame_number = cap.get(7) # 视频文件的帧数
info["rate"] = rate
info["duration"] = int(frame_number / rate)
info["durationHMS"] = timeConvert(info["duration"])
info["width"] = int(cap.get(3))
info["height"] = int(cap.get(4))
cap.release()
return info
def getFrames(file_path, cutTimes):
'''
file_path: 文件名
cutTimes： 抽取帧的时间数组，时间单位为s
return [时间, 帧图像]数组
'''
t_frames = []
info = getInfo(file_path)
cutFrames = [int(info["rate"] * x) + 1 for x in cutTimes]
print("cutFrames:", cutFrames)
cap = cv2.VideoCapture(file_path)
cut_cnt = 0
for cutFrame in cutFrames:
cap.set(cv2.CAP_PROP_POS_FRAMES, cutFrame - 1)
ret, frame = cap.read()
if ret:
t_frames.append((cutTimes[cut_cnt], frame))
cut_cnt += 1
print("截取视频第：" + str(cut_cnt) + " 帧")
else:
break
cap.release()
return t_frames
def timeConvert(seconds, str=True):
h = seconds // 3600
m = seconds % 3600 // 60
s = seconds % 60
if str:
if h > 0:
return '{:.0f}:{:.0f}:{:.0f}'.format(h, m, s)
else:
return '{:.0f}:{:.0f}s'.format(m, s)
else:
return h, m, s
def sizeConvert(size): # 单位换算
K, M, G = 1024, 1024 ** 2, 1024 ** 3
if size >= G:
return str(size // G) + 'GB'
elif size >= M:
return str(size // M) + 'MB'
elif size >= K:
return str(size // K) + 'KB'
else:
return str(size) + 'Bytes'
def imgResize(img, width=1980):
'''
将图片等比拉伸至宽为width
'''
h, w = img.shape[:2]
height = int(h * width / w)
if width > w: # 放大图像
img_new = cv2.resize(img, (width, height), interpolation=cv2.INTER_CUBIC)
else:
img_new = cv2.resize(img, (width, height), interpolation=cv2.INTER_AREA)
return img_new
def imgAddImg(imgDst, imgSrc, xOffset=0, yOffset=0, copy=False):
if copy:
imgDst = imgDst.copy()
imgDst[yOffset:yOffset + imgSrc.shape[0], xOffset:xOffset + imgSrc.shape[1]] = imgSrc
return imgDst
def imgAddText(img, text, xOffset=0, yOffset=0):
# cv2.putText(图像,需要添加字符串,需要绘制的坐标,字体类型,字号,字体颜色,字体粗细)
img2 = cv2.putText(img, text, (xOffset, yOffset), cv2.LINE_AA, 0.7, (249, 249, 249), 2)
return img2
def imgAddTextUTF8(img, text, xOffset=0, yOffset=0):
img_cv2_RGB = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) # cv2和PIL中颜色的hex码的储存顺序不同
img_PIL = Image.fromarray(img_cv2_RGB)
draw = ImageDraw.Draw(img_PIL)
font = ImageFont.truetype("simhei.ttf", 20, encoding="utf-8") # 参数1：字体文件路径，参数2：字体大小
draw.text((xOffset, yOffset), text, (0, 0, 0), font=font) # 参数1：打印坐标，参数2：文本，参数3：字体颜色，参数4：字体
# PIL图片转cv2 图片
img_cv2_textAdded = cv2.cvtColor(np.array(img_PIL), cv2.COLOR_RGB2BGR)
return img_cv2_textAdded
def imgWhite(height, width):
height = int(height)
width = int(width)
img = np.zeros((height, width), dtype=np.uint8)
# img = np.zeros((height,width,3), dtype=np.uint8)
img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
img[:, :, :] = 255
return img
def mp4toJpg(f_video, f_img):
start = cv2.getTickCount()
small_pic_width = 480
small_pic_cnt = 1
small_pic_per_line = 1
small_pic_gap_x = 0
small_pic_gap_y = 0
small_pic_top = 0
small_pic_bottom = 0
small_pic_left = 0
small_pic_right = 0
time_reduce = 0
# 获取信息
info = getInfo(f_video)
if info['duration'] == 0: # 预防不足一秒的场景
print('视频不足一秒 %s' % f_video)
t_imgs = getFrames(f_video, [0])
else:
t_imgs = getFrames(f_video, [1])
small_pic_cnt = len(t_imgs)
# 对图像进行缩放，添加时间戳水印
imgs_fixed = []
h, w = (0, 0)
for time, img in t_imgs:
# print(r"正在加工图片%d"%time)
img_resize = imgResize(img, small_pic_width)
img_add_time = imgAddText(img_resize, timeConvert(time), 10, 25)
h, w = img_add_time.shape[:2]
# cv2.imwrite(str(time) + "_time.png", img_add_time)
imgs_fixed.append(img_add_time)
# 先画一张空白图
height = (math.ceil(small_pic_cnt * 1.0 / small_pic_per_line)) * h
width = small_pic_left + small_pic_right + small_pic_per_line * (small_pic_gap_x + w) - small_pic_gap_x
print("h = %.0f, w = %.0f" % (h, w))
print("height = %.0f, width = %.0f" % (height, width))
img_backgroud = imgWhite(height, width)
# 在大图上加入视频信息
filepath, filename = os.path.split(f_video)
print('file name %s' % filename)
print('file path %s' % filepath)
# img_backgroud = imgAddTextUTF8(img_backgroud, "name : %s"%filename, 0, 0)
# img_backgroud = imgAddTextUTF8(img_backgroud, "time : %s"%info["durationHMS"], 20, 45)
# img_backgroud = imgAddTextUTF8(img_backgroud, "size : %s"%(info["sizeUnit"]), 20, 70)
# img_backgroud = imgAddTextUTF8(img_backgroud, "px : %dx%d"%(info["width"],info["height"]), 20, 95)
# 在大图上加入视频截图
i = 0
while i < small_pic_cnt:
xx = i % small_pic_per_line
yy = int(i / small_pic_per_line)
print(r"正在将第(%d,%d)放入背景" % (xx + 1, yy + 1))
off_x = int(small_pic_left + (small_pic_gap_x + w) * xx)
off_y = int(small_pic_top + (small_pic_gap_y + h) * yy)
img_backgroud = imgAddImg(img_backgroud, imgs_fixed[i], off_x, off_y)
# imgAddImg(img_backgroud, img_backgroud, 0, 0)
i += 1
end = cv2.getTickCount()
time_spent = (end - start) / cv2.getTickFrequency() / 60
print("耗时%.1f min" % time_spent)
cv2.imwrite(f_img, img_backgroud)
# cv2.imshow("img", img_backgroud)
# cv2.waitKey(0)
def curlGet(url):
# 请求接口
r = requests.get(url)
# json字符串转换字典格式
return json.loads(r.text)
def curlPost(url, in_value):
r = requests.post(url, in_value)
return json.loads(r.text)
# 获取phone需要审核的视频
def getphoneVideo():
url = 'http://127.0.0.1:8080/Aider/phone/getphoneVideo?channel_id=%d' % CHANNEL_ID
dictinfo = curlGet(url)
if dictinfo['status'] == 200:
# {"user_id":10000008058,"passport":"xkdkkd","full_name":"kdkdk","base64":""}
return dictinfo['data']
print('请求结果异常 %s : %s' % (dictinfo['status'], dictinfo['msg']))
return {}
# 保存phone视频截图
def savephoneVideoUrl(user_id, base64_str):
url = 'http://127.0.0.1:8080/Aider/phone/savephoneVideoUrl'
dictinfo = curlPost(url, {'channel_id': CHANNEL_ID, 'user_id': user_id, 'base64': base64_str})
if dictinfo['status'] == 200:
# {"user_id":10000008058,"passport":"xkdkkd","full_name":"kdkdk","base64":""}
return dictinfo['data']
print('请求结果异常 %s : %s' % (dictinfo['status'], dictinfo['msg']))
return {}
# base64 转换 mp4 文件
def createMp4File(base64_str, user_id):
img_data = base64.b64decode(base64_str)
mp4_file = '/tmp/phone/%s.mp4' % user_id
file = open(mp4_file, 'wb')
file.write(img_data)
file.close()
return mp4_file
# 文件 转换 base64 字符串
def fileToBase64(file_path):
with open(file_path, "rb") as f:
# b64encode是编码，b64decode是解码
base64_data = base64.b64encode(f.read())
return base64_data
# 检测 phone视频
def checkphone():
print("------ checkphone 时间：%s ------" % time.ctime())
print('------ 开始phone视频截取 ------')
try:
print('开始获取phone视频详情')
# 获取视频
phone_video_info = getphoneVideo()
if phone_video_info == {}:
raise Exception, " -- 暂无需要审核视频 -- "
user_id = phone_video_info['user_id']
mp4_file = createMp4File(phone_video_info['base64'], user_id)
print('mp4文件地址%s' % mp4_file)
# 截图文件地址
img_file = "/tmp/phone/%s.jpg" % user_id
print('Mp4 %s 转 Jpg ：%s' % (mp4_file, img_file))
mp4toJpg(mp4_file, img_file)
if os.path.exists(img_file) == False:
raise Exception, '-- 图片文件生成失败 --'
img_base64 = fileToBase64(img_file)
print('保存phone视频截图 %d' % user_id)
if savephoneVideoUrl(user_id, img_base64) == {}:
raise Exception, '-- 保存phone 视频接口抛错 --'
except Exception, err:
print err
except IOError:
print("Error: 没有找到文件或读取文件失败")
else:
print("phone 视频截图上传成功")
print('------ 结束phone视频截取 ------')
# 如果count小于10，开始下一次调度
Timer(10, checkphone).start()
#生产仅单一渠道使用
CHANNEL_ID = 123
if __name__ == '__main__':
# 指定10秒后执行print_time函数
checkphone()
