Added support for iframe context switching in all Playwright actions

Author: nasif-z

Framework/Built_In_Automation/Web/Playwright/BuiltInFunctions.py

@@ -46,6 +46,13 @@
 from Framework.Utilities.CommonUtil import passed_tag_list, failed_tag_list
 from . import locator as PlaywrightLocator
 
+def _get_frame_locator():
+    """Helper function to get current frame locator from shared variables."""
+    frame_locator = sr.Get_Shared_Variables("playwright_frame")
+    if frame_locator in failed_tag_list:
+        return None
+    return frame_locator
+
 #########################
 #                       #
 #    Global Variables   #
@@ -495,7 +502,7 @@ async def Click_Element(step_data):
                     right_click = True
 
         # Get element
-        locator = await PlaywrightLocator.Get_Element(step_data, current_page)
+        locator = await PlaywrightLocator.Get_Element(step_data, current_page, frame_locator=_get_frame_locator())
         if locator == "zeuz_failed":
             CommonUtil.ExecLog(sModuleInfo, "Element not found", 3)
             return "zeuz_failed"
@@ -609,7 +616,7 @@ async def Hover_Over_Element(step_data):
                 elif left_l == "timeout":
                     timeout = int(float(right_v) * 1000)
 
-        locator = await PlaywrightLocator.Get_Element(step_data, current_page)
+        locator = await PlaywrightLocator.Get_Element(step_data, current_page, frame_locator=_get_frame_locator())
         if locator == "zeuz_failed":
             CommonUtil.ExecLog(sModuleInfo, "Element not found", 3)
             return "zeuz_failed"
@@ -686,7 +693,7 @@ async def Enter_Text_In_Text_Box(step_data):
                 elif left_l == "timeout":
                     timeout = int(float(right.strip()) * 1000)
 
-        locator = await PlaywrightLocator.Get_Element(step_data, current_page)
+        locator = await PlaywrightLocator.Get_Element(step_data, current_page, frame_locator=_get_frame_locator())
         if locator == "zeuz_failed":
             CommonUtil.ExecLog(sModuleInfo, "Element not found", 3)
             return "zeuz_failed"
@@ -830,7 +837,7 @@ async def Keystroke_For_Element(step_data):
                 key = key_map.get(key, keystroke_value)
 
             if has_element:
-                locator = await PlaywrightLocator.Get_Element(step_data, current_page)
+                locator = await PlaywrightLocator.Get_Element(step_data, current_page, frame_locator=_get_frame_locator())
                 if locator == "zeuz_failed":
                     CommonUtil.ExecLog(sModuleInfo, "Element not found", 3)
                     return "zeuz_failed"
@@ -853,7 +860,7 @@ async def Keystroke_For_Element(step_data):
                 type_options["delay"] = int(delay * 1000)
 
             if has_element:
-                locator = await PlaywrightLocator.Get_Element(step_data, current_page)
+                locator = await PlaywrightLocator.Get_Element(step_data, current_page, frame_locator=_get_frame_locator())
                 if locator == "zeuz_failed":
                     CommonUtil.ExecLog(sModuleInfo, "Element not found", 3)
                     return "zeuz_failed"
@@ -930,7 +937,7 @@ async def Validate_Text(step_data):
                 if left_l == "timeout":
                     timeout = int(float(right_v) * 1000)
 
-        locator = await PlaywrightLocator.Get_Element(step_data, current_page)
+        locator = await PlaywrightLocator.Get_Element(step_data, current_page, frame_locator=_get_frame_locator())
         if locator == "zeuz_failed":
             CommonUtil.ExecLog(sModuleInfo, "Element not found", 3)
             return "zeuz_failed"
@@ -995,7 +1002,7 @@ async def if_element_exists(step_data):
             if mid_l == "optional parameter" and left_l == "timeout":
                 timeout = int(float(right_v) * 1000)
 
-        locator = await PlaywrightLocator.Get_Element(step_data, current_page, element_wait=timeout/1000)
+        locator = await PlaywrightLocator.Get_Element(step_data, current_page, element_wait=timeout/1000, frame_locator=_get_frame_locator())
 
         if locator == "zeuz_failed":
             CommonUtil.ExecLog(sModuleInfo, "Element does not exist", 1)
