111 lines
5.9 KiB
Python
111 lines
5.9 KiB
Python
import os
|
|
import asyncio
|
|
from playwright.async_api import async_playwright
|
|
import time
|
|
|
|
class ThumbnailHandler:
|
|
def __init__(self, locator_manager, browser_controller, logger, whale_translator, clipboardImageManager, toggle_states):
|
|
self.locator_manager = locator_manager
|
|
self.browser_controller = browser_controller
|
|
self.page = self.browser_controller.page
|
|
self.whale_translator = whale_translator
|
|
self.clipboardImageManager = clipboardImageManager
|
|
self.toggle_states = toggle_states
|
|
self.logger = logger
|
|
self.thumbnail_selector_template = "div#productMainContentContainerId div:nth-child({index}) > div > div:nth-child(2) > div > img"
|
|
self.delete_button_selector_template = "div#productMainContentContainerId div:nth-child({index}) > div > div.ant-row.ant-row-no-wrap.ant-row-space-between.ant-row-middle.css-1li46mu > div:nth-child(1) > div > span"
|
|
# self.delete_button_selector_for_Num1 = "div#productMainContentContainerId div:nth-child(1) > div > div.ant-row.ant-row-no-wrap.ant-row-space-between.ant-row-middle.css-1li46mu > div:nth-child(1) > div > span"
|
|
self.upload_button_selector = "div#productMainContentContainerId div:nth-child({index}) > div > div:nth-child(2) > span"
|
|
self.file_upload_button_selector = 'span.ant-upload-btn input[type="file"]'
|
|
|
|
self.confirm_upload_button_selector = "div.ant-modal-footer > button[type=\"button\"].ant-btn.css-1li46mu.ant-btn-primary"
|
|
|
|
# main.py 실행 폴더의 tmp_images 폴더 경로 설정
|
|
base_dir = os.path.dirname(os.path.abspath(__file__)) # main.py 위치 확인
|
|
self.temp_dir = os.path.join(base_dir, "tmp_images") # tmp_images 폴더 경로 설정
|
|
os.makedirs(self.temp_dir, exist_ok=True) # 폴더가 없으면 생성
|
|
|
|
def update_page(self, page1):
|
|
self.page = page1
|
|
self.logger.debug(f"page객체 업데이트 : {page1}")
|
|
|
|
def update_whale(self):
|
|
self.whale_translator = self.browser_controller.get_whale()
|
|
self.logger.debug(f"whale_translator 업데이트 : {self.whale_translator}")
|
|
|
|
async def process_thumbnails(self):
|
|
self.update_whale()
|
|
|
|
# 썸네일 카드 개수 수집
|
|
thumbnails = await self.page.query_selector_all("div#productMainContentContainerId div.ant-row.ant-row-bottom.css-1li46mu > div")
|
|
total_thumbnails = len(thumbnails)
|
|
self.logger.debug(f"총 썸네일 카드 수집 완료: {total_thumbnails}개")
|
|
|
|
for index in range(total_thumbnails-1):
|
|
|
|
thumbnail_img_selector = self.thumbnail_selector_template.format(index=1)
|
|
delete_button_selector = self.delete_button_selector_template.format(index=1) # 첫번째카드만 삭제를 반복하면 되므로
|
|
|
|
# 이미지 URL 수집
|
|
thumbnail_img_element = await self.page.query_selector(thumbnail_img_selector)
|
|
if thumbnail_img_element:
|
|
image_url = await thumbnail_img_element.get_attribute("src")
|
|
self.logger.debug(f"{index}번째 썸네일 이미지 URL: {image_url}")
|
|
|
|
# 이미지 번역 실행
|
|
translated_image_path = await self.translate_thumbnail_image(image_url, index)
|
|
|
|
# 기존 썸네일 삭제
|
|
await self.delete_thumbnail(delete_button_selector)
|
|
|
|
# 번역된 이미지 업로드
|
|
await self.upload_translated_image(translated_image_path, thumbnails)
|
|
|
|
async def translate_thumbnail_image(self, image_url, index):
|
|
try:
|
|
# 이미지 번역 수행 및 임시 저장 경로 반환
|
|
translated_image_path = os.path.join(self.temp_dir, f"translated_thumb_{index}.png") # 이미지 저장 경로 설정
|
|
|
|
is_success_translated = self.whale_translator.translate_image(image_url)
|
|
self.clipboardImageManager.process_clipboard(image_url, is_success_translated, self.toggle_states, translated_image_path, is_thumb=True)
|
|
self.logger.debug(f"{index}번째 썸네일 이미지 번역 완료: {translated_image_path}")
|
|
self.browser_controller.switch_to_chrome() # 크롬으로 포커스 이동
|
|
|
|
return translated_image_path
|
|
except Exception as e:
|
|
self.logger.error(f"{index}번째 썸네일 이미지 번역 중 오류 발생: {e}", exc_info=True)
|
|
return None
|
|
|
|
async def delete_thumbnail(self, delete_button_selector):
|
|
try:
|
|
delete_button = await self.page.query_selector(delete_button_selector)
|
|
if delete_button:
|
|
await delete_button.click()
|
|
self.logger.debug("썸네일 삭제 버튼 클릭 완료")
|
|
except Exception as e:
|
|
self.logger.error(f"썸네일 삭제 중 오류 발생: {e}", exc_info=True)
|
|
|
|
async def upload_translated_image(self, image_path, thumbnails):
|
|
try:
|
|
# 썸네일 카드 목록의 마지막 요소가 업로드 카드
|
|
last_thumbnail = thumbnails[-1]
|
|
|
|
upload_button = await last_thumbnail.query_selector("span:has-text('Upload')")
|
|
|
|
if upload_button:
|
|
await upload_button.click()
|
|
self.logger.debug("업로드 버튼 클릭 완료")
|
|
|
|
# 파일 업로드 수행
|
|
file_input = await self.page.query_selector(self.file_upload_button_selector)
|
|
if file_input:
|
|
await file_input.set_input_files(image_path)
|
|
self.logger.debug("이미지 파일 업로드 완료")
|
|
|
|
# '이미지 삽입' 버튼 클릭
|
|
confirm_button = await self.page.query_selector(self.confirm_upload_button_selector)
|
|
await confirm_button.click()
|
|
self.logger.debug("이미지 삽입 버튼 클릭 완료")
|
|
except Exception as e:
|
|
self.logger.error(f"이미지 업로드 중 오류 발생: {e}", exc_info=True)
|