feat: add cli playwright as an entry point. (microsoft#388)

pavelfeldman · web-flow · commit dc8b54b79aac · 2021-01-19T12:38:47.000-08:00
diff --git a/playwright/__main__.py b/playwright/__main__.py
@@ -18,9 +18,15 @@
 
 from playwright._impl._driver import compute_driver_executable
 
-driver_executable = compute_driver_executable()
-my_env = os.environ.copy()
-# VSCode's JavaScript Debug Terminal provides it but driver/pkg does not support it
-my_env.pop("NODE_OPTIONS", None)
-my_env["PW_CLI_TARGET_LANG"] = "python"
-subprocess.run([str(driver_executable), *sys.argv[1:]], env=my_env)
+
+def main() -> None:
+    driver_executable = compute_driver_executable()
+    my_env = os.environ.copy()
+    # VSCode's JavaScript Debug Terminal provides it but driver/pkg does not support it
+    my_env.pop("NODE_OPTIONS", None)
+    my_env["PW_CLI_TARGET_LANG"] = "python"
+    subprocess.run([str(driver_executable), *sys.argv[1:]], env=my_env)
+
+
+if __name__ == "__main__":
+    main()
diff --git a/setup.py b/setup.py
@@ -135,4 +135,9 @@ def run(self) -> None:
         "write_to_template": 'version = "{version}"\n',
     },
     setup_requires=["setuptools_scm"],
+    entry_points={
+        "console_scripts": [
+            "playwright=playwright.__main__:main",
+        ],
+    },
 )
