first commit

arbitrage/config.py

@@ -0,0 +1,76 @@
+from __future__ import annotations
+
+from decimal import Decimal
+from enum import Enum
+from pathlib import Path
+
+from pydantic import Field, SecretStr
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+
+class Mode(str, Enum):
+    PAPER = "paper"
+    LIVE = "live"
+
+
+class Settings(BaseSettings):
+    model_config = SettingsConfigDict(
+        env_file=".env",
+        env_file_encoding="utf-8",
+        env_prefix="ARB_",
+        extra="ignore",
+    )
+
+    mode: Mode = Mode.PAPER
+
+    clob_host: str = "https://clob.polymarket.com"
+    gamma_host: str = "https://gamma-api.polymarket.com"
+    data_host: str = "https://data-api.polymarket.com"
+    ws_url: str = "wss://ws-subscriptions-clob.polymarket.com/ws/market"
+    polygon_rpc: str = "https://polygon-rpc.com"
+    proxy: str | None = None
+
+    private_key: SecretStr | None = None
+    funder_address: str | None = None
+    signature_type: int = 0
+    api_key: SecretStr | None = None
+    api_secret: SecretStr | None = None
+    api_passphrase: SecretStr | None = None
+
+    min_net_edge_bps: int = 50
+    max_basket_usd: Decimal = Field(default=Decimal("50"))
+    max_open_baskets: int = 3
+    max_open_baskets_per_event: int = 1
+    daily_loss_stop_usd: Decimal = Field(default=Decimal("100"))
+    kill_switch_file: Path = Path("./KILL")
+    resolution_skip_hours: int = 24
+
+    paper_latency_ms: int = 250
+
+    db_path: Path = Path("./arbitrage.db")
+    web_host: str = "127.0.0.1"
+    web_port: int = 8000
+    log_level: str = "INFO"
+
+    @property
+    def is_live(self) -> bool:
+        return self.mode == Mode.LIVE
+
+    def require_live_credentials(self) -> None:
+        if not self.is_live:
+            return
+        missing = [
+            name
+            for name, val in [
+                ("ARB_PRIVATE_KEY", self.private_key),
+                ("ARB_FUNDER_ADDRESS", self.funder_address),
+            ]
+            if not val
+        ]
+        if missing:
+            raise RuntimeError(
+                f"Live mode requires: {', '.join(missing)}. Set them in .env or run paper mode."
+            )
+
+
+settings = Settings()