python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Python录制屏幕工具

利用Python编写简易的录制屏幕小工具

作者:Sir 老王

这篇文章主要为大家详细介绍了如何利用Python语言编写一个简易的录制屏幕小工具,文中的示例代码讲解详细,感兴趣的小伙伴可以动手尝试一下

由于最近测试需要录制系统界面的操作过程,因为都是全屏的操作,所以用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编写简易的录制屏幕小工具的文章就介绍到这了,更多相关Python录制屏幕工具内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文