文/IT可达鸭

图/IT可达鸭、网络

写在前面

这次不蹭热点了，简简单单分享一个录屏软件的源码。我在阅读了近十多个录屏软件的源码后，发现这个是最简单，最好练手的一个源码。

为了便于理解，里面的一些不必要的代码我作了删减，主要是用来学习研究。

• 环境配置

python版本： 3.6.0

编辑器： pycharm

项目所需要的环境安装包

pip install numpy
pip install opencv-python
pip install pillow

• 需要注意的点

要先安装numpy，再安装opencv-python。

第一步：导包

time: 是一个时间包，其中time.sleep(n) 表示主线程睡眠 n 秒钟，子线程可以并发执行。

threading: 是一个关于线程操作的包，包括常用的线程函数、线程对象、锁、事件等。

cv2: 是一个强大的图像处理库 opencv, 据说人脸识别、视频处理等操作都可以做到。

numpy: 是一个扩展包，支持大量的维度数组与矩阵运算，包含大量数学函数。

PIL: 是一个图像处理标准库，一般和opencv一起用。

• 进程和线程简单介绍

什么是进程？举个例子，音乐播放器是一个进程，浏览器也是一个进程，互相不干扰。

什么是线程？音乐播放器中，音频播放是一个线程、歌词播放也是一个线程，它们之间有交流，保证歌词和音频同步播放。

第二步：录屏线程编写

写一个屏幕录制线程工具类，继承 threading.Tread。重写__init__() 方法，调用threading.Thread的__init__() 方法。重写 run() 方法，不断截屏，并保存到video 对象中。利用 b_record 来控制是否停止录屏。这是一个线程开发的基本写法，只要涉及多线程，都可以这么写。

后续如果想加一个录制麦克风声音，可以写一个声音录制的线程类。

第三步：主函数

首先设定好录屏保存的路径，新建一个录屏对象。 start() 启动录制线程，time.sleep(10) 表示主线程停留10秒，但是录屏线程依旧运行。直到主线程运行十秒后，运行 stop_record()， 修改b_record 变量为Flase，录屏线程检测变化到就会立刻停止录屏，并输出保存视频文件。

完整代码

# encoding:utf-8


import time
import threading
from cv2 import cv2
import numpy as np
from PIL import ImageGrab


class VideoCapThread(threading.Thread):


    def __init__(self, video_file='record.avi'):
        threading.Thread.__init__(self)
        self.b_record = True
        self.video = cv2.VideoWriter(video_file, 
                cv2.VideoWriter_fourcc(*'XVID'),
                32, ImageGrab.grab().size)  # 帧率为32，可以调节


    def run(self):
        while self.b_record:
            im = ImageGrab.grab()
            imm = cv2.cvtColor(np.array(im), cv2.COLOR_RGB2BGR)
            self.video.write(imm)
        self.video.release()
        cv2.destroyAllWindows()


    def stop_record(self):
        self.b_record = False


if __name__ == "__main__":


    avi_file = 'data/avi_tmp.avi'
    t1 = VideoCapThread(avi_file)
    t1.start()
    time.sleep(10)  # 录制10s
    t1.stop_re

最后，附上完整代码。仅有三十多行，是不是非常简洁？其实，编程也没那么难。

结语

当然，距离完整的录屏软件还有很多东西可以做。如果有兴趣深究下去，可以在这个源码的基础上做优化和改进。例如，录制屏幕的分辨率设置、录屏位置坐标自由设置、增加UI界面（QtPython）、增加声音或者麦克风录制、视频音频合并（FFmpeg）等等。

如果有疑问想获取源码，可以关注后，在后台私信我，回复：python录屏。 我把源码发你。持续关注"IT可达鸭" ，每天分享有趣Python源码。最后，感谢大家的阅读，祝大家工作生活愉快！