我要评论本文将展示如何在有另一个python进程运行时，使用python自动录制屏幕生成视频，并用以检查其执行情况。
大多数时候，当需要对非常老旧的系统进行自动化时，将不得不通过gui来完成，因为它们没有api，gui自动化可能非常棘手并且会出现意外行为。
解决这个问题的最好方法是尝试处理代码中的每一个可能的异常，但也可能会发生意外错误。所以当发生异常时，最好保存一个异常情况的视频来以便于分析，并调试代码。
使用python的多进程库，在与运行自动化的线程不同的线程中，运行屏幕录制脚本。已经用三个不同的库测试了这个脚本的屏幕录制：mss，pillow，和pyautogui。在这些库中，mss是表现最好的一个。
以下是固定时间录制的代码片段：
from time import sleep
from mss.windows import mss as mss
import multiprocessing
import pyautogui
import cv2
import numpy as np
from os import remove
class fixedtimecapture:
def __init__(self, capture_time:int, video_path:str='c:/test') -> none:
self.fps = 15
self.capture_time = capture_time
self.video_path = video_path.replace('\\','/') + '.avi'
self.process = multiprocessing.process(
target=self.capture_screen,
args=(),
name='screen capture'
)
def capture_screen(self):
screen_size = tuple(pyautogui.size())
fourcc = cv2.videowriter_fourcc(*'xvid')
out = cv2.videowriter(self.video_path, fourcc, self.fps, (screen_size))
print(f'screen capture started')
w, h = pyautogui.size()
monitor = {'top':0, 'left':0, 'width':w, 'height':h}
with mss() as sct:
for i in range(int(self.capture_time * self.fps)):
try:
img = sct.grab(monitor=monitor)
frame = np.array(img)
frame = cv2.cvtcolor(frame, cv2.color_bgra2bgr)
out.write(frame)
except ioerror:
pass
cv2.destroyallwindows()
out.release()
sleep(3)
print('screen capture finished')
def start_capture(self):
self.process.start()
def abort_capture(self):
if not self.process.is_alive():
print('the process is not executing')
else:
print('aborting screen capture')
self.process.kill()
sleep(3)
try:
remove(self.video_path)
except:
print('couldnt delete the file')
if __name__ == '__main__':
ct = fixedtimecapture(capture_time=10, video_path=r'c:\test')
ct.start_capture()
# 如果想中止录制并在其完成之前删除文件
# ct = fixedtimecapture(capture_time=10, video_path=r'c:\test2')
# ct.start_capture()
# sleep(5)
# ct.abort_capture()请注意，其中包含了一个选项来中止进程并在必要时删除文件。使用的是avi视频编解码器，录制速度为15帧/秒，但可以自行更改这些设置。
不知道录制的过程要花多长时间的情况下，可以创建一个函数来开始录制，另一个函数来结束录制。
from time import sleep
from mss.windows import mss as mss
import multiprocessing
import pyautogui
import cv2
import numpy as np
from os import remove
class capturescreen:
def __init__(self, video_path:str) -> none:
self.fps = 15
self.video_path = video_path.replace('\\','/') + '.avi'
self.process = multiprocessing.process(
target=self.capture_screen,
args=(),
name='screen capture'
)
def capture_screen(self):
screen_size = tuple(pyautogui.size())
fourcc = cv2.videowriter_fourcc(*'xvid')
out = cv2.videowriter(self.video_path, fourcc, self.fps, (screen_size))
img = none
print(f'screen capture started')
# mss
w, h = pyautogui.size()
monitor = {'top':0, 'left':0, 'width':w, 'height':h}
with mss() as sct:
while true:
try:
try:
img = sct.grab(monitor=monitor)
frame = np.array(img)
frame = cv2.cvtcolor(frame, cv2.color_bgra2bgr)
out.write(frame)
except ioerror:
pass
except keyboardinterrupt:
break
cv2.destroyallwindows()
out.release()
# # pillow
# from pil import imagegrab
# w, h = pyautogui.size()
# while true:
# try:
# try:
# img = imagegrab.grab(bbox =(0, 0, w, h))
# frame = np.array(img)
# frame = cv2.cvtcolor(frame, cv2.color_bgr2rgb)
# out.write(frame)
# except ioerror:
# pass
# except keyboardinterrupt:
# break
# cv2.destroyallwindows()
# out.release()
# # pyautogui
# while true:
# try:
# try:
# img = pyautogui.screenshot()
# frame = np.array(img)
# frame = cv2.cvtcolor(frame, cv2.color_bgr2rgb)
# out.write(frame)
# except ioerror:
# pass
# except keyboardinterrupt:
# break
# cv2.destroyallwindows()
# out.release()
def start_capture(self):
self.process.start()
def finish_capture(self):
if not self.process.is_alive():
print('the process is not executing')
else:
self.process.terminate()
sleep(3)
print('screen capture finished')
if __name__ == '__main__':
c = capturescreen(video_path=r'c:\test')
c.start_capture()
sleep(15)
c.finish_capture()在这个例子中，还包含了pillow和pyautogui的实现，如果想进行一下测试，需要调整帧率以获得100%完美的视频再现速度。
到此这篇关于python实现自动化录制屏幕并生成视频的文章就介绍到这了,更多相关python自动化录屏内容请搜索代码网以前的文章或继续浏览下面的相关文章希望大家以后多多支持代码网！
发表评论