trading-cli 0.3.0__py3-none-any.whl

fastbooks_cli/commands/bots.py

@@ -0,0 +1,479 @@
+"""Bots command group — create, manage, monitor trading bots."""
+
+from __future__ import annotations
+
+import json
+
+import click
+
+from ..client import FastBooksClient
+from ..output import emit, emit_error, format_table
+
+VENUES = [
+    "binance", "bybit", "okx", "kraken", "coinbase", "kucoin",
+    "hyperliquid", "ibkr", "alpaca",
+]
+
+STRATEGIES = [
+    "conviction_swing", "dca", "momentum", "mean_reversion",
+    "grid", "tech_momentum", "morning_scalper",
+]
+
+ASSET_TYPES = ["stock", "crypto", "forex"]
+
+
+@click.group()
+def bots():
+    """Bot management: create, start, stop, monitor trading bots.
+
+    Bots execute automated strategies on supported venues. They can be
+    configured for paper or live trading with conviction-based entries.
+    """
+    pass
+
+
+# ── create ───────────────────────────────────────────────────────────────
+
+@bots.command()
+@click.option("--name", "-n", required=True, help="Bot display name.")
+@click.option("--venue", "-v", type=click.Choice(VENUES, case_sensitive=False), required=True)
+@click.option("--strategy", "-s", type=click.Choice(STRATEGIES, case_sensitive=False), default="conviction_swing")
+@click.option("--asset-type", type=click.Choice(ASSET_TYPES, case_sensitive=False), default="stock")
+@click.option("--capital", "-c", type=float, required=True, help="Starting capital (in quote currency).")
+@click.option("--quote-currency", "-q", type=str, default="USD", help="Quote currency (USD, USDT, USDC).")
+@click.option("--paper/--live", default=True, help="Paper or live trading (default: paper).")
+@click.option("--conviction-threshold", type=float, default=65.0, help="Min conviction score to enter (0-100).")
+@click.option("--market-hours-only", is_flag=True, help="Only trade during market hours.")
+@click.option("--auto-rebalance", is_flag=True, help="Enable automatic portfolio rebalancing.")
+@click.option("--config-json", type=str, default=None, help="Full bot config as JSON string (overrides other options).")
+@click.pass_context
+def create(ctx, name, venue, strategy, asset_type, capital, quote_currency, paper, conviction_threshold, market_hours_only, auto_rebalance, config_json):
+    """Create a new trading bot.
+
+    Examples:
+        fastbooks bots create --name "BTC DCA" --venue binance --strategy dca --capital 1000 --paper
+        fastbooks bots create --name "Perp Grid" --venue hyperliquid --strategy grid --capital 5000 -q USDC
+        fastbooks bots create --name "IBKR Swing" --venue ibkr --strategy conviction_swing --capital 50000 --live
+        fastbooks bots create --config-json '{"name":"Custom","venue":"alpaca","capital":10000}'
+    """
+    client: FastBooksClient = ctx.obj
+
+    if config_json:
+        try:
+            bot_config = json.loads(config_json)
+        except json.JSONDecodeError as e:
+            emit_error(ctx, f"Invalid JSON: {e}")
+            ctx.exit(1)
+            return
+    else:
+        bot_config = {
+            "name": name,
+            "venue": venue,
+            "strategy": strategy,
+            "asset_type": asset_type,
+            "capital": capital,
+            "quote_currency": quote_currency,
+            "is_paper_trading": paper,
+            "conviction_threshold": conviction_threshold,
+            "market_hours_only": market_hours_only,
+            "auto_rebalance": auto_rebalance,
+        }
+
+    try:
+        data = client.create_bot(bot_config)
+        emit(ctx, data, lambda d: f"  Bot created: {d.get('id', d.get('bot_id', '?'))} ({name})")
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+# ── list ─────────────────────────────────────────────────────────────────
+
+@bots.command("list")
+@click.option("--investor-id", type=str, default=None, help="Filter by investor ID.")
+@click.pass_context
+def list_bots(ctx, investor_id):
+    """List all bots.
+
+    Examples:
+        fastbooks bots list
+        fastbooks --json bots list
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.list_bots(investor_id)
+        emit(ctx, data, lambda d: _format_bots(d))
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+def _format_bots(data) -> str:
+    items = data if isinstance(data, list) else data.get("bots", data.get("data", []))
+    if not isinstance(items, list):
+        return str(data)
+    if not items:
+        return "  No bots found."
+    return format_table(
+        items,
+        ["id", "name", "venue", "status", "capital", "is_paper_trading"],
+        ["ID", "Name", "Venue", "Status", "Capital", "Paper?"],
+    )
+
+
+# ── info ─────────────────────────────────────────────────────────────────
+
+@bots.command()
+@click.argument("bot_id")
+@click.pass_context
+def info(ctx, bot_id):
+    """Get detailed bot information.
+
+    Example:
+        fastbooks bots info abc-123
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.get_bot(bot_id)
+        emit(ctx, data)
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+# ── lifecycle commands ───────────────────────────────────────────────────
+
+@bots.command()
+@click.argument("bot_id")
+@click.pass_context
+def start(ctx, bot_id):
+    """Start a bot.
+
+    Example: fastbooks bots start abc-123
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.start_bot(bot_id)
+        emit(ctx, data, lambda d: f"  Bot {bot_id} started.")
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+@bots.command()
+@click.argument("bot_id")
+@click.pass_context
+def stop(ctx, bot_id):
+    """Stop a running bot.
+
+    Example: fastbooks bots stop abc-123
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.stop_bot(bot_id)
+        emit(ctx, data, lambda d: f"  Bot {bot_id} stopped.")
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+@bots.command()
+@click.argument("bot_id")
+@click.pass_context
+def pause(ctx, bot_id):
+    """Pause a running bot (keeps state).
+
+    Example: fastbooks bots pause abc-123
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.pause_bot(bot_id)
+        emit(ctx, data, lambda d: f"  Bot {bot_id} paused.")
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+@bots.command()
+@click.argument("bot_id")
+@click.pass_context
+def resume(ctx, bot_id):
+    """Resume a paused bot.
+
+    Example: fastbooks bots resume abc-123
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.resume_bot(bot_id)
+        emit(ctx, data, lambda d: f"  Bot {bot_id} resumed.")
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+# ── update ───────────────────────────────────────────────────────────────
+
+@bots.command()
+@click.argument("bot_id")
+@click.option("--capital", type=float, default=None, help="Update capital.")
+@click.option("--conviction-threshold", type=float, default=None, help="Update conviction threshold.")
+@click.option("--paper/--live", default=None, help="Switch paper/live mode.")
+@click.option("--config-json", type=str, default=None, help="Partial config as JSON.")
+@click.pass_context
+def update(ctx, bot_id, capital, conviction_threshold, paper, config_json):
+    """Update bot configuration.
+
+    Examples:
+        fastbooks bots update abc-123 --capital 20000
+        fastbooks bots update abc-123 --conviction-threshold 70
+        fastbooks bots update abc-123 --config-json '{"auto_rebalance": true}'
+    """
+    client: FastBooksClient = ctx.obj
+
+    if config_json:
+        try:
+            updates = json.loads(config_json)
+        except json.JSONDecodeError as e:
+            emit_error(ctx, f"Invalid JSON: {e}")
+            ctx.exit(1)
+            return
+    else:
+        updates = {}
+        if capital is not None:
+            updates["capital"] = capital
+        if conviction_threshold is not None:
+            updates["conviction_threshold"] = conviction_threshold
+        if paper is not None:
+            updates["is_paper_trading"] = paper
+
+    if not updates:
+        emit_error(ctx, "No updates provided.")
+        ctx.exit(1)
+        return
+
+    try:
+        data = client.update_bot(bot_id, updates)
+        emit(ctx, data, lambda d: f"  Bot {bot_id} updated.")
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+# ── delete ───────────────────────────────────────────────────────────────
+
+@bots.command()
+@click.argument("bot_id")
+@click.option("--confirm/--no-confirm", default=True, help="Require confirmation.")
+@click.pass_context
+def delete(ctx, bot_id, confirm):
+    """Delete a bot permanently.
+
+    Example: fastbooks bots delete abc-123
+    """
+    client: FastBooksClient = ctx.obj
+
+    if confirm and not ctx.find_root().params.get("json_output", False):
+        if not click.confirm(f"  Delete bot {bot_id}? This cannot be undone."):
+            click.echo("  Cancelled.")
+            return
+
+    try:
+        data = client.delete_bot(bot_id)
+        emit(ctx, data, lambda d: f"  Bot {bot_id} deleted.")
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+# ── monitoring ───────────────────────────────────────────────────────────
+
+@bots.command()
+@click.argument("bot_id")
+@click.option("--limit", "-n", type=int, default=50, help="Max events to return.")
+@click.pass_context
+def events(ctx, bot_id, limit):
+    """Get bot events / decision log.
+
+    Example: fastbooks bots events abc-123 --limit 20
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.bot_events(bot_id, limit)
+        emit(ctx, data)
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+@bots.command()
+@click.argument("bot_id")
+@click.option("--since", type=int, default=0, help="Show logs since timestamp.")
+@click.pass_context
+def logs(ctx, bot_id, since):
+    """Get bot subprocess logs (stdout/stderr).
+
+    Example: fastbooks bots logs abc-123
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.bot_logs(bot_id, since)
+        emit(ctx, data, lambda d: d.get("logs", str(d)) if isinstance(d, dict) else str(d))
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+@bots.command()
+@click.argument("bot_id")
+@click.pass_context
+def report(ctx, bot_id):
+    """Generate a performance report for a bot.
+
+    Example: fastbooks bots report abc-123
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.bot_report(bot_id)
+        emit(ctx, data)
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+# ── rebalance / trade ────────────────────────────────────────────────────
+
+@bots.command()
+@click.argument("bot_id")
+@click.pass_context
+def rebalance(ctx, bot_id):
+    """Trigger a portfolio rebalance for a bot.
+
+    Example: fastbooks bots rebalance abc-123
+    """
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.rebalance_bot(bot_id)
+        emit(ctx, data, lambda d: f"  Rebalance triggered for bot {bot_id}.")
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+@bots.command()
+@click.argument("bot_id")
+@click.argument("symbol")
+@click.argument("side", type=click.Choice(["buy", "sell"], case_sensitive=False))
+@click.argument("amount", type=float)
+@click.option("--reason", type=str, default=None, help="Trade reason.")
+@click.pass_context
+def trade(ctx, bot_id, symbol, side, amount, reason):
+    """Submit a manual trade intent to a bot.
+
+    Example: fastbooks bots trade abc-123 AAPL buy 10 --reason "earnings play"
+    """
+    client: FastBooksClient = ctx.obj
+    intent = {"symbol": symbol, "side": side, "amount": amount}
+    if reason:
+        intent["reason"] = reason
+    try:
+        data = client.bot_trade(bot_id, intent)
+        emit(ctx, data, lambda d: f"  Trade intent submitted to bot {bot_id}.")
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+# ── deploy (template → running bot) ───────────────────────────────────────
+
+@bots.command()
+@click.option("--name", "-n", required=True, help="Bot display name.")
+@click.option("--venue", "-v", type=click.Choice(VENUES, case_sensitive=False), required=True)
+@click.option("--strategy", "-s", type=click.Choice(STRATEGIES, case_sensitive=False), default="conviction_swing")
+@click.option("--asset-type", type=click.Choice(ASSET_TYPES, case_sensitive=False), default="stock")
+@click.option("--capital", "-c", type=float, required=True, help="Starting capital (in quote currency).")
+@click.option("--quote-currency", "-q", type=str, default="USD", help="Quote currency (USD, USDT, USDC).")
+@click.option("--paper/--live", default=True, help="Paper or live trading (default: paper).")
+@click.option("--start/--no-start", default=True, help="Start the bot immediately after deploy.")
+@click.option("--config-json", type=str, default=None, help="Full deploy config as JSON (overrides options).")
+@click.pass_context
+def deploy(ctx, name, venue, strategy, asset_type, capital, quote_currency, paper, start, config_json):
+    """Deploy a bot from a template and (by default) start it on paper.
+
+    Examples:
+        fastbooks bots deploy -n "Paper Momentum" -v alpaca -s momentum -c 50000 --paper
+        fastbooks bots deploy -n "HL Grid" -v hyperliquid -s grid -c 5000 -q USDC --live
+    """
+    client: FastBooksClient = ctx.obj
+    if config_json:
+        try:
+            cfg = json.loads(config_json)
+        except json.JSONDecodeError as e:
+            emit_error(ctx, f"Invalid JSON: {e}")
+            ctx.exit(1)
+            return
+    else:
+        cfg = {
+            "name": name,
+            "venue": venue,
+            "strategy": strategy,
+            "asset_type": asset_type,
+            "capital": capital,
+            "quote_currency": quote_currency,
+            "is_paper_trading": paper,
+            "auto_start": start,
+        }
+    try:
+        data = client.deploy_bot(cfg)
+        bot_id = data.get("id") or data.get("bot_id") or "?"
+        mode = "paper" if paper else "live"
+        emit(ctx, data, lambda d: f"  Bot deployed ({mode}): {bot_id} ({name})")
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+# ── command (control channel) ─────────────────────────────────────────────
+
+@bots.command("command")
+@click.argument("bot_id")
+@click.argument("command")
+@click.option("--args-json", type=str, default=None, help="Command args as JSON object.")
+@click.pass_context
+def send_command(ctx, bot_id, command, args_json):
+    """Send a control command to a running bot.
+
+    Example: fastbooks bots command abc-123 flatten
+    """
+    client: FastBooksClient = ctx.obj
+    args = None
+    if args_json:
+        try:
+            args = json.loads(args_json)
+        except json.JSONDecodeError as e:
+            emit_error(ctx, f"Invalid JSON: {e}")
+            ctx.exit(1)
+            return
+    try:
+        data = client.bot_command(bot_id, command, args)
+        emit(ctx, data, lambda d: f"  Command '{command}' sent to bot {bot_id}.")
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)
+
+
+# ── emergency stop-all ─────────────────────────────────────────────────────
+
+@bots.command("emergency-stop")
+@click.option("--confirm", is_flag=True, help="Skip the confirmation prompt.")
+@click.pass_context
+def emergency_stop(ctx, confirm):
+    """Emergency-stop ALL running bots."""
+    if not confirm and not click.confirm("Stop ALL running bots?"):
+        ctx.exit(0)
+    client: FastBooksClient = ctx.obj
+    try:
+        data = client.emergency_stop_all()
+        emit(ctx, data, lambda d: f"  Emergency stop issued. {d.get('stopped', '?')} bots stopped.")
+    except Exception as e:
+        emit_error(ctx, str(e))
+        ctx.exit(1)