由于最近测试需要录制系统界面的操作过程,因为都是全屏的操作,所以用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
至此,录制屏幕小工具就大功告成了。 「Python 集中营」,只做知识分享 !
|