mirror of
https://github.com/trycua/computer.git
synced 2026-01-01 11:00:31 -06:00
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
"""
|
|
Command-line interface for the Computer API server.
|
|
"""
|
|
|
|
import argparse
|
|
import logging
|
|
import sys
|
|
from typing import List, Optional
|
|
|
|
from .server import Server
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def parse_args(args: Optional[List[str]] = None) -> argparse.Namespace:
|
|
"""Parse command-line arguments."""
|
|
parser = argparse.ArgumentParser(description="Start the Computer API server")
|
|
parser.add_argument(
|
|
"--host", default="0.0.0.0", help="Host to bind the server to (default: 0.0.0.0)"
|
|
)
|
|
parser.add_argument(
|
|
"--port", type=int, default=8000, help="Port to bind the server to (default: 8000)"
|
|
)
|
|
parser.add_argument(
|
|
"--log-level",
|
|
choices=["debug", "info", "warning", "error", "critical"],
|
|
default="info",
|
|
help="Logging level (default: info)",
|
|
)
|
|
|
|
return parser.parse_args(args)
|
|
|
|
|
|
def main() -> None:
|
|
"""Main entry point for the CLI."""
|
|
args = parse_args()
|
|
|
|
# Configure logging
|
|
logging.basicConfig(
|
|
level=getattr(logging, args.log_level.upper()),
|
|
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
|
|
)
|
|
|
|
# Create and start the server
|
|
logger.info(f"Starting CUA Computer API server on {args.host}:{args.port}...")
|
|
server = Server(host=args.host, port=args.port, log_level=args.log_level)
|
|
|
|
try:
|
|
server.start()
|
|
except KeyboardInterrupt:
|
|
logger.info("Server stopped by user")
|
|
sys.exit(0)
|
|
except Exception as e:
|
|
logger.error(f"Error starting server: {e}")
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|