화면 캡처 프로그램¶

PyAutoGUI 라이브러리의 스크릿샷 기능을 이용해서 주어진 시간 간격과 횟수에 따라 화면을 캡처하는 프로그램을 만들어 보겠습니다.

예제¶

## Ex 10-4. 화면 캡처 프로그램.

import sys
from PyQt5.QtWidgets import *
from PyQt5.QtCore import QTimer
import pyautogui
import datetime


class MyApp(QWidget):

    def __init__(self):
        super().__init__()
        self.delay_le = QLineEdit(self)
        self.num_le = QLineEdit(self)
        self.start_btn = QPushButton('Capture', self)

        self.delay, self.num_cap = 0, 0
        self.initUI()

    def initUI(self):
        self.delay_le.setPlaceholderText('캡처 간격 (초)')
        self.num_le.setPlaceholderText('캡처 횟수')
        self.start_btn.clicked.connect(self.start_btn_click)

        hbox = QHBoxLayout()
        hbox.addWidget(self.delay_le)
        hbox.addWidget(self.num_le)

        vbox = QVBoxLayout()
        vbox.addLayout(hbox)
        vbox.addWidget(self.start_btn)

        self.setLayout(vbox)

        self.setWindowTitle('Capture')
        self.setGeometry(200, 200, 300, 150)
        self.show()

    def start_btn_click(self):
        self.delay = int(self.delay_le.text())
        self.num_cap = 0
        self.timer = QTimer()
        self.timer.start(self.delay * 1000)
        self.timer.timeout.connect(self.capture)

    def capture(self):
        today = str(datetime.datetime.today())
        pyautogui.screenshot(today + '.png')
        self.num_cap += 1

        if self.num_cap == int(self.num_le.text()):
            self.timer.stop()


if __name__ == '__main__':
    app = QApplication(sys.argv)
    ex = MyApp()
    sys.exit(app.exec_())

설명¶

self.delay_le = QLineEdit(self)
self.num_le = QLineEdit(self)
self.start_btn = QPushButton('Capture', self)

self.delay, self.num_cap = 0, 0

캡처 사이의 시간 간격과 캡처 횟수를 입력받을 줄 편집기, 그리고 시작 버튼을 만듭니다.

self.delay와 self.num_cap은 시간 간격와 캡처 횟수를 저장하는 변수입니다.

self.delay_le.setPlaceholderText('캡처 간격 (초)')
self.num_le.setPlaceholderText('캡처 횟수')
self.start_btn.clicked.connect(self.start_btn_click)

setPlaceholderText() 메서드는 줄 편집기 위젯에 표시할 텍스트를 설정합니다.

시작 버튼을 클릭하면 self.start_btn_click() 메서드가 호출되도록 합니다.

hbox = QHBoxLayout()
hbox.addWidget(self.delay_le)
hbox.addWidget(self.num_le)

vbox = QVBoxLayout()
vbox.addLayout(hbox)
vbox.addWidget(self.start_btn)

self.setLayout(vbox)

위젯들을 배치하기 위해 박스 레이아웃을 사용합니다.

QHBoxLayout()을 이용해서 수평 박스를 하나 만들고 두 개의 줄 편집기 위젯을 배치합니다.

QVBoxLayout()을 이용해서 수직 박스를 하나 만들어서 수직 박스 (hbox)와 시작 버튼 (self.start_btn)을 수직으로 배치합니다.

그리고 setLayout()을 이용해서 이 수직 박스 (vbox)를 전체 레이아웃으로 설정합니다.

(참고: PyQt5 레이아웃 - 박스 레이아웃)

def start_btn_click(self):
    self.delay = int(self.delay_le.text())
    self.num_cap = 0
    self.timer = QTimer()
    self.timer.start(self.delay * 1000)
    self.timer.timeout.connect(self.capture)

앞의 예제에서와 같이 타이머를 하나 만들고, 입력한 딜레이 값을 가져와서 self.delay에 저장합니다.

start()를 이용해서 self.delay (초) 마다 self.capture 함수가 반복해서 호출되도록 합니다.

def capture(self):
    today = str(datetime.datetime.today())
    pyautogui.screenshot(today + '.png')
    self.num_cap += 1

    if self.num_cap == int(self.num_le.text()):
        self.timer.stop()