使用Python来做一个屏幕录制工具的操作代码

一、写在前面

作为一名测试,有时候经常会遇到需要录屏记录自己操作,方便后续开发同学定位。以前都是用ScreenToGif来录屏制作成动态图,偶尔的机会看到python也能实现。那就赶紧学习下。

二、效果展示

使用Python来做一个屏幕录制工具的操作代码

三、知识串讲

这次要讲的东西可能比较多了,涉及到pyqt5 GUI软件的制作、QThread多线程的使用、Sikuli库的图形操作、win32库的模拟键盘操作、cv2库的写视频文件等。下面我们一点点来蚕食我这次写的代码。

1、GUI界面制作

这次我用的是现成的Pyqt5界面布局类,QVBoxLayout。这个类可以快速协助我完成按钮的垂直分布,而且按钮添加也更方便。

button1 = QPushButton("自定义录屏")
layout.addWidget(button1)

两行代码就完成了按钮的命名和添加。我之前玩qt时,用的都是qt的UI界面,对应生成的组件代码也比较复杂。因此,在开发一些少量按钮、简单布局时可以用QVBoxLayout类。如果喜欢水平布局,可以用QHBoxLayout类,使用方法是一样的。

另外,在按钮点击关联的功能函数,即work()方法时,如果想带参数,可以通过lambda匿名函数来实现。这 也是个小技巧。

# 不带参数
button1.clicked.connect(self.work)
# 带参数
button1.clicked.connect(lambda: self.work(1))

2、QThread类的多线程使用

因为录屏工具有开始和停止两个功能,一开始时我用的是单线程,发现工具就会卡死。查了一些资料,发现针对这种情况,应该要使用多线程来实现,而QT库中本身就有多线程类--QThread。

使用方法是通过继承QThread类,重写run方法来实现的。

(但是其实这种使用方法,QT大神们是不赞成这样使用的,我会在第2篇文章中再简单说明更好的多线程使用方法)

这 里要注意,work()函数必须是Ui_Mainwindow类方法,因为如果不是类方法,会在运行GUI时导致生命周期直接结束,导致录屏代码没见运行就报错退出。

class WorkThread(QThread):
  def __init__(self,n):
    super(WorkThread,self).__init__()
    self.n = n

  def run(self):
    XXXXX

3、sikuli库图形识别

由于这个库的使用方法和介绍,我在之前的博客里已经提过 了。因此只简单地呈现下代码。这段代码主要是为了自定义录屏时,可以获取选择范围的坐标值,并传值给recording函数,从而完成自定义录屏功能。

def SelectRegion():
  jvmPath = jpype.get_default_jvm_path()
  jpype.startJVM(jvmPath,'-ea','-Djava.class.path=F:\\sikuli\\1\\sikulixapi.jar') #加载jar包路径
  Screen = jpype.JClass('org.sikuli.script.Screen')
  myscreen = Screen()
  region = myscreen.selectRegion() # 自定义获取屏幕范围
  return region

4、win32库模拟键盘操作

其实这个库不用也是可以的,我为什么要用呢?主要是为了方便用户在进行录屏时,能自动将工具界面缩小。一切为了用户嘛!

以下这段代码 是为了缩小工具窗口,其中91表示左win键,40表示方向向下键。****即win+向下键是可以实现窗口缩小功能的。****keybd_event(91,0)表示按下win键,

keybd_event(91,win32con.KEYEVENTF_KEYUP,0)则是松开win键。

另外,这里为什么要加 上sleep(0.5)?这是因为在按下win键后要延迟按方向键,不然是 不起作用的。

def Minimize_Window():
  win32api.keybd_event(91,0)
  time.sleep(0.5)
  win32api.keybd_event(40,0)
  time.sleep(0.5)
  win32api.keybd_event(91,0)
  win32api.keybd_event(40,0)

5、录屏主代码

这段代码其实网上已经有很多类似的代码,并且我已经加了注释,相信大家应该能理解。这里我想注明下的是:如何停止录屏。

如果大家有去 网上查如何停止录屏的方法,很多人都会写以下代码:

if cv2.waitKey(1) & 0xFF == ord('q'):
  break

然后告诉你,按q键就会停止录屏。但是你会发现,实际情况根本停止不了,为什么呢?因为还 有一句屏幕显示的代码:

cv2.imshow('imm',img_bgr)
if cv2.waitKey(1) & 0xFF == ord('q'):
  break

如果你不亲自执行一次,你以为会万事大吉,但你错了。这样写,会导致你的电脑屏幕被每一帧画面给撑暴!因为用的while True,因此每一帧画面都会显示,即1S 25帧画面会不停地显示在你桌面上!

因此,综上的问题,我采用了一种取巧的方法:在录屏开始时生成一个标记文件,通过标记文件是否被删除来判断是否要停止录屏功能。

四、示例代码

1、工具GUI界面代码:

# coding=utf-8
# @Software : PyCharm
#Python学习群827513319

