feat: support dynamic browser name (microsoft#748)

kumaraditya303 · web-flow · commit aa0b4677527e · 2021-06-27T22:04:43.000-07:00
diff --git a/playwright/async_api/_generated.py b/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)
 
diff --git a/playwright/sync_api/_generated.py b/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)
 
diff --git a/scripts/generate_async_api.py b/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})")
 
diff --git a/scripts/generate_sync_api.py b/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})")
 
