mirror of
https://github.com/PR0M3TH3AN/SeedPass.git
synced 2026-10-27 07:52:03 +00:00
google-labs-jules[bot]andPR0M3TH3AN
8e33f9c5e1
- Apply black formatting to 17 files. - Fix F821 undefined name errors in: - src/main.py: Add missing `queue` import. - src/seedpass/core/encryption.py: Fix `JSONDecodeError` reference. - src/utils/__init__.py: Fix exception variable scope in fallback code. - src/utils/key_derivation.py: Add `TYPE_CHECKING` imports. - src/seedpass/core/config_manager.py: Add `TYPE_CHECKING` imports. - Initialize torch environment and TODO file. Co-authored-by: PR0M3TH3AN <[email protected]>
38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
from __future__ import annotations
|
|
|
|
import typer
|
|
import uvicorn
|
|
|
|
from .. import api as api_module
|
|
|
|
app = typer.Typer(help="Run the API server")
|
|
|
|
|
|
@app.command("start")
|
|
def api_start(ctx: typer.Context, host: str = "127.0.0.1", port: int = 8000) -> None:
|
|
"""Start the SeedPass API server."""
|
|
token = api_module.start_server(ctx.obj.get("fingerprint"))
|
|
typer.echo(
|
|
f"API token: {token}\nWARNING: Store this token securely; it cannot be recovered."
|
|
)
|
|
uvicorn.run(api_module.app, host=host, port=port)
|
|
|
|
|
|
@app.command("stop")
|
|
def api_stop(
|
|
token: str = typer.Option(..., help="API token"),
|
|
host: str = "127.0.0.1",
|
|
port: int = 8000,
|
|
) -> None:
|
|
"""Stop the SeedPass API server."""
|
|
import requests
|
|
|
|
try:
|
|
requests.post(
|
|
f"http://{host}:{port}/api/v1/shutdown",
|
|
headers={"Authorization": f"Bearer {token}"},
|
|
timeout=2,
|
|
)
|
|
except Exception as exc: # pragma: no cover - best effort
|
|
typer.echo(f"Failed to stop server: {exc}")
|