@@ -1069,7 +1076,7 @@ async def Save_Attribute(step_data):
             CommonUtil.ExecLog(sModuleInfo, "No save variable specified", 3)
             return "zeuz_failed"
 
-        locator = await PlaywrightLocator.Get_Element(step_data, current_page)
+        locator = await PlaywrightLocator.Get_Element(step_data, current_page, frame_locator=_get_frame_locator())
         if locator == "zeuz_failed":
             CommonUtil.ExecLog(sModuleInfo, "Element not found", 3)
             return "zeuz_failed"
@@ -1131,7 +1138,7 @@ async def get_element_info(step_data):
             if mid.strip().lower() == "save parameter":
                 save_variable = left.strip()
 
-        locator = await PlaywrightLocator.Get_Element(step_data, current_page)
+        locator = await PlaywrightLocator.Get_Element(step_data, current_page, frame_locator=_get_frame_locator())
         if locator == "zeuz_failed":
             CommonUtil.ExecLog(sModuleInfo, "Element not found", 3)
             return "zeuz_failed"
@@ -1363,7 +1370,7 @@ async def scroll_to_element(step_data):
                 elif left_l == "align to top":
                     align_to_top = right_v.lower() in ("true", "yes", "1")
 
-        locator = await PlaywrightLocator.Get_Element(step_data, current_page)
+        locator = await PlaywrightLocator.Get_Element(step_data, current_page, frame_locator=_get_frame_locator())
         if locator == "zeuz_failed":
             CommonUtil.ExecLog(sModuleInfo, "Element not found", 3)
             return "zeuz_failed"
@@ -1443,7 +1450,7 @@ async def Select_Deselect(step_data):
             CommonUtil.ExecLog(sModuleInfo, "No selection value provided", 3)
             return "zeuz_failed"
 
-        locator = await PlaywrightLocator.Get_Element(step_data, current_page)
+        locator = await PlaywrightLocator.Get_Element(step_data, current_page, frame_locator=_get_frame_locator())
         if locator == "zeuz_failed":
             CommonUtil.ExecLog(sModuleInfo, "Element not found", 3)
             return "zeuz_failed"
@@ -1521,7 +1528,7 @@ async def check_uncheck(step_data):
                 if left_l == "use js":
                     use_js = right_v in ("true", "yes", "1")
 
-        locator = await PlaywrightLocator.Get_Element(step_data, current_page)
+        locator = await PlaywrightLocator.Get_Element(step_data, current_page, frame_locator=_get_frame_locator())
         if locator == "zeuz_failed":
             CommonUtil.ExecLog(sModuleInfo, "Element not found", 3)
             return "zeuz_failed"
@@ -1758,7 +1765,7 @@ def close_tab(step_data):
 #########################
 
 @logger
