fix: update dom handling code (#184)

* fix: update dom handling code * Edit comment
eggplants · Oct 20, 2023 · d1c7060 · d1c7060
1 parent 8a6d684
commit d1c7060
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/deepl/deepl.py b/deepl/deepl.py
@@ -128,6 +128,8 @@ async def __translate(self, script: str) -> str:
                 raise DeepLCLIPageLoadError(msg) from e
 
             if self.use_dom_submit:
+                # banner prevents clicking on language buttons, close the banner first
+                await page.click("button[data-testid=cookie-banner-lax-close-button]")
                 # select input / output language
                 await page.click("button[data-testid=translator-source-lang-btn]")
                 await page.click(f"button[data-testid=translator-lang-option-{self.fr_lang}]")
@@ -141,7 +143,7 @@ async def __translate(self, script: str) -> str:
                 await page.wait_for_function(
                     """
                     () => document.querySelector(
-                    'd-textarea[aria-labelledby=translation-results-heading]')?.value?.length > 0
+                    'd-textarea[aria-labelledby=translation-target-heading]')?.value?.length > 0
                     """,
                 )
             except PlaywrightError as e: