import time,threading
from datetime import datetime
from PIL import ImageGrab
from cv2 import *
import numpy as np
from pynput import keyboard
def video_record():   # 錄入視頻
  global name
  name = datetime.now().strftime('%Y-%m-%d %H-%M-%S') # 當前的時間（當文件名）
  screen = ImageGrab.grab() # 獲取當前屏幕
  width, high = screen.size # 獲取當前屏幕的大小
  fourcc = VideoWriter_fourcc('X', 'V', 'I', 'D') # MPEG-4編碼,文件后綴可為.avi .asf .mov等
  video = VideoWriter('%s.avi' % name, fourcc, 15, (width, high)) # （文件名，編碼器，幀率，視頻寬高）
  #print('3秒后開始錄制----')  # 可選
  #time.sleep(3)
  print('開始錄制!')
  global start_time
  start_time = time.time()
  while True:
    if flag:
      print("錄制結束！")
      global final_time
      final_time = time.time()
      video.release() #釋放
      break
    im = ImageGrab.grab()  # 圖片為RGB模式
    imm = cvtColor(np.array(im), COLOR_RGB2BGR) # 轉為opencv的BGR模式
    video.write(imm)  #寫入
    # time.sleep(5) # 等待5秒再次循環
def on_press(key):   # 監聽按鍵
  global flag
  if key == keyboard.Key.home:
    flag = True # 改變
    return False # 返回False，鍵盤監聽結束！
def video_info():   # 視頻信息
  video = VideoCapture('%s.avi' % name)  # 記得文件名加格式不要錯！
  fps = video.get(CAP_PROP_FPS)
  Count = video.get(CAP_PROP_FRAME_COUNT)
  size = (int(video.get(CAP_PROP_FRAME_WIDTH)), int(video.get(CAP_PROP_FRAME_HEIGHT)))
  print('幀率=%.1f'%fps)
  print('幀數=%.1f'%Count)
  print('分辨率',size)
  print('視頻時間=%.3f秒'%(int(Count)/fps))
  print('錄制時間=%.3f秒'%(final_time-start_time))
  print('推薦幀率=%.2f'%(fps*((int(Count)/fps)/(final_time-start_time))))
if __name__ == '__main__':
  flag = False
  th = threading.Thread(target=video_record)
  th.start()
  with keyboard.Listener(on_press=on_press) as listener:
    listener.join()
  time.sleep(1)  # 等待視頻釋放過后
  video_info()