-def switch_iframe(step_data):
+async def switch_iframe(step_data):
     """
     Switch to an iframe or back to main content.
 
@@ -1806,6 +1813,7 @@ def switch_iframe(step_data):
         if switch_to_default:
             # In Playwright, we work with the main page directly
             # Store a flag or reset frame locator
+            sr.Set_Shared_Variables("playwright_frame", None)
             CommonUtil.ExecLog(sModuleInfo, "Switched to default content", 1)
             return "passed"
 
@@ -1974,13 +1982,13 @@ async def drag_and_drop(step_data):
                     source_param = left.strip()
 
         # Get source element
-        source_locator = await PlaywrightLocator.Get_Element(source_param, current_page)
+        source_locator = await PlaywrightLocator.Get_Element(source_param, current_page, frame_locator=_get_frame_locator())
         if source_locator == "zeuz_failed":
             CommonUtil.ExecLog(sModuleInfo, "Source element not found", 3)
             return "zeuz_failed"
 
         # Get target element
-        target_locator = await PlaywrightLocator.Get_Element(target_param, current_page)
+        target_locator = await PlaywrightLocator.Get_Element(target_param, current_page, frame_locator=_get_frame_locator())
         if target_locator == "zeuz_failed":
             CommonUtil.ExecLog(sModuleInfo, "Target element not found", 3)
             return "zeuz_failed"
@@ -2053,7 +2061,7 @@ async def take_screenshot_playwright(step_data):
 
         # Take screenshot
         if has_element:
-            locator = await PlaywrightLocator.Get_Element(step_data, current_page)
+            locator = await PlaywrightLocator.Get_Element(step_data, current_page, frame_locator=_get_frame_locator())
             if locator == "zeuz_failed":
                 CommonUtil.ExecLog(sModuleInfo, "Element not found", 3)
                 return "zeuz_failed"
@@ -2123,7 +2131,7 @@ async def execute_javascript(step_data):
 
         # Execute JS
         if has_element:
-            locator = await PlaywrightLocator.Get_Element(step_data, current_page)
+            locator = await PlaywrightLocator.Get_Element(step_data, current_page, frame_locator=_get_frame_locator())
             if locator == "zeuz_failed":
                 CommonUtil.ExecLog(sModuleInfo, "Element not found", 3)
                 return "zeuz_failed"
@@ -2193,7 +2201,7 @@ async def upload_file(step_data):
                 CommonUtil.ExecLog(sModuleInfo, f"File not found: {file_path}", 3)
                 return "zeuz_failed"
 
-        locator = await PlaywrightLocator.Get_Element(step_data, current_page)
+        locator = await PlaywrightLocator.Get_Element(step_data, current_page, frame_locator=_get_frame_locator())
         if locator == "zeuz_failed":
             CommonUtil.ExecLog(sModuleInfo, "Element not found", 3)
             return "zeuz_failed"
@@ -2320,7 +2328,7 @@ async def Wait_For_Element(step_data):
         if timeout:
             await asyncio.sleep(timeout)
 
-        locator = await PlaywrightLocator.Get_Element(step_data, current_page)
+        locator = await PlaywrightLocator.Get_Element(step_data, current_page, frame_locator=_get_frame_locator())
 
         if locator == "zeuz_failed":
             # For hidden/detached states, element not found is actually success
@@ -2539,7 +2547,7 @@ async def Extract_Table_Data(step_data):
                 elif left_l == "column":
                     col_filter = right_v
 
-        locator = await PlaywrightLocator.Get_Element(step_data, current_page)
+        locator = await PlaywrightLocator.Get_Element(step_data, current_page, frame_locator=_get_frame_locator())
         if locator == "zeuz_failed":
             CommonUtil.ExecLog(sModuleInfo, "Table element not found", 3)
             return "zeuz_failed"

Framework/Built_In_Automation/Web/Playwright/locator.py

@@ -25,7 +25,7 @@
 MODULE_NAME = inspect.getmodulename(__file__)
 
 
-async def Get_Element(step_data, page, return_all=False, element_wait=None):
+async def Get_Element(step_data, page, return_all=False, element_wait=None, frame_locator=None):
     """
     Get element using Playwright's native Locator API.
 
@@ -38,6 +38,7 @@ async def Get_Element(step_data, page, return_all=False, element_wait=None):
         page: Playwright Page object
         return_all: If True, return list of all matching ElementHandles
         element_wait: Override default wait timeout (in seconds)
+        frame_locator: Optional frame locator for iframe context
 
     Returns:
         Locator | List[ElementHandle] | "zeuz_failed"
@@ -75,7 +76,7 @@ async def Get_Element(step_data, page, return_all=False, element_wait=None):
                 return "zeuz_failed"
 
         # Build the locator
-        locator = _build_locator(page, step_data, params)
+        locator = _build_locator(page, step_data, params, frame_locator)
 
         if locator is None:
             CommonUtil.ExecLog(sModuleInfo, "Could not build locator from step data", 3)
@@ -230,23 +231,32 @@ def _parse_element_params(step_data):
     return params
 
 
