利用Python编写简易的录制屏幕小工具_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