用python轻轻松松写一个录屏工具

import cv2
import numpy as np
import pyautogui

def record_screen(output_file, fps=30):
    # 获取屏幕宽度和高度
    screen_width, screen_height = pyautogui.size()
    
    # 创建视频编码器
    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
    out = cv2.VideoWriter(output_file, fourcc, fps, (screen_width, screen_height))
    
    while True:
        # 捕获屏幕图像
        img = pyautogui.screenshot()
        
        # 转换图像颜色空间
        frame = np.array(img)
        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
        
        # 将图像写入视频文件
        out.write(frame)
        
        # 按下q键停止录屏
        if cv2.waitKey(1) == ord('q'):
            break
    
    # 释放资源
    out.release()
    cv2.destroyAllWindows()
    print(f'录制已保存为{output_file}')

def main():
    output_file = 'screencast.mp4'
    fps = 30
    
    print('按下q键停止录屏')
    record_screen(output_file, fps)

if __name__ == '__main__':
    main()

这段代码使用pyautogui库的screenshot()方法来捕获屏幕图像，并使用opencv-python库的VideoWriter类来创建一个视频编码器，将捕获到的图像写入视频文件。

运行代码后，按下"q"键即可停止录屏，并将录制的视频保存为screencast.mp4文件。

请注意，这只是一个简单的示例，可能在某些系统上无法正常工作或存在一些限制。如果需要更高级的功能，可以考虑使用第三方库，如pyav、imageio等。

为了实现该功能，你需要安装opencv-python和pyautogui库。可以使用以下命令在终端中安装这些库：

pip install opencv-python pyautogui