|
|
import time |
|
|
from typing import Literal |
|
|
|
|
|
from playwright.sync_api import Playwright, sync_playwright |
|
|
|
|
|
from lib.pages import TranslatorPage |
|
|
from environment import DEBUG_PORT, RunType |
|
|
|
|
|
class PageRunner: |
|
|
def __init__(self, run_type: RunType): |
|
|
self.run_type = run_type |
|
|
|
|
|
def start(self, p: Playwright)-> TranslatorPage: |
|
|
if self.run_type == RunType.electron: |
|
|
return TranslatorPage(self._start_electron(p)) |
|
|
elif self.run_type == RunType.code: |
|
|
return TranslatorPage(self._start_web(p)) |
|
|
elif self.run_type == RunType.dev: |
|
|
return TranslatorPage(self._start_web(p)) |
|
|
else: |
|
|
raise TypeError(f"invalid run_type: {self.run_type}") |
|
|
|
|
|
def _start_web(self, p: Playwright): |
|
|
url = "http://127.0.0.1:9191/app" |
|
|
browser = p.chromium.launch(headless=False) |
|
|
context = browser.new_context(permissions=['microphone']) |
|
|
page = context.new_page() |
|
|
for i in range(20): |
|
|
try: |
|
|
page.goto(url) |
|
|
break |
|
|
except Exception as e: |
|
|
print(f"Exception happened: {e}. retry {i+1}...") |
|
|
time.sleep(3) |
|
|
return page |
|
|
|
|
|
def _start_electron(self, p: Playwright): |
|
|
|
|
|
browser = p.chromium.connect_over_cdp(f"http://localhost:{DEBUG_PORT}") |
|
|
assert browser.is_connected() |
|
|
print("connected to electron debugging port") |
|
|
context = browser.contexts[0] |
|
|
page = context.pages[0] |
|
|
return page |
|
|
|
|
|
if __name__ == '__main__': |
|
|
with sync_playwright() as p: |
|
|
page = PageRunner(RunType.code).start(p) |
|
|
page.start_zh2en() |
|
|
page.set_off() |
