Phase 1: initial codes

stockmind/brokers/schwab.py

@@ -0,0 +1,105 @@
+from __future__ import annotations
+from brokers.base_broker import BaseBroker, Position, AccountBalance, Order
+
+
+class SchwabBroker(BaseBroker):
+    name = "schwab"
+
+    def __init__(self, app_key: str, app_secret: str):
+        self._app_key = app_key
+        self._app_secret = app_secret
+        self._client = None
+        self._connected = False
+
+    def connect(self) -> bool:
+        if not self._app_key or not self._app_secret:
+            return False
+        try:
+            import schwabdev
+            self._client = schwabdev.Client(self._app_key, self._app_secret)
+            self._connected = True
+            return True
+        except Exception as e:
+            self._connected = False
+            raise RuntimeError(f"Schwab connection failed: {e}")
+
+    def is_connected(self) -> bool:
+        return self._connected and self._client is not None
+
+    def get_positions(self) -> list[Position]:
+        if not self.is_connected():
+            return []
+        try:
+            response = self._client.account_linked().json()
+            account_hash = response[0].get("hashValue", "")
+            positions_resp = self._client.account(account_hash, fields="positions").json()
+            raw_positions = positions_resp.get("securitiesAccount", {}).get("positions", [])
+            result = []
+            for p in raw_positions:
+                instr = p.get("instrument", {})
+                symbol = instr.get("symbol", "")
+                shares = float(p.get("longQuantity", 0))
+                avg_cost = float(p.get("averagePrice", 0))
+                market_value = float(p.get("marketValue", 0))
+                current_price = market_value / shares if shares else 0
+                cost_basis = avg_cost * shares
+                pnl = market_value - cost_basis
+                pnl_pct = (pnl / cost_basis * 100) if cost_basis else 0
+                result.append(Position(
+                    symbol=symbol,
+                    shares=shares,
+                    avg_cost=avg_cost,
+                    market_value=market_value,
+                    current_price=current_price,
+                    pnl=pnl,
+                    pnl_pct=pnl_pct,
+                    description=instr.get("description", ""),
+                ))
+            return result
+        except Exception:
+            return []
+
+    def get_balance(self) -> AccountBalance:
+        if not self.is_connected():
+            return AccountBalance()
+        try:
+            response = self._client.account_linked().json()
+            account_hash = response[0].get("hashValue", "")
+            acct = self._client.account(account_hash).json()
+            balances = acct.get("securitiesAccount", {}).get("currentBalances", {})
+            return AccountBalance(
+                total_equity=float(balances.get("equity", 0)),
+                cash=float(balances.get("cashBalance", 0)),
+                buying_power=float(balances.get("buyingPower", 0)),
+                day_pnl=float(acct.get("securitiesAccount", {}).get("currentBalances", {}).get("dayProfitLoss", 0)),
+            )
+        except Exception:
+            return AccountBalance()
+
+    def get_orders(self, limit: int = 25) -> list[Order]:
+        if not self.is_connected():
+            return []
+        try:
+            response = self._client.account_linked().json()
+            account_hash = response[0].get("hashValue", "")
+            from datetime import datetime, timedelta
+            from_time = (datetime.now() - timedelta(days=30)).strftime("%Y-%m-%dT%H:%M:%S+0000")
+            to_time = datetime.now().strftime("%Y-%m-%dT%H:%M:%S+0000")
+            orders_resp = self._client.account_orders(account_hash, from_time, to_time, max_results=limit).json()
+            result = []
+            for o in orders_resp:
+                legs = o.get("orderLegCollection", [{}])
+                instr = legs[0].get("instrument", {}) if legs else {}
+                result.append(Order(
+                    order_id=str(o.get("orderId", "")),
+                    symbol=instr.get("symbol", ""),
+                    side=legs[0].get("instruction", "").lower() if legs else "",
+                    quantity=float(o.get("quantity", 0)),
+                    price=float(o.get("price", 0)),
+                    status=o.get("status", ""),
+                    order_type=o.get("orderType", ""),
+                    placed_at=o.get("enteredTime", ""),
+                ))
+            return result
+        except Exception:
+            return []