feat: support dynamic browser name (#748)

Author: kumaraditya303

playwright/async_api/_generated.py

@@ -9930,6 +9930,15 @@ def stop(self) -> NoneType:
 
         return mapping.from_maybe_impl(self._impl_obj.stop())
 
+    def __getitem__(self, value: str) -> "BrowserType":
+        if value == "chromium":
+            return self.chromium
+        elif value == "firefox":
+            return self.firefox
+        elif value == "webkit":
+            return self.webkit
+        raise ValueError("Invalid browser " + value)
+
 
 mapping.register(PlaywrightImpl, Playwright)
 

playwright/sync_api/_generated.py

@@ -9872,6 +9872,15 @@ def stop(self) -> NoneType:
 
         return mapping.from_maybe_impl(self._impl_obj.stop())
 
+    def __getitem__(self, value: str) -> "BrowserType":
+        if value == "chromium":
+            return self.chromium
+        elif value == "firefox":
+            return self.firefox
+        elif value == "webkit":
+            return self.webkit
+        raise ValueError("Invalid browser " + value)
+
 
 mapping.register(PlaywrightImpl, Playwright)
 

scripts/generate_async_api.py

@@ -115,7 +115,19 @@ def generate(t: Any) -> None:
                     f"""
         return {prefix}{arguments(value, len(prefix))}{suffix}"""
                 )
-
+    if class_name == "Playwright":
+        print(
+            """
+    def __getitem__(self, value: str) -> "BrowserType":
+        if value == "chromium":
+            return self.chromium
+        elif value == "firefox":
+            return self.firefox
+        elif value == "webkit":
+            return self.webkit
+        raise ValueError("Invalid browser "+value)
+            """
+        )
     print("")
     print(f"mapping.register({class_name}Impl, {class_name})")
 

scripts/generate_sync_api.py

@@ -114,7 +114,19 @@ def generate(t: Any) -> None:
                     f"""
         return {prefix}{arguments(value, len(prefix))}{suffix}"""
                 )
-
+    if class_name == "Playwright":
+        print(
+            """
+    def __getitem__(self, value: str) -> "BrowserType":
+        if value == "chromium":
+            return self.chromium
+        elif value == "firefox":
+            return self.firefox
+        elif value == "webkit":
+            return self.webkit
+        raise ValueError("Invalid browser "+value)
+            """
+        )
     print("")
     print(f"mapping.register({class_name}Impl, {class_name})")