python 实现屏幕录制

用python实现屏幕录制

PIL 即pollow 的安装命令如下：

pip install pillow

其中cv2的安装是下面这条命令

pip install opencv-python

代码实现：

# coding: utf-8
from PIL import ImageGrab
import  numpy as np
import  cv2

fps = 20
start = 3  # 延时录制
end = 15  # 自动结束时间

curScreen = ImageGrab.grab()  # 获取屏幕对象
height, width = curScreen.size

video = cv2.VideoWriter('video02.avi', cv2.VideoWriter_fourcc(*'XVID'), fps, (height, width))

imageNum = 0
while True:
    imageNum += 1
    captureImage = ImageGrab.grab()  # 抓取屏幕
    frame = cv2.cvtColor(np.array(captureImage), cv2.COLOR_RGB2BGR)

    # 显示无图像的窗口
    cv2.imshow('capturing', np.zeros((1, 255), np.uint8))

    # 控制窗口显示位置，方便通过按键方式退出
    cv2.moveWindow('capturing', height - 100, width - 100)
    if imageNum > fps * start:
        video.write(frame)
    # 退出条件
    if cv2.waitKey(50) == ord('q') or imageNum > fps * end:
        break
video.release()
cv2.destroyAllWindows()