add cli for installing browsers (#20)

Author: coffecup25

dendrite_sdk/_cli/main.py

@@ -0,0 +1,31 @@
+import argparse
+import subprocess
+import sys
+
+
+def run_playwright_install():
+    try:
+        subprocess.run(["playwright", "install"], check=True)
+        print("Playwright browser installation completed successfully.")
+    except subprocess.CalledProcessError as e:
+        print(f"Error during Playwright browser installation: {e}")
+        sys.exit(1)
+    except FileNotFoundError:
+        print(
+            "Playwright command not found. Please ensure Playwright is installed in your environment."
+        )
+        sys.exit(1)
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Dendrite SDK CLI tool")
+    parser.add_argument("command", choices=["install"], help="Command to execute")
+
+    args = parser.parse_args()
+
+    if args.command == "install":
+        run_playwright_install()
+
+
+if __name__ == "__main__":
+    main()

pyproject.toml

@@ -39,6 +39,8 @@ autopep8 = "^2.0.4"
 pylint = "^3.2.6"
 black = "^24.8.0"
 
+[tool.poetry.scripts]
+dendrite = "dendrite_sdk._cli.main:main"
 
 [build-system]
 requires = ["poetry-core", "requests", "python-dotenv", "pydantic"]