robin-sdk 0.1.0__py3-none-any.whl

robin_sdk/__init__.py

@@ -0,0 +1,37 @@
+"""
+Robin SDK — Python SDK for Robinhood MCP Trading API.
+
+Zero third-party runtime dependencies. Uses only the Python standard library.
+
+Usage:
+    >>> from robin_sdk import RobinClient
+    >>> async with RobinClient(client_id="your_id") as client:
+    ...     accounts = await client.get_accounts()
+"""
+
+__version__ = "0.1.0"
+
+from robin_sdk.client import RobinClient
+from robin_sdk.exceptions import (
+    RobinSDKError,
+    AuthenticationError,
+    TokenExpiredError,
+    TransportError,
+    APIError,
+    OrderError,
+    AccountError,
+    ValidationError,
+)
+
+__all__ = [
+    "RobinClient",
+    "RobinSDKError",
+    "AuthenticationError",
+    "TokenExpiredError",
+    "TransportError",
+    "APIError",
+    "OrderError",
+    "AccountError",
+    "ValidationError",
+    "__version__",
+]

robin_sdk/_compat.py

@@ -0,0 +1,31 @@
+"""
+Python version compatibility shims.
+
+Keeps the rest of the codebase clean by centralising version checks here.
+"""
+
+from __future__ import annotations
+
+import sys
+
+PYTHON_39 = sys.version_info >= (3, 9)
+PYTHON_310 = sys.version_info >= (3, 10)
+PYTHON_311 = sys.version_info >= (3, 11)
+
+
+def removeprefix(s: str, prefix: str) -> str:
+    """str.removeprefix backport for Python < 3.9."""
+    if PYTHON_39:
+        return s.removeprefix(prefix)  # type: ignore[attr-defined]
+    if s.startswith(prefix):
+        return s[len(prefix):]
+    return s
+
+
+def removesuffix(s: str, suffix: str) -> str:
+    """str.removesuffix backport for Python < 3.9."""
+    if PYTHON_39:
+        return s.removesuffix(suffix)  # type: ignore[attr-defined]
+    if suffix and s.endswith(suffix):
+        return s[:-len(suffix)]
+    return s

robin_sdk/_types.py

