refactor: flatten CLI directory structure - remove 'box in a box'
BEFORE: /opt/aitbc/cli/ ├── aitbc_cli/ # Python package (box in a box) │ ├── commands/ │ ├── main.py │ └── ... ├── setup.py AFTER: /opt/aitbc/cli/ # Flat structure ├── commands/ # Direct access ├── main.py # Direct access ├── auth/ ├── config/ ├── core/ ├── models/ ├── utils/ ├── plugins.py └── setup.py CHANGES MADE: - Moved all files from aitbc_cli/ to cli/ root - Fixed all relative imports (from . to absolute imports) - Updated setup.py entry point: aitbc_cli.main → main - Added CLI directory to Python path in entry script - Simplified deployment.py to remove dependency on deleted core.deployment - Fixed import paths in all command files - Recreated virtual environment with new structure BENEFITS: - Eliminated 'box in a box' nesting - Simpler directory structure - Direct access to all modules - Cleaner imports - Easier maintenance and development - CLI works with both 'python main.py' and 'aitbc' commands
This commit is contained in:
315
cli/main.py
Executable file
315
cli/main.py
Executable file
@@ -0,0 +1,315 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
AITBC CLI - Main entry point for the AITBC Command Line Interface
|
||||
"""
|
||||
|
||||
import click
|
||||
import sys
|
||||
from typing import Optional
|
||||
|
||||
from __init__ import __version__
|
||||
from config import get_config
|
||||
|
||||
|
||||
def with_role(role: str):
|
||||
"""Decorator to set role for command groups"""
|
||||
def decorator(func):
|
||||
@click.pass_context
|
||||
def wrapper(ctx, *args, **kwargs):
|
||||
ctx.parent.detected_role = role
|
||||
return func(ctx, *args, **kwargs)
|
||||
return wrapper
|
||||
return decorator
|
||||
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 commands.governance import governance
|
||||
from commands.exchange import exchange
|
||||
from commands.oracle import oracle
|
||||
from commands.market_maker import market_maker
|
||||
from commands.multisig import multisig
|
||||
from commands.genesis_protection import genesis_protection
|
||||
from commands.transfer_control import transfer_control
|
||||
from commands.agent import agent
|
||||
from commands.multimodal import multimodal
|
||||
from commands.optimize import optimize
|
||||
# from commands.openclaw import openclaw # Temporarily disabled due to naming conflict
|
||||
from commands.marketplace_advanced import advanced # Re-enabled after fixing registration issues
|
||||
from commands.swarm import swarm
|
||||
from commands.chain import chain
|
||||
from commands.genesis import genesis
|
||||
from commands.keystore import keystore
|
||||
from commands.test_cli import test
|
||||
from commands.node import node
|
||||
from commands.analytics import analytics
|
||||
from commands.agent_comm import agent_comm
|
||||
from commands.deployment import deploy
|
||||
from commands.cross_chain import cross_chain
|
||||
from commands.compliance import compliance
|
||||
from commands.surveillance import surveillance
|
||||
from commands.regulatory import regulatory
|
||||
from commands.ai_trading import ai_trading
|
||||
from commands.advanced_analytics import advanced_analytics_group
|
||||
from commands.ai_surveillance import ai_surveillance_group
|
||||
|
||||
# AI provider commands
|
||||
from commands.ai import ai_group
|
||||
|
||||
# Enterprise integration (optional)
|
||||
try:
|
||||
from commands.enterprise_integration import enterprise_integration_group
|
||||
except ImportError:
|
||||
enterprise_integration_group = None
|
||||
|
||||
from commands.sync import sync
|
||||
from commands.explorer import explorer
|
||||
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.option(
|
||||
"--test-mode",
|
||||
is_flag=True,
|
||||
help="Enable test mode (uses mock data and test endpoints)"
|
||||
)
|
||||
@click.option(
|
||||
"--dry-run",
|
||||
is_flag=True,
|
||||
help="Dry run mode (show what would be done without executing)"
|
||||
)
|
||||
@click.option(
|
||||
"--timeout",
|
||||
type=int,
|
||||
default=30,
|
||||
help="Request timeout in seconds (useful for testing)"
|
||||
)
|
||||
@click.option(
|
||||
"--no-verify",
|
||||
is_flag=True,
|
||||
help="Skip SSL certificate verification (testing only)"
|
||||
)
|
||||
@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], test_mode: bool,
|
||||
dry_run: bool, timeout: int, no_verify: bool):
|
||||
"""
|
||||
AITBC CLI - Command Line Interface for AITBC Network
|
||||
|
||||
Manage jobs, mining, wallets, blockchain operations, marketplaces, and AI services.
|
||||
|
||||
CORE COMMANDS:
|
||||
client Submit and manage AI compute jobs
|
||||
miner GPU mining operations and status
|
||||
wallet Wallet management and transactions
|
||||
marketplace GPU marketplace and trading
|
||||
blockchain Blockchain operations and queries
|
||||
exchange Real exchange integration (Binance, Coinbase, etc.)
|
||||
explorer Blockchain explorer and analytics
|
||||
|
||||
ADVANCED FEATURES:
|
||||
analytics Chain performance monitoring and predictions
|
||||
ai-trading AI-powered trading strategies
|
||||
surveillance Market surveillance and compliance
|
||||
compliance Regulatory compliance and reporting
|
||||
governance Network governance and proposals
|
||||
|
||||
DEVELOPMENT TOOLS:
|
||||
admin Administrative operations
|
||||
config Configuration management
|
||||
monitor System monitoring and health
|
||||
test CLI testing and validation
|
||||
deploy Deployment and infrastructure management
|
||||
|
||||
SPECIALIZED SERVICES:
|
||||
agent AI agent operations
|
||||
multimodal Multi-modal AI processing
|
||||
oracle Price discovery and data feeds
|
||||
market-maker Automated market making
|
||||
genesis-protection Advanced security features
|
||||
|
||||
Use 'aitbc <command> --help' for detailed help on any command.
|
||||
|
||||
Examples:
|
||||
aitbc client submit --prompt "Generate an image" --model llama2
|
||||
aitbc miner status
|
||||
aitbc wallet create --type hd
|
||||
aitbc marketplace list
|
||||
aitbc exchange create-pair --pair AITBC/BTC --base-asset AITBC --quote-asset BTC
|
||||
aitbc analytics summary
|
||||
aitbc explorer status
|
||||
aitbc explorer block 12345
|
||||
aitbc explorer transaction 0x123...
|
||||
aitbc explorer search --address 0xabc...
|
||||
"""
|
||||
# Ensure context object exists
|
||||
ctx.ensure_object(dict)
|
||||
|
||||
# Setup logging based on verbosity
|
||||
log_level = setup_logging(verbose, debug)
|
||||
|
||||
# Detect role from command name (before config is loaded)
|
||||
role = None
|
||||
|
||||
# Check invoked_subcommand first
|
||||
if ctx.invoked_subcommand:
|
||||
if ctx.invoked_subcommand == 'client':
|
||||
role = 'client'
|
||||
elif ctx.invoked_subcommand == 'miner':
|
||||
role = 'miner'
|
||||
elif ctx.invoked_subcommand == 'blockchain':
|
||||
role = 'blockchain'
|
||||
elif ctx.invoked_subcommand == 'admin':
|
||||
role = 'admin'
|
||||
|
||||
# Also check if role was already set by command group
|
||||
if not role:
|
||||
role = getattr(ctx, 'detected_role', None)
|
||||
|
||||
# Load configuration with role
|
||||
config = get_config(config_file, role=role)
|
||||
|
||||
# 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
|
||||
ctx.obj['test_mode'] = test_mode
|
||||
ctx.obj['dry_run'] = dry_run
|
||||
ctx.obj['timeout'] = timeout
|
||||
ctx.obj['no_verify'] = no_verify
|
||||
|
||||
# Apply test mode settings
|
||||
if test_mode:
|
||||
config.coordinator_url = config.coordinator_url or "http://localhost:8000"
|
||||
config.api_key = config.api_key or "test-api-key"
|
||||
|
||||
|
||||
# Add command groups
|
||||
cli.add_command(client)
|
||||
cli.add_command(miner)
|
||||
cli.add_command(wallet)
|
||||
cli.add_command(plugin)
|
||||
# cli.add_command(openclaw) # Temporarily disabled due to naming conflict
|
||||
cli.add_command(advanced) # Re-enabled after fixing registration issues
|
||||
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(governance)
|
||||
cli.add_command(exchange)
|
||||
cli.add_command(oracle)
|
||||
cli.add_command(market_maker)
|
||||
cli.add_command(multisig)
|
||||
cli.add_command(genesis_protection)
|
||||
cli.add_command(transfer_control)
|
||||
cli.add_command(agent)
|
||||
cli.add_command(multimodal)
|
||||
cli.add_command(optimize)
|
||||
cli.add_command(ai_group)
|
||||
# cli.add_command(openclaw) # Temporarily disabled
|
||||
cli.add_command(swarm)
|
||||
cli.add_command(chain)
|
||||
cli.add_command(genesis)
|
||||
cli.add_command(keystore)
|
||||
cli.add_command(test)
|
||||
cli.add_command(node)
|
||||
cli.add_command(analytics)
|
||||
cli.add_command(agent_comm)
|
||||
cli.add_command(deploy)
|
||||
cli.add_command(cross_chain)
|
||||
cli.add_command(compliance)
|
||||
cli.add_command(surveillance)
|
||||
cli.add_command(regulatory)
|
||||
cli.add_command(ai_trading)
|
||||
cli.add_command(advanced_analytics_group)
|
||||
cli.add_command(ai_surveillance_group)
|
||||
if enterprise_integration_group is not None:
|
||||
cli.add_command(enterprise_integration_group)
|
||||
cli.add_command(sync)
|
||||
cli.add_command(explorer)
|
||||
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()
|
||||
Reference in New Issue
Block a user