import sys
from PyQt5.QtCore import *
from PyQt5.QtWidgets import *
import time
import win32api,win32con
from recording import *

class WorkThread(QThread):
  def __init__(self,self).__init__()
    self.n = n

  def run(self):
    if self.n == 1:
      Minimize_Window()
      Recording(1)
    elif self.n == 2:
      Minimize_Window()
      Recording(2)
    else:
      StopRecording()

def Minimize_Window():
  win32api.keybd_event(91,0)

class Ui_Mainwindow():
  def setupUi(self,top):
    # 垂直布局类QVBoxLayout
    layout = QVBoxLayout(top)
    # 添加录屏相关按钮
    button1 = QPushButton("自定义录屏")
    layout.addWidget(button1)
    button2 = QPushButton("全屏录屏")
    layout.addWidget(button2)
    button3 = QPushButton("停止录屏")
    layout.addWidget(button3)
    self.text = QPlainTextEdit('欢迎使用!')
    layout.addWidget(self.text)
    button1.clicked.connect(lambda: self.work(1))
    button2.clicked.connect(lambda: self.work(2))
    button3.clicked.connect(lambda: self.work(3))

  def work(self,n):
    if n == 1 :
      print('已选择自定义录屏:')
      self.text.setPlainText('正在录屏中,请等待……')
    elif n == 2 :
      print('已选择全屏录屏:')
      self.text.setPlainText('正在录屏中,请等待……')
    else:
      print('已选择结束录屏:')
      self.text.setPlainText('录屏结束!(点击关闭按钮,可退出程序!)')
    self.workThread = WorkThread(n)
    self.workThread.start()

if __name__ == "__main__":
  app = QApplication(sys.argv)
  top = QWidget()
  top.setWindowTitle('录屏小工具')
  top.resize(300,170)
  ui = Ui_Mainwindow()
  ui.setupUi(top)
  top.show()
  sys.exit(app.exec_())# coding=utf-8

2、录屏函数

# coding=utf-8
# @Software : PyCharm

from PIL import ImageGrab
import numpy as np
import cv2
import os
import jpype

def Recording(tag=1):
  # 录屏开始时创建test.txt,作为结束录屏的条件
  #Python学习群827513319
  if not os.path.exists('test.txt'):
    f = open('test.txt','w')
    f.close()
  # 根据tag值判断自定义录屏或全录屏
  if tag == 1:
    r = SelectRegion()
    record_region = (r.x,r.y,r.w + r.x,r.h + r.y) # 自定义录屏的范围(左上坐标、右下坐标)
  elif tag == 2:
    record_region = None
  image = ImageGrab.grab(record_region) # 获取指定范围的屏幕对象
  width,height = image.size
  fourcc = cv2.VideoWriter_fourcc(*'XVID')
  video = cv2.VideoWriter('test.avi',fourcc,25,(width,height)) # 默认视频为25帧
  while True:
    captureImage = ImageGrab.grab(record_region) # 抓取指定范围的屏幕
    frame = cv2.cvtColor(np.array(captureImage),cv2.COLOR_RGB2BGR)
    video.write(frame) # 将每帧画面写视频文件
    # 停止录屏的条件:test.txt被删除
    if not os.path.exists('test.txt'):
      break
  video.release()
  cv2.destroyAllWindows()

def SelectRegion():
  jvmPath = jpype.get_default_jvm_path()
  jpype.startJVM(jvmPath,'-Djava.class.path=F:\\sikuli\\1\\sikulixapi.jar') #加载jar包路径
  Screen = jpype.JClass('org.sikuli.script.Screen')
  myscreen = Screen()
  region = myscreen.selectRegion() # 自定义获取屏幕范围
  return region

def StopRecording():
  os.remove('test.txt') #停止录屏的触发条件

if __name__ == "__main__":
  Recording()

五、总结

至此,基本实现了录屏小工具的代码开发。但是如果你是对代码中的相关库不熟悉,或者都没下载相关的库,那我相信你还会遇到很多坑。因此,为了方便一些小伙伴能快速把代码跑起来,我将在下一篇文章中讲讲我在开发时遇到的一些坑,方便大家能避免这些问题。好了,今天就先到这里!Bye!

以上所述是小编给大家介绍的使用Python来做一个屏幕录制工具的操作代码,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对我们网站的支持!
如果你觉得本文对你有帮助,欢迎转载,烦请注明出处,谢谢!

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐


