[python] 테스트 실행 중 화면 비디오 녹화하기

이 글에서는 Python을 사용하여 테스트 실행 중 화면을 비디오로 녹화하는 방법에 대해 알아보겠습니다.

녹화하기 위한 라이브러리 설치

먼저, 화면 녹화를 위해 pyautoguiopencv-python이라는 라이브러리를 설치해야 합니다. 아래의 명령어를 사용하여 설치해주세요.

pip install pyautogui opencv-python

코드 작성

이제 실제로 화면을 녹화하는 코드를 작성해보겠습니다. 아래의 예제 코드를 참고해주세요.

import cv2
import pyautogui

# 화면 크기 설정
screen_size = (1920, 1080)

# 비디오 저장할 경로 및 이름 설정
output = "recording.avi"

# 비디오 코덱 설정
fourcc = cv2.VideoWriter_fourcc(*"XVID")

# 비디오 저장 객체 생성
video_writer = cv2.VideoWriter(output, fourcc, 20.0, screen_size)

while True:
    # 화면 캡처
    image = pyautogui.screenshot()
    
    # 이미지 프레임으로 변환
    frame = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
    
    # 비디오에 프레임 추가
    video.write(frame)
    
    # escape 키를 누르면 반복문 종료
    if cv2.waitKey(1) == 27:
        break

# 객체 해제
video.release()
cv2.destroyAllWindows()

코드 설명

위의 코드를 간단히 설명하면 다음과 같습니다.

  1. 필요한 라이브러리를 임포트합니다.
  2. 화면 크기와 저장될 비디오 파일의 경로, 코덱 정보를 설정합니다.
  3. 비디오 저장 객체를 생성합니다.
  4. 계속해서 화면을 캡처하여 비디오에 추가합니다.
  5. ESC 키를 누르면 녹화를 종료합니다.
  6. 객체를 해제하고 창을 닫습니다.

실행하기

작성한 코드를 실행하기 전에, 테스트를 원하는 화면으로 이동한 뒤 아래의 명령어를 실행하여 화면을 녹화해보세요.

python recording.py

화면이 녹화되는 동안 recording.avi 파일이 생성될 것입니다. 녹화를 중지하려면 ESC 키를 누르세요.

결론

위의 코드를 통해 Python으로 테스트 실행 중 화면을 비디오로 녹화하는 방법을 배웠습니다. 이를 활용하여 테스트 수행 결과를 저장하거나 제품 데모 영상을 만들어 볼 수 있습니다.