@@ -0,0 +1,231 @@
+"""
+Type definitions for Robin SDK.
+
+Uses TypedDict for structured response data so callers get editor
+autocompletion without requiring runtime dependencies.
+"""
+
+from __future__ import annotations
+
+import sys
+from typing import Any, Dict, List, Optional
+
+if sys.version_info >= (3, 12):
+    from typing import TypedDict, NotRequired
+elif sys.version_info >= (3, 11):
+    from typing import TypedDict, NotRequired
+else:
+    from typing import TypedDict
+
+    # Python 3.9-3.10 backport for NotRequired
+    class _NotRequiredMeta(type):
+        def __getitem__(cls, item):
+            return item
+
+    class NotRequired(metaclass=_NotRequiredMeta):  # type: ignore[no-redef]
+        pass
+
+
+# ---------------------------------------------------------------------------
+# OAuth / Auth types
+# ---------------------------------------------------------------------------
+
+class TokenData(TypedDict):
+    """Cached token data stored on disk."""
+    access_token: str
+    refresh_token: Optional[str]
+    timestamp: str
+    expires_in: int
+
+
+class OAuthEndpoints(TypedDict):
+    """Discovered OAuth2 endpoint URLs."""
+    auth_url: str
+    token_url: str
+    scopes: List[str]
+
+
+# ---------------------------------------------------------------------------
+# Account & Portfolio types
+# ---------------------------------------------------------------------------
+
+class Account(TypedDict, total=False):
+    """A Robinhood brokerage account."""
+    account_number: str
+    id: str
+    brokerage_account_type: str
+    type: str
+    nickname: str
+    is_default: bool
+    agentic_allowed: bool
+
+
+class PortfolioData(TypedDict, total=False):
+    """Portfolio snapshot values."""
+    portfolio_value: float
+    cash_value: float
+    buying_power: float
+    equity_value: float
+    market_value: float
+    total_value: float
+    total_equity: float
+    extended_hours_equity: float
+    dividend_total: float
+    total_return: float
+
+
+# ---------------------------------------------------------------------------
+# Equity types
+# ---------------------------------------------------------------------------
+
+class EquityPosition(TypedDict, total=False):
+    """An open equity position."""
+    symbol: str
+    ticker: str
+    quantity: float
+    average_buy_price: float
+    value: float
+
+
+class EquityQuote(TypedDict, total=False):
+    """Real-time equity quote."""
+    symbol: str
+    last_trade_price: float
+    previous_close: float
+    bid_price: float
+    ask_price: float
+
+
+class EquityOrder(TypedDict, total=False):
+    """An equity order record."""
+    order_id: str
+    id: str
+    state: str
+    status: str
+    side: str
+    symbol: str
+    quantity: str
+    price: str
+    average_fill_price: str
+    created_at: str
+    updated_at: str
+
+
+class TradabilityInfo(TypedDict, total=False):
+    """Whether a symbol can be traded."""
+    symbol: str
+    tradable: bool
+    fractional: bool
+
+
+# ---------------------------------------------------------------------------
+# Option types
+# ---------------------------------------------------------------------------
+
+class OptionChain(TypedDict, total=False):
+    """Option chain data for a symbol."""
+    symbol: str
+    expiration_dates: List[str]
+    chain_id: str
+
+
+class OptionInstrument(TypedDict, total=False):
+    """An individual option contract."""
+    instrument_id: str
+    symbol: str
+    expiration_date: str
+    strike_price: float
+    option_type: str
+
+
+class OptionQuote(TypedDict, total=False):
+    """Real-time option quote."""
+    instrument_id: str
+    mark_price: float
+    bid_price: float
+    ask_price: float
+
+
+class OptionPosition(TypedDict, total=False):
+    """An open or closed option position."""
+    instrument_id: str
+    symbol: str
+    quantity: float
+    average_price: float
+
+
+class OptionOrder(TypedDict, total=False):
+    """An option order record."""
+    order_id: str
+    id: str
+    state: str
+    side: str
+    quantity: str
+    price: str
+    created_at: str
+
+
+# ---------------------------------------------------------------------------
+# Watchlist types
+# ---------------------------------------------------------------------------
+
+class Watchlist(TypedDict, total=False):
+    """A user watchlist."""
+    watchlist_id: str
+    id: str
+    name: str
+    item_count: int
+
+
+class WatchlistItem(TypedDict, total=False):
+    """An item in a watchlist."""
+    symbol: str
+    instrument_id: str
+
+
+# ---------------------------------------------------------------------------
+# Market Data types
+# ---------------------------------------------------------------------------
+
+class HistoricalBar(TypedDict, total=False):
+    """An OHLCV price bar."""
+    begins_at: str
+    open_price: float
+    close_price: float
+    high_price: float
+    low_price: float
+    volume: int
+
+
+class IndexData(TypedDict, total=False):
+    """Market index information."""
+    symbol: str
+    name: str
+    value: float
+
+
+# ---------------------------------------------------------------------------
+# JSON-RPC types (internal)
+# ---------------------------------------------------------------------------
+
+class JSONRPCRequest(TypedDict):
+    """A JSON-RPC 2.0 request."""
+    jsonrpc: str
+    id: int
+    method: str
+    params: Dict[str, Any]
+
+
+class JSONRPCError(TypedDict, total=False):
+    """A JSON-RPC 2.0 error object."""
+    code: int
+    message: str
+    data: Any
+
+
+class JSONRPCResponse(TypedDict, total=False):
+    """A JSON-RPC 2.0 response."""
+    jsonrpc: str
+    id: int
+    result: Any
+    error: JSONRPCError