利用Python编写简易的录制屏幕小工具

from PIL import ImageGrab

import numpy as np

import cv2

import datetime

from pynput import keyboard

import threading

from loguru import logger

import time

is_running = True

def generate_video():
    '''
    生成录制视频函数
    :return:
    '''
    file_name = datetime.datetime.now().strftime('%Y-%m-%d %H-%M-%S')
    screen = ImageGrab.grab()
    width, height = screen.size
    fourcc = cv2.VideoWriter_fourcc(*'XVID')
    video = cv2.VideoWriter('%s.avi' % file_name, fourcc, 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_RGB2BGR)
        video.write(im2)
        if is_running is False:
            logger.debug('屏幕录制已经结束！')
            break
    video.release()

is_running=False
def press_keyboard(key):
    '''
    键盘监听函数
    :param key:
    :return:
    '''
    global is_running
    if key == keyboard.Key.esc:
        logger.debug('ESC已经被按下，马上结束录制！')
        is_running = False
        return False

if __name__ == '__main__':
    thread_ = threading.Thread(target=generate_video)
    thread_.start()
    logger.debug(' 开始进入视频录制！')
    with keyboard.Listener(on_press=press_keyboard) as listener:
        listener.join()

pyinstaller -i .\video.ico -Fw .\test2.py