"""Portfolio CLI commands.""" from __future__ import annotations import json from typing import Any import typer from app.cli.core.utils import execute_command, render_output from app.schemas.portfolio import ( BuyRequest, CommentCreateRequest, CommentUpdateRequest, CreatePortfolioRequest, InvestRequest, PortfolioTagMutationRequest, ResetPortfolioRequest, SellRequest, TransferRequest, UpdatePortfolioRequest, ) from app.services.portfolio_service import PortfolioService portfolio_app = typer.Typer(help="Portfolio lifecycle and trade commands.") comment_app = typer.Typer(help="Portfolio comment history commands.") tag_app = typer.Typer(help="Portfolio tag commands.") portfolio_app.add_typer(comment_app, name="comment") portfolio_app.add_typer(tag_app, name="tag") def _parse_strategy_params(value: str) -> dict[str, Any]: if not value.strip(): return {} parsed = json.loads(value) if not isinstance(parsed, dict): raise ValueError("strategy params must be a JSON object") return parsed def _with_service(ctx: typer.Context, callback): app_context = ctx.obj repository = app_context.repository_factory(app_context.settings) service = PortfolioService(repository) try: result = execute_command(app_context.console, callback=lambda: callback(service, app_context)) return result finally: if callable(getattr(type(repository), "close", None)): repository.close() def _with_service_and_repo(ctx: typer.Context, callback): app_context = ctx.obj repository = app_context.repository_factory(app_context.settings) service = PortfolioService(repository) try: result = execute_command(app_context.console, callback=lambda: callback(service, repository, app_context)) return result finally: if callable(getattr(type(repository), "close", None)): repository.close() def _confirm(force: bool, message: str) -> None: if force: return confirmed = typer.confirm(message) if not confirmed: raise typer.Exit(code=0) @portfolio_app.command("list") def list_portfolios( ctx: typer.Context, user: str | None = typer.Option(None, "--user", help="Filter by username (required)."), broker: str | None = typer.Option(None, help="Filter by broker."), status: str | None = typer.Option(None, help="Filter by status."), include_tags: list[str] | None = typer.Option(None, "--include-tag", help="Include tag filter. Repeatable."), exclude_tags: list[str] | None = typer.Option(None, "--exclude-tag", help="Exclude tag filter. Repeatable."), match_mode: str | None = typer.Option(None, "--match-mode", help="AND or OR for include tags."), ) -> None: if user is None: ctx.obj.console.print("Error: --user 옵션은 필수입니다.") raise typer.Exit(code=1) def _callback(service, repository, app_context): owner = repository.get_user_by_username(user) if owner is None: raise ValueError(f"사용자를 찾을 수 없습니다: {user}") return service.list_portfolios( owner_user_id=owner.id, broker=broker, status=status, include_tags=include_tags, exclude_tags=exclude_tags, match_mode=match_mode, ) detail = _with_service_and_repo(ctx, _callback) render_output(ctx.obj.console, ctx.obj.output_format, detail, empty_message="No portfolios found.") @portfolio_app.command("status") def portfolio_status( ctx: typer.Context, portfolio_id: str = typer.Argument(..., help="Portfolio identifier."), recent: int = typer.Option(5, "--recent", min=1, help="Number of recent ledger entries."), ) -> None: detail = _with_service( ctx, lambda service, app_context: service.get_portfolio_status(portfolio_id=portfolio_id, recent_trades=recent), ) render_output(ctx.obj.console, ctx.obj.output_format, detail) @portfolio_app.command("create") def create_portfolio( ctx: typer.Context, portfolio_id: str = typer.Argument(..., help="Portfolio identifier."), user: str | None = typer.Option(None, "--user", help="Owner username (required)."), name: str | None = typer.Option(None, help="Asset display name."), symbol: str = typer.Option(..., help="Trading symbol."), broker: str = typer.Option("kis", help="Broker code."), status: str = typer.Option("enabled", help="Portfolio status."), cash_balance: float = typer.Option(0.0, min=0.0, help="Starting cash balance."), asset_quantity: float = typer.Option(0.0, min=0.0, help="Starting asset quantity."), asset_price: float = typer.Option(0.0, min=0.0, help="Starting asset price."), profit_buffer: float = typer.Option(0.0, min=0.0, help="Initial profit buffer."), strategy_name: str = typer.Option("target_value", help="Strategy name."), strategy_params: str = typer.Option("{}", help="Strategy params as JSON object."), force: bool = typer.Option(False, "--force", help="Skip confirmation prompt."), ) -> None: if user is None: ctx.obj.console.print("Error: --user 옵션은 필수입니다.") raise typer.Exit(code=1) _confirm(force, f"Create portfolio {portfolio_id} for symbol {symbol}?") def _callback(service, repository, app_context): owner = repository.get_user_by_username(user) if owner is None: raise ValueError(f"사용자를 찾을 수 없습니다: {user}") return service.create_portfolio( CreatePortfolioRequest( portfolio_id=portfolio_id, name=name, symbol=symbol, broker=broker, status=status, cash_balance=cash_balance, asset_quantity=asset_quantity, asset_price=asset_price, profit_buffer=profit_buffer, strategy_name=strategy_name, strategy_params=_parse_strategy_params(strategy_params), ), owner_user_id=owner.id, ) detail = _with_service_and_repo(ctx, _callback) render_output(ctx.obj.console, ctx.obj.output_format, detail) @portfolio_app.command("update") def update_portfolio( ctx: typer.Context, portfolio_id: str = typer.Argument(..., help="Portfolio identifier."), broker: str | None = typer.Option(None, help="Broker code."), status: str | None = typer.Option(None, help="Portfolio status."), name: str | None = typer.Option(None, help="Asset display name."), symbol: str | None = typer.Option(None, help="Trading symbol."), strategy_name: str | None = typer.Option(None, help="Strategy name."), strategy_params: str | None = typer.Option(None, help="Strategy params as JSON object."), asset_price: float | None = typer.Option(None, min=0.0, help="Current asset price."), force: bool = typer.Option(False, "--force", help="Skip confirmation prompt."), ) -> None: _confirm(force, f"Update portfolio {portfolio_id}?") parsed_strategy_params = None if strategy_params is None else _parse_strategy_params(strategy_params) detail = _with_service( ctx, lambda service, app_context: service.update_portfolio( portfolio_id, UpdatePortfolioRequest( broker=broker, status=status, name=name, symbol=symbol, strategy_name=strategy_name, strategy_params=parsed_strategy_params, asset_price=asset_price, ), ), ) render_output(ctx.obj.console, ctx.obj.output_format, detail) @portfolio_app.command("buy") def buy_portfolio( ctx: typer.Context, portfolio_id: str = typer.Argument(..., help="Portfolio identifier."), quantity: float = typer.Option(..., min=0.0000001, help="Quantity to buy."), price: float = typer.Option(..., min=0.0000001, help="Execution price."), fee: float = typer.Option(0.0, min=0.0, help="Fee amount."), tax: float = typer.Option(0.0, min=0.0, help="Tax amount."), note: str = typer.Option("", help="Ledger note."), force: bool = typer.Option(False, "--force", help="Skip confirmation prompt."), ) -> None: total_cost = quantity * price + fee + tax _confirm(force, f"Buy {quantity:g} units of {portfolio_id} at {price:,.0f} (total {total_cost:,.0f})?") detail = _with_service( ctx, lambda service, app_context: service.buy( portfolio_id, BuyRequest(quantity=quantity, price=price, fee=fee, tax=tax, note=note), ), ) render_output(ctx.obj.console, ctx.obj.output_format, detail) @portfolio_app.command("sell") def sell_portfolio( ctx: typer.Context, portfolio_id: str = typer.Argument(..., help="Portfolio identifier."), quantity: float = typer.Option(..., min=0.0000001, help="Quantity to sell."), price: float = typer.Option(..., min=0.0000001, help="Execution price."), fee: float = typer.Option(0.0, min=0.0, help="Fee amount."), tax: float = typer.Option(0.0, min=0.0, help="Tax amount."), note: str = typer.Option("", help="Ledger note."), force: bool = typer.Option(False, "--force", help="Skip confirmation prompt."), ) -> None: net_amount = quantity * price - fee - tax _confirm(force, f"Sell {quantity:g} units of {portfolio_id} at {price:,.0f} (net {net_amount:,.0f})?") detail = _with_service( ctx, lambda service, app_context: service.sell( portfolio_id, SellRequest(quantity=quantity, price=price, fee=fee, tax=tax, note=note), ), ) render_output(ctx.obj.console, ctx.obj.output_format, detail) @portfolio_app.command("invest") def invest_portfolio( ctx: typer.Context, portfolio_id: str = typer.Argument(..., help="Portfolio identifier."), movement_type: str = typer.Option(..., "--type", help="deposit, profit, or withdraw."), amount: float = typer.Option(..., min=0.0000001, help="Buffer adjustment amount."), note: str = typer.Option("", help="Ledger note."), force: bool = typer.Option(False, "--force", help="Skip confirmation prompt."), ) -> None: _confirm(force, f"Apply {movement_type} of {amount:,.0f} to portfolio {portfolio_id}?") detail = _with_service( ctx, lambda service, app_context: service.invest( portfolio_id, InvestRequest(movement_type=movement_type, amount=amount, note=note), ), ) render_output(ctx.obj.console, ctx.obj.output_format, detail) @portfolio_app.command("transfer") def transfer_portfolio( ctx: typer.Context, portfolio_id: str = typer.Argument(..., help="Portfolio identifier."), direction: str = typer.Option(..., help="to-buffer or from-buffer."), amount: float = typer.Option(..., min=0.0000001, help="Transfer amount."), note: str = typer.Option("", help="Ledger note."), force: bool = typer.Option(False, "--force", help="Skip confirmation prompt."), ) -> None: _confirm(force, f"Transfer {amount:,.0f} {direction} for portfolio {portfolio_id}?") detail = _with_service( ctx, lambda service, app_context: service.transfer( portfolio_id, TransferRequest(direction=direction, amount=amount, note=note), ), ) render_output(ctx.obj.console, ctx.obj.output_format, detail) @portfolio_app.command("reset") def reset_portfolio( ctx: typer.Context, portfolio_id: str = typer.Argument(..., help="Portfolio identifier."), infer_symbol: str | None = typer.Option(None, help="Fallback symbol when portfolio does not exist."), force: bool = typer.Option(False, "--force", help="Skip confirmation prompt."), ) -> None: _confirm(force, f"Reset portfolio {portfolio_id}?") detail = _with_service( ctx, lambda service, app_context: service.reset(portfolio_id, ResetPortfolioRequest(infer_symbol=infer_symbol)), ) render_output(ctx.obj.console, ctx.obj.output_format, detail) @portfolio_app.command("delete") def delete_portfolio( ctx: typer.Context, portfolio_id: str = typer.Argument(..., help="Portfolio identifier."), keep_ledger: bool = typer.Option(False, help="Preserve ledger history."), force: bool = typer.Option(False, "--force", help="Skip confirmation prompt."), ) -> None: suffix = " and keep ledger" if keep_ledger else "" _confirm(force, f"Delete portfolio {portfolio_id}{suffix}?") _with_service( ctx, lambda service, app_context: service.delete(portfolio_id, keep_ledger=keep_ledger), ) if ctx.obj.output_format == "json": render_output(ctx.obj.console, ctx.obj.output_format, {"deleted": portfolio_id, "keep_ledger": keep_ledger}) else: ctx.obj.console.print(f"Deleted portfolio: {portfolio_id}") @comment_app.command("add") def comment_add( ctx: typer.Context, portfolio_id: str = typer.Argument(..., help="Portfolio identifier."), content: str = typer.Option(..., "--content", help="Comment text (supports multiline)."), ) -> None: detail = _with_service( ctx, lambda service, app_context: service.create_comment( portfolio_id, CommentCreateRequest(content=content), ), ) render_output(ctx.obj.console, ctx.obj.output_format, detail) @comment_app.command("list") def comment_list( ctx: typer.Context, portfolio_id: str = typer.Argument(..., help="Portfolio identifier."), ) -> None: detail = _with_service( ctx, lambda service, app_context: service.list_comments(portfolio_id), ) render_output(ctx.obj.console, ctx.obj.output_format, detail, empty_message="No comments found.") @comment_app.command("update") def comment_update( ctx: typer.Context, portfolio_id: str = typer.Argument(..., help="Portfolio identifier."), comment_id: str = typer.Argument(..., help="Comment identifier."), content: str = typer.Option(..., "--content", help="Updated comment text."), force: bool = typer.Option(False, "--force", help="Skip confirmation prompt."), ) -> None: _confirm(force, f"Update comment {comment_id} for portfolio {portfolio_id}?") detail = _with_service( ctx, lambda service, app_context: service.update_comment( portfolio_id, comment_id, CommentUpdateRequest(content=content), ), ) render_output(ctx.obj.console, ctx.obj.output_format, detail) @comment_app.command("delete") def comment_delete( ctx: typer.Context, portfolio_id: str = typer.Argument(..., help="Portfolio identifier."), comment_id: str = typer.Argument(..., help="Comment identifier."), force: bool = typer.Option(False, "--force", help="Skip confirmation prompt."), ) -> None: _confirm(force, f"Delete comment {comment_id} for portfolio {portfolio_id}?") _with_service( ctx, lambda service, app_context: service.delete_comment(portfolio_id, comment_id), ) if ctx.obj.output_format == "json": render_output(ctx.obj.console, ctx.obj.output_format, {"deleted": comment_id, "portfolio_id": portfolio_id}) else: ctx.obj.console.print(f"Deleted comment: {comment_id}") @tag_app.command("add") def tag_add( ctx: typer.Context, portfolio_id: str = typer.Argument(..., help="Portfolio identifier."), tag: str = typer.Option(..., "--tag", help="Tag value."), ) -> None: detail = _with_service( ctx, lambda service, app_context: service.add_tag(portfolio_id, PortfolioTagMutationRequest(tag=tag)), ) render_output(ctx.obj.console, ctx.obj.output_format, detail) @tag_app.command("list") def tag_list( ctx: typer.Context, portfolio_id: str = typer.Argument(..., help="Portfolio identifier."), ) -> None: detail = _with_service(ctx, lambda service, app_context: service.list_tags(portfolio_id)) render_output(ctx.obj.console, ctx.obj.output_format, detail, empty_message="No tags found.") @tag_app.command("remove") def tag_remove( ctx: typer.Context, portfolio_id: str = typer.Argument(..., help="Portfolio identifier."), tag: str = typer.Option(..., "--tag", help="Tag value."), force: bool = typer.Option(False, "--force", help="Skip confirmation prompt."), ) -> None: _confirm(force, f"Remove tag {tag!r} from portfolio {portfolio_id}?") _with_service( ctx, lambda service, app_context: service.remove_tag(portfolio_id, PortfolioTagMutationRequest(tag=tag)), ) if ctx.obj.output_format == "json": render_output(ctx.obj.console, ctx.obj.output_format, {"portfolio_id": portfolio_id, "deleted": tag.strip()}) else: ctx.obj.console.print(f"Removed tag: {tag.strip()}")