-def _build_locator(page, step_data, params):
+def _build_locator(page, step_data, params, frame_locator=None):
     """
     Build a Playwright Locator from step data.
 
     Attempts these strategies in order:
     1. Playwright-native selectors (test-id, role, text, etc.) - fastest
     2. Direct xpath/css if provided
     3. Build xpath from element parameters using existing logic
+    
+    Args:
+        page: Playwright Page object
+        step_data: Step data for building xpath
+        params: Parsed element parameters
+        frame_locator: Optional frame locator for iframe context
     """
 
+    # Use frame locator if provided, otherwise use page
+    base_locator = frame_locator if frame_locator else page
+
     # Strategy 1: Check for Playwright-native selectors (fastest path)
     for left, right in params['element_params']:
         left_lower = left.lower()
 
         # Test ID selectors
         if left_lower in ("test-id", "testid", "data-testid", "data-test-id"):
-            return page.get_by_test_id(right)
+            return base_locator.get_by_test_id(right)
 
         # Role selector
         if left_lower == "role":
@@ -257,54 +267,54 @@ def _build_locator(page, step_data, params):
                     name = r
                     break
             if name:
-                return page.get_by_role(right, name=name)
-            return page.get_by_role(right)
+                return base_locator.get_by_role(right, name=name)
+            return base_locator.get_by_role(right)
 
         # Text selectors
         if left_lower == "text":
-            return page.get_by_text(right, exact=True)
+            return base_locator.get_by_text(right, exact=True)
         if left_lower == "*text":
-            return page.get_by_text(right, exact=False)
+            return base_locator.get_by_text(right, exact=False)
         if left_lower == "**text":
             # Case-insensitive partial match
-            return page.get_by_text(re.compile(re.escape(right), re.IGNORECASE))
+            return base_locator.get_by_text(re.compile(re.escape(right), re.IGNORECASE))
 
         # Label selector
         if left_lower == "label":
-            return page.get_by_label(right)
+            return base_locator.get_by_label(right)
 
         # Placeholder selector
         if left_lower == "placeholder":
-            return page.get_by_placeholder(right)
+            return base_locator.get_by_placeholder(right)
 
         # Alt text selector
         if left_lower in ("alt", "alt text", "alt-text"):
-            return page.get_by_alt_text(right)
+            return base_locator.get_by_alt_text(right)
 
         # Title selector
         if left_lower == "title" and "parameter" not in params.get('mid', ''):
-            return page.get_by_title(right)
+            return base_locator.get_by_title(right)
 
         # Direct xpath
         if left_lower == "xpath":
-            return page.locator(f"xpath={right}")
+            return base_locator.locator(f"xpath={right}")
 
         # Direct CSS selector
         if left_lower in ("css", "css selector", "css_selector"):
-            return page.locator(right)
+            return base_locator.locator(right)
 
     # Strategy 2: Check for unique parameters
     for left, right in params['unique_params']:
         left_lower = left.lower()
 
         if left_lower == "id":
-            return page.locator(f"#{right}")
+            return base_locator.locator(f"#{right}")
         elif left_lower == "name":
-            return page.locator(f"[name='{right}']")
+            return base_locator.locator(f"[name='{right}']")
         elif left_lower == "class":
-            return page.locator(f".{right}")
+            return base_locator.locator(f".{right}")
         elif left_lower == "tag":
-            return page.locator(right)
+            return base_locator.locator(right)
 
     # Strategy 3: Build xpath from element/parent/child parameters
     xpath = _build_xpath_from_params(step_data, params)
@@ -314,7 +324,7 @@ def _build_locator(page, step_data, params):
             f"Built xpath from parameters: {xpath}",
             5
         )
-        return page.locator(f"xpath={xpath}")
+        return base_locator.locator(f"xpath={xpath}")
 
     # Strategy 4: Simple element parameters as xpath
     if params['element_params']:
@@ -348,9 +358,9 @@ def _build_locator(page, step_data, params):
 
         if xpath_parts:
             xpath = f"//{tag}[{' and '.join(xpath_parts)}]"
-            return page.locator(f"xpath={xpath}")
+            return base_locator.locator(f"xpath={xpath}")
         elif tag != "*":
-            return page.locator(tag)
+            return base_locator.locator(tag)
 
     return None