Merge pull request #12 from hhslepicka/handle_iframe

ENH: Add enter_iframe and leave_iframe to handle iFrames with WebBot.

botcity/web/bot.py

@@ -939,6 +939,19 @@ def set_file_input_element(self, element: WebElement, filepath: str):
         fpath = os.path.abspath(os.path.expanduser(os.path.expandvars(filepath)))
         element.send_keys(fpath)
 
+    def enter_iframe(self, iframe: WebElement):
+        """Switch the WebBot driver to the specified iframe.
+
+        Args:
+            iframe (WebElement): The desired iFrame.
+        """
+        self._driver.switch_to.frame(iframe)
+
+    def leave_iframe(self):
+        """Leave the iframe and switch the WebBot driver to the default content.
+        """
+        self._driver.switch_to.default_content()
+
     #######
     # Mouse
     #######