pykalshi 0.1.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

pykalshi/__init__.py

@@ -0,0 +1,144 @@
+"""
+Kalshi API Client Library
+
+A clean, modular interface for the Kalshi trading API.
+"""
+
+import logging
+
+from .client import KalshiClient
+from .events import Event
+from .markets import Market, Series
+from .orders import Order
+from .portfolio import Portfolio
+from .exchange import Exchange
+from .api_keys import APIKeys
+from .feed import (
+    Feed,
+    TickerMessage,
+    OrderbookSnapshotMessage,
+    OrderbookDeltaMessage,
+    OrderbookMessage,
+    TradeMessage,
+    FillMessage,
+    PositionMessage,
+    MarketLifecycleMessage,
+    OrderGroupUpdateMessage,
+)
+from .enums import (
+    Side,
+    Action,
+    OrderType,
+    OrderStatus,
+    MarketStatus,
+    CandlestickPeriod,
+    TimeInForce,
+    SelfTradePrevention,
+)
+from .models import (
+    PositionModel,
+    FillModel,
+    OrderModel,
+    BalanceModel,
+    MarketModel,
+    EventModel,
+    OrderbookResponse,
+    CandlestickResponse,
+    ExchangeStatus,
+    Announcement,
+    APILimits,
+    APIKey,
+    GeneratedAPIKey,
+    SeriesModel,
+    TradeModel,
+    SettlementModel,
+    QueuePositionModel,
+    OrderGroupModel,
+    SubaccountModel,
+    SubaccountBalanceModel,
+    SubaccountTransferModel,
+    ForecastPercentileHistory,
+)
+from .orderbook import OrderbookManager
+from .rate_limiter import RateLimiter, NoOpRateLimiter
+from .exceptions import (
+    KalshiError,
+    KalshiAPIError,
+    AuthenticationError,
+    InsufficientFundsError,
+    ResourceNotFoundError,
+    RateLimitError,
+    OrderRejectedError,
+)
+
+# Set up logging to NullHandler by default to avoid "No handler found" warnings.
+logging.getLogger(__name__).addHandler(logging.NullHandler())
+
+__all__ = [
+    # Client
+    "KalshiClient",
+    # Domain objects
+    "Event",
+    "Market",
+    "Series",
+    "Order",
+    "Portfolio",
+    "Exchange",
+    "APIKeys",
+    # Feed (WebSocket)
+    "Feed",
+    "TickerMessage",
+    "OrderbookSnapshotMessage",
+    "OrderbookDeltaMessage",
+    "OrderbookMessage",
+    "TradeMessage",
+    "FillMessage",
+    "PositionMessage",
+    "MarketLifecycleMessage",
+    "OrderGroupUpdateMessage",
+    # Enums
+    "Side",
+    "Action",
+    "OrderType",
+    "OrderStatus",
+    "MarketStatus",
+    "CandlestickPeriod",
+    "TimeInForce",
+    "SelfTradePrevention",
+    # Models
+    "PositionModel",
+    "FillModel",
+    "OrderModel",
+    "BalanceModel",
+    "MarketModel",
+    "EventModel",
+    "OrderbookResponse",
+    "CandlestickResponse",
+    "ExchangeStatus",
+    "Announcement",
+    "APILimits",
+    "APIKey",
+    "GeneratedAPIKey",
+    "SeriesModel",
+    "TradeModel",
+    "SettlementModel",
+    "QueuePositionModel",
+    "OrderGroupModel",
+    "ForecastPercentileHistory",
+    # Utilities
+    "OrderbookManager",
+    "RateLimiter",
+    "NoOpRateLimiter",
+    # Subaccount Models
+    "SubaccountModel",
+    "SubaccountBalanceModel",
+    "SubaccountTransferModel",
+    # Exceptions
+    "KalshiError",
+    "KalshiAPIError",
+    "AuthenticationError",
+    "InsufficientFundsError",
+    "ResourceNotFoundError",
+    "RateLimitError",
+    "OrderRejectedError",
+]

pykalshi/api_keys.py

@@ -0,0 +1,59 @@
+from __future__ import annotations
+from typing import TYPE_CHECKING
+from .models import APIKey, GeneratedAPIKey, APILimits
+
+if TYPE_CHECKING:
+    from .client import KalshiClient
+
+
+class APIKeys:
+    """API key management and account limits."""
+
+    def __init__(self, client: KalshiClient) -> None:
+        self._client = client
+
+    def list(self) -> list[APIKey]:
+        """List all API keys for this account."""
+        data = self._client.get("/api_keys")
+        return [APIKey.model_validate(k) for k in data.get("api_keys", [])]
+
+    def create(self, public_key: str, name: str | None = None) -> APIKey:
+        """Create an API key with a provided RSA public key.
+
+        Args:
+            public_key: PEM-encoded RSA public key.
+            name: Optional name for the key.
+        """
+        body: dict = {"public_key": public_key}
+        if name:
+            body["name"] = name
+        data = self._client.post("/api_keys", body)
+        return APIKey.model_validate(data["api_key"])
+
+    def generate(self, name: str | None = None) -> GeneratedAPIKey:
+        """Generate a new API key pair (Kalshi creates both keys).
+
+        Returns a GeneratedAPIKey with the private_key field populated.
+        The private key is only returned ONCE - store it securely.
+
+        Args:
+            name: Optional name for the key.
+        """
+        body: dict = {}
+        if name:
+            body["name"] = name
+        data = self._client.post("/api_keys/generate", body)
+        return GeneratedAPIKey.model_validate(data)
+
+    def delete(self, key_id: str) -> None:
+        """Delete an API key.
+
+        Args:
+            key_id: The API key ID to delete.
+        """
+        self._client.delete(f"/api_keys/{key_id}")
+
+    def get_limits(self) -> APILimits:
+        """Get API rate limits for this account."""
+        data = self._client.get("/account/limits")
+        return APILimits.model_validate(data)