由于最近测试需要录制系统界面的操作过程,因为都是全屏的操作,所以用python做一个简单的录屏小工具。
实现过程也是比较简单,就是通过对屏幕操作进行不断的截图,最后将截图合成一个操作视频的过程。由于我们只是做简单的截屏功能,并没有加入音频效果。
1、准备
开始之前我们还是按照以往的方式介绍一下使用到的第三方的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
由于实现过程比较少,这里我们不再创建标准的class实现,直接在.py文件写相关的函数来实现。
2、代码
实现过程主要是通过两个函数来实现的,一个是实现具体的屏幕截图操作,并且将其写入视频。另一个则是用来监听键盘的输入,若是按下esc键时则退出当前的录制操作。
初始化一个变量作为停止标志。
is_running = True
创建录制过程生成函数generate_video,用以生成录制视频。
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()
创建键盘监听函数press_keyboard,监听输入,若是按下esc键则改变运行状态。
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
执行主函数main,启动则执行屏幕录制。
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将其打包成exe的可执行应用,使用时直接双击打开即可开始录制屏幕的操作过程。
pyinstaller -i .\video.ico -Fw .\test2.py
至此,录制屏幕小工具就大功告成了。