feat(api): CLI: add --host, --port

saveweb · Oct 30, 2023 · 9750899 · 9750899
1 parent 04edd49
commit 9750899
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/biliarchiver/cli_tools/biliarchiver.py b/biliarchiver/cli_tools/biliarchiver.py
@@ -82,7 +82,19 @@ def auth():
 
 
 @biliarchiver.command(help=click.style(_("运行 API"), fg="cyan"))
-def api():
+@click.option(
+    "--host",
+    type=str,
+    default="127.0.0.1",
+    show_default=True,
+)
+@click.option(
+    "--port",
+    type=int,
+    default=8000,
+    show_default=True,
+)
+def api(**kwargs):
     try:
         import fastapi
         import uvicorn
@@ -93,7 +105,7 @@ def api():
 
     from biliarchiver.rest_api.main import app
 
-    uvicorn.run(app)
+    uvicorn.run(app, **kwargs)
 
 
 if __name__ == "__main__":