本文从多个角度分析了vi编辑器保存退出命令。我们介绍了保存和退出vi编辑器的命令,以及如何撤销更改、移动光标、查找和替换文本等实用命令。希望这些技巧能帮助你更好地使用vi编辑器。
Python中的回车和换行是计算机中文本处理中的两个重要概念,它们在代码编写中扮演着非常重要的角色。本文从多个角度分析了Python中的回车和换行,包括回车和换行的概念、使用方法、使用场景和注意事项。通过本文的介绍,读者可以更好地理解和掌握Python中的回车和换行,从而编写出更加高效和规范的Python代码。
SQL Server启动不了错误1067是一种比较常见的故障,主要原因是数据库服务启动失败、权限不足和数据库文件损坏等。要解决这个问题,我们需要检查服务日志、重启服务器、检查文件权限和恢复数据库文件等。在日常的数据库运维工作中,我们应该时刻关注数据库的运行状况,及时发现并解决问题,以确保数据库的正常运行。
信息模块是一种可重复使用的、可编程的、可扩展的、可维护的、可测试的、可重构的软件组件。信息模块的端接需要从接口设计、数据格式、消息传递、函数调用等方面进行考虑。信息模块的端接需要满足高内聚、低耦合的原则,以保证系统的可扩展性和可维护性。
本文从电脑配置、PyCharm版本、Java版本、配置文件以及程序冲突等多个角度分析了Win10启动不了PyCharm的可能原因,并提供了解决方法。
本文主要从多个角度分析了安装SQL Server 2012时可能出现的错误,并提供了解决方法。
Pycharm是一款非常优秀的Python集成开发环境,它可以让Python开发者更加高效地进行代码编写、调试和测试。在Pycharm中设置解释器非常简单,我们可以通过创建新项目、修改项目解释器、设置全局解释器等多种方式进行设置。
Python中有多种方法可以将字符串转换为整数,包括使用int()函数、try-except语句、正则表达式、map()函数、ord()函数和reduce()函数。在实际应用中,应根据具体情况选择最合适的方法。
本文介绍了导入CSV文件的多种方法,包括使用Excel、Python和R等工具。同时,还介绍了导入CSV文件时需要注意的一些细节和问题。CSV文件是数据处理和分析中不可或缺的一部分,希望本文能够对读者有所帮助。
mongodb是一种新型的数据库,它采用了面向文档的数据模型,具有灵活性、高性能和高可用性等优势。但是,mongodb也存在数据结构混乱、安全性和学习成本高等问题。
当Python运行不了时,我们应该从代码、Python环境、操作系统和硬件设备等多个角度来排查问题,并采取相应的解决措施。
Python列表是一种常见的数据类型,排序是列表操作中的一个重要部分。本文介绍了Python列表降序排序的方法,包括使用sort()函数、sorted()函数以及自定义函数进行排序。使用sort()函数可以简单方便地实现降序排序,但会改变原始列表的顺序;使用sorted()函数可以保留原始列表的顺序,但需要创建一个新的列表;使用自定义函数可以灵活地控制排序的方式,但需要编写额外的代码。
本文介绍了如何使用Python输入一段英文并统计其中的单词个数,从去除标点符号、忽略单词大小写、排除常用词汇等多个角度进行了分析。此外,还介绍了使用NLTK库进行单词统计的方法。
虚拟环境可以帮助我们在同一台机器上运行不同版本的Python、安装不同的Python包,并且不会相互影响。创建虚拟环境的命令是python3 -m venv myenv,进入虚拟环境的命令是source myenv/bin/activate,退出虚拟环境的命令是deactivate。在虚拟环境中可以使用pip安装包,也可以使用Python运行程序。
本文从XHR对象、fetch API和jQuery三个方面分析了JS获取响应状态的方法及其应用。以上三种方法都可以轻松地发送HTTP请求,并处理响应数据。
桌面的命令包括常见的操作命令、系统命令、批处理命令以及第三方应用程序提供的命令。我们可以通过鼠标右键点击桌面、创建快捷方式、创建批处理文件等方式来运用这些命令,从而更好地管理计算机,提高工作效率。
本文分析了应用程序闪退的多个原因,包括应用程序本身存在问题、手机或平板电脑系统问题、硬件问题、网络问题和其他原因。同时,本文提供了解决闪退问题的多种方式,包括更新或卸载重新下载应用程序、升级系统或进行修复、清理手机缓存、清理不必要的文件或者是更换电池等方式来解决、确保网络信号的稳定性、注意用户隐私和安全问题。
本文介绍了使用Python下载图片的多种方法,包括使用Python标准库urllib.request、第三方库requests、多线程和异步IO。这些方法在不同情况下都有它们的优缺点。使用这些方法,我们可以轻松地将网络上的图片下载到本地,方便我们在离线状态下查看或处理这些图片。
MySQL数据文件是指存储MySQL数据库中数据的文件,存储位置的选择对数据库的性能、可靠性和安全性都有着重要的影响。本文从存储位置的选择、存储设备的选择、存储空间的管理和存储位置的安全性等多个角度对MySQL数据文件的存储位置进行分析,最后得出需要根据实际情况综合考虑多个因素,选择合适的存储位置和存储设备,并进行有效的存储空间管理和安全措施的结论。
AS400是一种主机操作系统,每个库都包含多个表。查询库表总数是一项基本任务。可以使用命令行、系统管理界面以及数据库管理工具来查询库表总数。查询库表总数可以帮助用户更好地管理和优化数据,包括规划数据存储、优化查询性能以及管理空间资源。