5120861e17
- Add GPU fields (model, memory, count, CUDA version, price, region) to MarketplaceOffer model - Create new marketplace_gpu router for GPU-specific operations - Update offer sync to populate GPU fields from miner capabilities - Move GPU attributes from generic attributes dict to dedicated fields - Update MarketplaceOfferView schema with GPU fields - Expand CLI README with comprehensive documentation and
137 lines
3.2 KiB
Python
137 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
AITBC CLI - Main entry point for the AITBC Command Line Interface
|
|
"""
|
|
|
|
import click
|
|
import sys
|
|
from typing import Optional
|
|
|
|
from . import __version__
|
|
from .config import get_config
|
|
from .utils import output, setup_logging
|
|
from .commands.client import client
|
|
from .commands.miner import miner
|
|
from .commands.wallet import wallet
|
|
from .commands.auth import auth
|
|
from .commands.blockchain import blockchain
|
|
from .commands.marketplace import marketplace
|
|
from .commands.simulate import simulate
|
|
from .commands.admin import admin
|
|
from .commands.config import config
|
|
from .commands.monitor import monitor
|
|
from .plugins import plugin, load_plugins
|
|
|
|
|
|
@click.group()
|
|
@click.option(
|
|
"--url",
|
|
default=None,
|
|
help="Coordinator API URL (overrides config)"
|
|
)
|
|
@click.option(
|
|
"--api-key",
|
|
default=None,
|
|
help="API key (overrides config)"
|
|
)
|
|
@click.option(
|
|
"--output",
|
|
type=click.Choice(["table", "json", "yaml"]),
|
|
default="table",
|
|
help="Output format"
|
|
)
|
|
@click.option(
|
|
"--verbose", "-v",
|
|
count=True,
|
|
help="Increase verbosity (use -v, -vv, -vvv)"
|
|
)
|
|
@click.option(
|
|
"--debug",
|
|
is_flag=True,
|
|
help="Enable debug mode"
|
|
)
|
|
@click.option(
|
|
"--config-file",
|
|
default=None,
|
|
help="Path to config file"
|
|
)
|
|
@click.version_option(version=__version__, prog_name="aitbc")
|
|
@click.pass_context
|
|
def cli(ctx, url: Optional[str], api_key: Optional[str], output: str,
|
|
verbose: int, debug: bool, config_file: Optional[str]):
|
|
"""
|
|
AITBC CLI - Command Line Interface for AITBC Network
|
|
|
|
Manage jobs, mining, wallets, and blockchain operations from the command line.
|
|
"""
|
|
# Ensure context object exists
|
|
ctx.ensure_object(dict)
|
|
|
|
# Setup logging based on verbosity
|
|
log_level = setup_logging(verbose, debug)
|
|
|
|
# Load configuration
|
|
config = get_config(config_file)
|
|
|
|
# Override config with command line options
|
|
if url:
|
|
config.coordinator_url = url
|
|
if api_key:
|
|
config.api_key = api_key
|
|
|
|
# Store in context for subcommands
|
|
ctx.obj['config'] = config
|
|
ctx.obj['output_format'] = output
|
|
ctx.obj['log_level'] = log_level
|
|
|
|
|
|
# Add command groups
|
|
cli.add_command(client)
|
|
cli.add_command(miner)
|
|
cli.add_command(wallet)
|
|
cli.add_command(auth)
|
|
cli.add_command(blockchain)
|
|
cli.add_command(marketplace)
|
|
cli.add_command(simulate)
|
|
cli.add_command(admin)
|
|
cli.add_command(config)
|
|
cli.add_command(monitor)
|
|
cli.add_command(plugin)
|
|
load_plugins(cli)
|
|
|
|
|
|
@cli.command()
|
|
@click.pass_context
|
|
def version(ctx):
|
|
"""Show version information"""
|
|
output(f"AITBC CLI version {__version__}", ctx.obj['output_format'])
|
|
|
|
|
|
@cli.command()
|
|
@click.pass_context
|
|
def config_show(ctx):
|
|
"""Show current configuration"""
|
|
config = ctx.obj['config']
|
|
output({
|
|
"coordinator_url": config.coordinator_url,
|
|
"api_key": "***REDACTED***" if config.api_key else None,
|
|
"output_format": ctx.obj['output_format'],
|
|
"config_file": config.config_file
|
|
}, ctx.obj['output_format'])
|
|
|
|
|
|
def main():
|
|
"""Main entry point"""
|
|
try:
|
|
cli()
|
|
except KeyboardInterrupt:
|
|
click.echo("\nAborted by user", err=True)
|
|
sys.exit(1)
|
|
except Exception as e:
|
|
click.echo(f"Error: {e}", err=True)
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|