# -*- coding:utf-8 -*-
# @Time   : 2022-08-10
# @Author : Carl_DJ

'''
制作視頻錄制器
實(shí)現(xiàn)功能：
1、倒計(jì)時(shí)開始進(jìn)行錄制
2、點(diǎn)擊 鍵盤的 esc按鍵，就停止錄制
3、打包成.exe， 直接點(diǎn)擊就可以運(yùn)行
'''
import cv2
from PIL import ImageGrab
import numpy as np
import datetime
from pynput import keyboard
import threading
from loguru import logger
import  time
import  os
# 初始化變量
is_running = True

'#創(chuàng)建錄制過程生成的generate_video,用于生成錄制視頻'

def generate_video():
    '''
    生成錄制視頻
    :return:
    '''
    #當(dāng)前時(shí)間戳為文件名字
    start_time = datetime.datetime.now().strftime('%Y-%m-%d %H%M%S')
    file_name = (start_time + '錄制視頻')
    # file_name= input("輸入錄制文件的名稱")
    screen = ImageGrab.grab()
    width,height = screen.size
    fource = cv2.VideoWriter_fourcc(*'XVID')
    video = cv2.VideoWriter('%s.avi'% file_name,fource,20,(width,height))
    for n in range(3):
        logger.debug(str(3-n)+ '秒后開始錄制')
        time.sleep(1)
    while True:
        im1 = ImageGrab.grab()
        im2 = cv2.cvtColor(np.array(im1),cv2.COLOR_RGBA2RGB)
        video.write(im2)
        if is_running is False:
            logger.debug('屏幕錄制結(jié)束')
            break
    video.release()

'創(chuàng)建鍵盤監(jiān)聽函數(shù) press_keyboard,監(jiān)聽輸入，如果按下esc則改變運(yùn)行狀態(tài)'
# is_running = False #注釋掉
def press_keyboard(key):
    '''
    鍵盤監(jiān)聽函數(shù)
    :param key:
    :return:
    '''
    global  is_running
    if key == keyboard.Key.esc:
        logger.debug('已按下ESC鍵，錄制結(jié)束')
        is_running = False
        return False

if __name__ == '__main__':
    thread_ = threading.Thread(target=generate_video)
    thread_.start()
    logger.debug('開始進(jìn)入錄制視頻')
    with keyboard.Listener(on_press = press_keyboard) as listener:
        listener.join()