Small fix in iframe logic

yaroslav4167 · Sep 15, 2023 · aecf1e6 · aecf1e6
1 parent e6a43b6
commit aecf1e6
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,5 @@
+# Seleniumbase
+downloaded_files/
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

diff --git a/pyproject.toml b/pyproject.toml
@@ -17,7 +17,7 @@ name = "youchat-py"  # Required
 #
 # For a discussion on single-sourcing the version, see
 # https://packaging.python.org/guides/single-sourcing-package-version/
-version = "1.1.2"  # Required
+version = "1.1.3"  # Required
 
 # This is a one-line description or tagline of what your project does. This
 # corresponds to the "Summary" metadata field:

diff --git a/youchat.py b/youchat.py
@@ -36,16 +36,16 @@ def you_message(text: str, out_type: str = 'json'):
                 if sb.assert_element('iframe'):
                     sb.switch_to_frame("iframe")
                     sb.find_element(".ctp-checkbox-label", timeout=0.1).click()
+                    sb.switch_to_default_content()
                     # sb.save_screenshot('sel2.png') # Debug
             except Exception:
                 result['error'] = 'Selenium was detected! Try again later. Captcha not solved automaticly.'
 
             if stream_available:
                 result.pop('error')
-                sb.switch_to_default_content()
                 data = sb.get_text("body pre")
                 break
-            sb.switch_to_default_content()
+
 
         res_message = ""
         for line in data.split("\n"):