Add Context Manager frame_switch(frame). ("with" required)

Author: mdmintz

help_docs/method_summary.md

@@ -209,6 +209,9 @@ self.switch_to_default_content()
 
 self.switch_to_parent_frame()
 
+with self.frame_switch(frame, timeout=None):
+    # Indented Code Block for Context Manager (Must use "with")
+
 self.set_content_to_frame(frame, timeout=None)
 
 self.set_content_to_default(nested=False)

seleniumbase/fixtures/base_case.py

@@ -42,6 +42,7 @@ def test_anything(self):
 import time
 import unittest
 import urllib3
+from contextlib import contextmanager
 from selenium.common.exceptions import (
     ElementClickInterceptedException as ECI_Exception,
     ElementNotInteractableException as ENI_Exception,
@@ -2775,6 +2776,18 @@ def switch_to_parent_frame(self):
                         return
         self.driver.switch_to.parent_frame()
 
+    @contextmanager
+    def frame_switch(self, frame, timeout=None):
+        """ Context Manager for switching into iframes.
+        Usage example:
+            with self.frame_switch("iframe"):
+                # Perform actions here that should be done within the iframe.
+            # The iframe is automatically exited after the "with" block ends.
+        """
+        self.switch_to_frame(frame, timeout=timeout)
+        yield
+        self.switch_to_parent_frame()
+
     def set_content_to_frame(self, frame, timeout=None):
         """Replaces the page html with an iframe's html from that page.
         If the iframe contains an "src" field that includes a valid URL,