126 lines
4.7 KiB
Python
126 lines
4.7 KiB
Python
import sys
|
|
import asyncio
|
|
from PySide6.QtWidgets import (QApplication, QWidget, QVBoxLayout, QPushButton, QLineEdit, QTextEdit, QLabel)
|
|
from PySide6.QtCore import Qt, QEvent
|
|
from PySide6.QtGui import QTextCursor
|
|
from playwright.async_api import async_playwright
|
|
import qasync
|
|
|
|
class WebAutomationApp(QWidget):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.browser = None # 브라우저 객체를 저장할 변수
|
|
self.page = None # 페이지 객체를 저장할 변수
|
|
self.playwright_context = None # Playwright 컨텍스트 유지
|
|
self.initUI()
|
|
|
|
def initUI(self):
|
|
self.setWindowTitle('Web Automation with Playwright')
|
|
|
|
# 레이아웃 설정
|
|
layout = QVBoxLayout()
|
|
|
|
# 입력창
|
|
self.url_input = QLineEdit(self)
|
|
self.url_input.setPlaceholderText("URL을 입력하세요 (예: https://www.naver.com)")
|
|
layout.addWidget(QLabel("URL 입력:"))
|
|
layout.addWidget(self.url_input)
|
|
|
|
# 로그 출력 창
|
|
self.log_output = QTextEdit(self)
|
|
self.log_output.setReadOnly(True)
|
|
layout.addWidget(QLabel("로그 출력:"))
|
|
layout.addWidget(self.log_output)
|
|
|
|
# 네이버 연결 버튼
|
|
self.start_button = QPushButton("네이버 열기", self)
|
|
self.start_button.clicked.connect(self.on_start_button_click)
|
|
layout.addWidget(self.start_button)
|
|
|
|
# 브라우저 닫기 버튼
|
|
self.close_browser_button = QPushButton("브라우저 닫기", self)
|
|
self.close_browser_button.clicked.connect(self.on_close_browser_click)
|
|
layout.addWidget(self.close_browser_button)
|
|
|
|
# 종료 버튼
|
|
self.exit_button = QPushButton("종료", self)
|
|
self.exit_button.clicked.connect(self.close_app)
|
|
layout.addWidget(self.exit_button)
|
|
|
|
self.setLayout(layout)
|
|
|
|
async def open_naver(self, url):
|
|
# Playwright를 이용해 네이버 웹페이지 열기
|
|
self.playwright_context = await async_playwright().start()
|
|
self.log_message("Playwright 시작 중...")
|
|
self.browser = await self.playwright_context.chromium.launch(headless=False)
|
|
self.page = await self.browser.new_page()
|
|
await self.page.goto(url)
|
|
self.log_message(f"{url} 로 이동 중...")
|
|
|
|
# 브라우저가 닫힐 때 감지
|
|
self.browser.on("disconnected", lambda: asyncio.ensure_future(self.close_app()))
|
|
|
|
def log_message(self, message):
|
|
self.log_output.append(message)
|
|
self.log_output.moveCursor(QTextCursor.End)
|
|
|
|
def on_start_button_click(self):
|
|
url = self.url_input.text()
|
|
if url == "":
|
|
self.log_message("URL을 입력하세요.")
|
|
return
|
|
|
|
# 비동기 작업 시작
|
|
self.log_message(f"{url} 연결 시도 중...")
|
|
asyncio.ensure_future(self.start_playwright_task(url))
|
|
|
|
async def start_playwright_task(self, url):
|
|
try:
|
|
await self.open_naver(url)
|
|
except Exception as e:
|
|
self.log_message(f"오류 발생: {e}")
|
|
|
|
async def close_browser(self):
|
|
if self.browser:
|
|
await self.browser.close()
|
|
await self.playwright_context.stop()
|
|
self.log_message("브라우저 닫힘")
|
|
self.browser = None
|
|
else:
|
|
self.log_message("열려 있는 브라우저가 없습니다.")
|
|
|
|
def on_close_browser_click(self):
|
|
asyncio.ensure_future(self.close_browser())
|
|
|
|
def close_app(self):
|
|
self.log_message("프로그램 종료 중...")
|
|
asyncio.ensure_future(self.close_browser()) # 프로그램 종료 전 브라우저 닫기
|
|
QApplication.instance().quit() # 애플리케이션 종료
|
|
|
|
def closeEvent(self, event):
|
|
"""GUI 창이 닫힐 때 호출되는 함수"""
|
|
self.log_message("GUI 창이 닫히고 있습니다...")
|
|
asyncio.ensure_future(self.close_browser()) # GUI 종료 시 브라우저 닫기
|
|
QApplication.instance().quit() # 이벤트 루프 종료
|
|
|
|
async def main():
|
|
app = QApplication(sys.argv)
|
|
loop = qasync.QEventLoop(app) # qasync를 이용한 이벤트 루프 통합
|
|
asyncio.set_event_loop(loop) # asyncio 이벤트 루프 설정
|
|
|
|
main_window = WebAutomationApp()
|
|
main_window.show()
|
|
|
|
with loop: # 이벤트 루프 시작
|
|
try:
|
|
loop.run_forever() # 이벤트 루프를 종료하지 않고 계속 유지
|
|
except (KeyboardInterrupt, SystemExit):
|
|
print("이벤트 루프 종료 중...")
|
|
|
|
if __name__ == "__main__":
|
|
if sys.platform == "win32":
|
|
asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy()) # Windows에서 SelectorEventLoop 사용 설정
|
|
|
|
asyncio.run(main())
|