| from datetime import datetime |
| from time import sleep |
| from typing import List |
| from playwright.sync_api import Page |
|
|
| from lib.log_parser import WebItem |
|
|
|
|
| class TranslatorPage: |
| def __init__(self, page:Page): |
| self.page = page |
|
|
| @property |
| def _switch(self): |
| return self.page.get_by_role("switch") |
|
|
| @property |
| def _trans_selector(self): |
| return self.page.locator(".ant-select-selector") |
|
|
| def start(self, translation_lang): |
| if translation_lang == "zh2en": |
| self.start_zh2en() |
| elif translation_lang == "en2zh": |
| self.start_en2zh() |
| else: |
| raise TypeError(f"invalid translation_lang: {translation_lang}") |
|
|
| def start_zh2en(self): |
| self._trans_selector.click() |
| self.page.locator(".ant-select-item").get_by_text("Chinese -> English").click() |
| self.set_on() |
| print("page started zh2en translation") |
|
|
| def start_en2zh(self): |
| self._trans_selector.click() |
| self.page.locator(".ant-select-item").get_by_text("English -> Chinese").click() |
| self.set_on() |
| print("page started en2zh translation") |
|
|
| def _is_on(self): |
| state = self._switch.get_attribute("aria-checked") |
| return True if state=="true" else False |
|
|
| def set_on(self): |
| if not self._is_on(): |
| self._switch.click() |
| print("click button to set translation on") |
|
|
| def set_off(self): |
| if self._is_on(): |
| self._switch.click() |
| print("click button to set translation off") |
|
|
|
|
| def get_current_node_text(self, duration=0, interval=0.05)-> List[WebItem]: |
| """在一定时间内持续读取页面最新的 node的内容""" |
| print(f"capture page latest content for duration: {duration}s") |
| translate_items = [] |
| last_src, last_dst = None, None |
| for i in range(int(duration // interval) + 1): |
| current_node = self.page.locator(".trans-list").locator(".current_node") |
| src_lang = current_node.locator(".trans-src-lang").inner_text() |
| dst_lang = current_node.locator(".trans-dst-lang").inner_text() |
| tsp = datetime.now() |
|
|
| sleep(interval) |
| if src_lang == last_src and dst_lang == last_dst: |
| continue |
| |
| |
| translate_items.append(WebItem(tsp, src_lang, dst_lang)) |
| last_src, last_dst = src_lang, dst_lang |
| print(f"capture page latest content finished") |
| return translate_items |
|
|
| def get_translated_texts(self, translation_lang): |
| src_all = self.page.locator(".trans-list").locator(".trans-src-lang").all_inner_texts() |
| dst_all = self.page.locator(".trans-list").locator(".trans-dst-lang").all_inner_texts() |
| |
| if translation_lang == "zh2en": |
| return "".join(src_all), " ".join(dst_all) |
| elif translation_lang == "en2zh": |
| return " ".join(src_all), "".join(dst_all) |
| return "".join(src_all), "".join(dst_all) |
|
|
|
|
|
|
