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})")