#!/usr/bin/env python 
# -*- coding: utf-8 -*- 
"""'定時任務每五分鐘發送上一個5分鐘視頻 
 目標視頻：10.1.1.25 /usr/local/checkVideo  audited、auditing、black、white 
 find 
 """ 
import linecache 
import os 
import os.path 
import requests 
import time 
import datetime 
import sys 
reload(sys) 
sys.setdefaultencoding('utf8') 
 
#openAPI現網配置 
url='http://***/nudityRecog' 
app_key = '***' 
access_token = '***' 
imagedir='/opt/tomcat_api/video_sendto_api/image/' 
 
audited_dir='/usr/local/checkVideo/audited' 
auditing_dir='/usr/local/checkVideo/auditing' 
black_dir='/usr/local/checkVideo/black' 
white_dir='/usr/local/checkVideo/white' 
 
#時間差5分鐘執行一次 
subtime=300 
 
#生成審核中截圖 
def create_auditing_image(auditing_dir): 
 #掃描視頻目錄生成截圖 
 for parent, dirnames, filenames in os.walk(auditing_dir): # 三個參數：分別返回1.父目錄 2.所有文件夾名字（不含路徑） 3.所有文件名字 
  for filename in filenames: # 輸出文件信息 
   video_path = os.path.join(parent, filename) # 輸出文件路徑信息 
   filePath = unicode(video_path, 'utf8') #中文編碼 
   filetime= os.path.getmtime(filePath) #獲取修改時間 
   localtime=time.time() #獲取當前系統時間 
   t=localtime-filetime #兩者差值 
   #判斷差值是否小于300s 
   if t<=subtime: 
    print t,filePath 
    filename=unicode(filename, 'utf8') #下載視頻名稱 
    #生成視頻md5 
    str_md5= 'md5sum'+' ' + filePath +' '+ "| awk '{print $1}'" 
    video_md5 = os.popen(str_md5).readline(32) 
    print filePath,video_md5 
    #拼接截圖命令行， 
    str_video= "ffmpeg -ss 0:1:00 -i "+" " +filePath + " "+"-r 0.01 -f image2 "+imagedir+video_md5+"-image-%5d.jpeg" 
    images = os.popen(str_video) # 調用命令行生成截圖 
    print str_video 
 
#生成審核完截圖 
def create_audited_image(audited_dir): 
 #掃描視頻目錄生成截圖 
 for parent, dirnames, filenames in os.walk(audited_dir): # 三個參數：分別返回1.父目錄 2.所有文件夾名字（不含路徑） 3.所有文件名字 
  for filename in filenames: # 輸出文件信息 
   video_path = os.path.join(parent, filename) # 輸出文件路徑信息 
   filePath = unicode(video_path, 'utf8') #中文編碼 
   filetime= os.path.getmtime(filePath) #獲取修改時間 
   localtime=time.time() #獲取當前系統時間 
   t=localtime-filetime #兩者差值 
   #判斷差值是否小于300s 
   if t<=subtime: 
    print t,filePath 
    filename=unicode(filename, 'utf8') #下載視頻名稱 
    #生成視頻md5 
    str_md5= 'md5sum'+' ' + filePath +' '+ "| awk '{print $1}'" 
    video_md5 = os.popen(str_md5).readline(32) 
    #拼接命令行， 
    str_video= "ffmpeg -ss 0:1:00 -i "+" " +filePath + " "+"-r 0.01 -f image2 "+imagedir+video_md5+"-image-%5d.jpeg" 
    images = os.popen(str_video) # 調用命令行生成截圖 
    print str_video 
 
#生成黑名單截圖 
def create_black_image(black_dir): 
 #掃描視頻目錄生成截圖 
 for parent, dirnames, filenames in os.walk(black_dir): # 三個參數：分別返回1.父目錄 2.所有文件夾名字（不含路徑） 3.所有文件名字 
  for filename in filenames: # 輸出文件信息 
   video_path = os.path.join(parent, filename) # 輸出文件路徑信息 
   filePath = unicode(video_path, 'utf8') #中文編碼 
   filetime= os.path.getmtime(filePath) #獲取修改時間 
   localtime=time.time() #獲取當前系統時間 
   t=localtime-filetime #兩者差值 
   #判斷差值是否小于300s 
   if t<=subtime: 
    print t,filePath 
    filename=unicode(filename, 'utf8') #下載視頻名稱 
    #生成視頻md5 
    str_md5= 'md5sum'+' ' + filePath +' '+ "| awk '{print $1}'" 
    video_md5 = os.popen(str_md5).readline(32) 
    #拼接命令行， 
    str_video= "ffmpeg -ss 0:1:00 -i "+" " +filePath + " "+"-r 0.01 -f image2 "+imagedir+video_md5+"-image-%5d.jpeg" 
    images = os.popen(str_video) # 調用命令行生成截圖 
    print str_video 
 
 
#生成白名單截圖 
def create_white_image(white_dir): 
 #掃描視頻目錄生成截圖 
 for parent, dirnames, filenames in os.walk(white_dir): # 三個參數：分別返回1.父目錄 2.所有文件夾名字（不含路徑） 3.所有文件名字 
  for filename in filenames: # 輸出文件信息 
   video_path = os.path.join(parent, filename) # 輸出文件路徑信息 
   filePath = unicode(video_path, 'utf8') #中文編碼 
   filetime= os.path.getmtime(filePath) #獲取修改時間 
   localtime=time.time() #獲取當前系統時間 
   t=localtime-filetime #兩者差值 
   #判斷差值是否小于300s 
   if t<=subtime: 
    print t,filePath 
    filename=unicode(filename, 'utf8') #下載視頻名稱 
    #生成視頻md5 
    str_md5= 'md5sum'+' ' + filePath +' '+ "| awk '{print $1}'" 
    video_md5 = os.popen(str_md5).readline(32) 
    #拼接命令行， 
    str_video= "ffmpeg -ss 0:1:00 -i "+" " +filePath + " "+"-r 0.01 -f image2 "+imagedir+video_md5+"-image-%5d.jpeg" 
    images = os.popen(str_video) # 調用命令行生成截圖 
    print str_video 
#發送圖片進程 
def send_image(imagedir): 
 #掃描圖片路徑 
 for img_parent, img_dir_names, img_names in os.walk(imagedir): 
  for img_name in img_names: 
   image = os.path.join(img_parent, img_name) #拼接圖片完整路徑 
   print time.strftime("%Y-%m-%d %X"), image 
   #準備發送圖片 
   file = dict(file=open(image, 'rb')) 
   post_data = {'mark': 'room-201', 'timestamp': 1846123456, 'random': 123} 
   headers = {'app_key': app_key, 'access_token': access_token} 
   result = requests.post(url, files=file, data=post_data, headers=headers, verify=False) 
   print result.content 
   #刪除發送的圖片 
   str_img = "rm -f " + " " + image 
   del_img = os.popen(str_img).readline() 
   print del_img 
 
if __name__ == "__main__": 
 #create_auditing_image(auditing_dir) 
 #create_audited_image(audited_dir) 
 #create_black_image(black_dir) 
 #create_white_image(white_dir) 
 send_image(imagedir)