From b8392817fa507c08f64f0c30f2a8cedd038298af Mon Sep 17 00:00:00 2001
From: "zhiheng.hu" <zhiheng.hu@klook.com>
Date: Thu, 16 Feb 2023 11:37:22 +0800
Subject: [PATCH] feat: support DrissionPage CLI

---
 DrissionPage/cli.py | 15 +++++++++++++++
 setup.py            | 11 +++++++++--
 2 files changed, 24 insertions(+), 2 deletions(-)
 create mode 100644 DrissionPage/cli.py

diff --git a/DrissionPage/cli.py b/DrissionPage/cli.py
new file mode 100644
index 0000000..43eadd7
--- /dev/null
+++ b/DrissionPage/cli.py
@@ -0,0 +1,15 @@
+import click
+from DrissionPage.easy_set import set_paths
+
+
+@click.command()
+@click.option("-p", "--browser-path", help="Setting browser path.")
+def main(browser_path):
+    """DrissionPage CLI."""
+    if browser_path:
+        set_paths(browser_path=browser_path)
+        return 0
+
+
+if __name__ == '__main__':
+    main()
diff --git a/setup.py b/setup.py
index 4823d9a..7ba3aa3 100644
--- a/setup.py
+++ b/setup.py
@@ -17,6 +17,7 @@ setup(
     url="https://gitee.com/g1879/DrissionPage",
     include_package_data=True,
     packages=find_packages(),
+    zip_safe=False,
     install_requires=[
         "selenium",
         "lxml",
@@ -24,7 +25,8 @@ setup(
         "requests",
         "DownloadKit>=0.5.0",
         "FlowViewer",
-        "websocket-client"
+        "websocket-client",
+        'click~=8.1.3'
     ],
     classifiers=[
         "Programming Language :: Python :: 3.6",
@@ -32,5 +34,10 @@ setup(
         "Topic :: Utilities",
         "License :: OSI Approved :: BSD License",
     ],
-    python_requires='>=3.6'
+    python_requires='>=3.6',
+    entry_points={
+        'console_scripts': [
+            'dp = DrissionPage.cli:main',
+        ],
+    },
 )